fix: mysql数据库为空时无法备份

tnb-labs · Oct 28, 2024 · 9541ff8 · 9541ff8
1 parent 8ad7867
commit 9541ff8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pkg/db/mysql.go b/pkg/db/mysql.go
@@ -94,7 +94,7 @@ func (m *MySQL) DatabaseExists(name string) (bool, error) {
 
 func (m *MySQL) DatabaseSize(name string) (int64, error) {
 	var size int64
-	err := m.QueryRow(fmt.Sprintf("SELECT SUM(data_length + index_length) FROM information_schema.tables WHERE table_schema = '%s'", name)).Scan(&size)
+	err := m.QueryRow(fmt.Sprintf("SELECT COALESCE(SUM(data_length) + SUM(index_length), 0) FROM information_schema.tables WHERE table_schema = '%s'", name)).Scan(&size)
 	return size, err
 }