feat(mysql): mysql清档重命名增加flush tables TencentBlueKing#8718

xfwduke · Dec 19, 2024 · b04c17d · b04c17d
1 parent 0352875
commit b04c17d
Show file tree

Hide file tree

Showing 2 changed files with 39 additions and 0 deletions.
diff --git a/dbm-services/mysql/db-tools/dbactuator/pkg/components/rename_dbs/on_mysql.go b/dbm-services/mysql/db-tools/dbactuator/pkg/components/rename_dbs/on_mysql.go
@@ -8,6 +8,7 @@ import (
 	"os"
 	"os/exec"
 	"strings"
+	"time"
 
 	"dbm-services/common/go-pubpkg/logger"
 	"dbm-services/mysql/db-tools/dbactuator/pkg/components"
@@ -62,6 +63,10 @@ func (c *OnMySQLComponent) Do() error {
 }
 
 func (c *OnMySQLComponent) oneInstance(port int) error {
+	defer func() {
+		c.instanceFlushTable()
+	}()
+
 	err := c.instanceInit(port)
 	if err != nil {
 		return err
@@ -279,6 +284,21 @@ func (c *OnMySQLComponent) instanceDropFromDBs(port int) error {
 	return nil
 }
 
+func (c *OnMySQLComponent) instanceFlushTable() {
+	if c.dbConn != nil {
+		ctx, cancel := context.WithTimeout(context.Background(), 200*time.Second)
+		defer cancel()
+
+		_, err := c.dbConn.ExecContext(
+			ctx,
+			`FLUSH TABLES`,
+		)
+		if err != nil {
+			logger.Error("flush table failed: %v", err)
+		}
+	}
+}
+
 func (c *OnMySQLComponent) Example() interface{} {
 	f := false
 	return OnMySQLComponent{

diff --git a/dbm-services/mysql/db-tools/dbactuator/pkg/components/truncate/on_mysql.go b/dbm-services/mysql/db-tools/dbactuator/pkg/components/truncate/on_mysql.go
@@ -69,7 +69,26 @@ func (c *OnMySQLComponent) Truncate() error {
 	return nil
 }
 
+func (c *OnMySQLComponent) instanceFlushTable() {
+	if c.dbConn != nil {
+		ctx, cancel := context.WithTimeout(context.Background(), 200*time.Second)
+		defer cancel()
+
+		_, err := c.dbConn.ExecContext(
+			ctx,
+			`FLUSH TABLES`,
+		)
+		if err != nil {
+			logger.Error("flush table failed: %v", err)
+		}
+	}
+}
+
 func (c *OnMySQLComponent) oneInstance(port int) error {
+	defer func() {
+		c.instanceFlushTable()
+	}()
+
 	err := c.instanceInit(port)
 	if err != nil {
 		logger.Error("init instance %d truncate failed: %s", port, err.Error())