feat(dbm-services): 模拟执行只备份导入变更SQL相关库 TencentBlueKing#8318

dbm-services/mysql/db-simulation/app/keyworld/mysql56.go

@@ -21,7 +21,6 @@ var Keywords56 = []Keyword{
 	{"ALGORITHM", false},
 	{"ALL", false},
 	{"ALTER", false},
-	{"ANALYSE", false},
 	{"ANALYZE", false},
 	{"AND", false},
 	{"ANY", false},

dbm-services/mysql/db-simulation/app/keyworld/mysql57.go

@@ -23,7 +23,6 @@ var Keywords57 = []Keyword{
 	{"ALL", false},
 	{"ALTER", true},
 	{"ALWAYS", false},
-	{"ANALYSE", false},
 	{"ANALYZE", false},
 	{"AND", false},
 	{"ANY", false},

dbm-services/mysql/db-simulation/app/keyworld/mysql80.go

@@ -25,7 +25,6 @@ var Keywords80 = []Keyword{
 	{"ALL", false},
 	{"ALTER", true},
 	{"ALWAYS", false},
-	{"ANALYSE", false},
 	{"ANALYZE", false},
 	{"AND", false},
 	{"ANY", false},

dbm-services/mysql/db-simulation/app/syntax/parse_relation_db.go

@@ -0,0 +1,173 @@
+/*
+ * TencentBlueKing is pleased to support the open source community by making 蓝鲸智云-DB管理系统(BlueKing-BK-DBM) available.
+ * Copyright (C) 2017-2023 THL A29 Limited, a Tencent company. All rights reserved.
+ * Licensed under the MIT License (the "License"); you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at https://opensource.org/licenses/MIT
+ * Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on
+ * an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the
+ * specific language governing permissions and limitations under the License.
+ */
+
+package syntax
+
+import (
+	"bufio"
+	"encoding/json"
+	"errors"
+	"io"
+	"os"
+	"runtime/debug"
+	"slices"
+	"sync"
+
+	"github.com/samber/lo"
+
+	"dbm-services/common/go-pubpkg/logger"
+)
+
+// DoParseRelationDbs parse relation db from sql file
+func (tf *TmysqlParseFile) DoParseRelationDbs(version string) (createDbs, relationDbs []string, dumpAll bool,
+	err error) {
+	logger.Info("doing....")
+	tf.result = make(map[string]*CheckInfo)
+	tf.tmpWorkdir = tf.BaseWorkdir
+	tf.mu = sync.Mutex{}
+
+	if !tf.IsLocalFile {
+		if err = tf.Init(); err != nil {
+			logger.Error("Do init failed %s", err.Error())
+			return nil, nil, false, err
+		}
+		if err = tf.Downloadfile(); err != nil {
+			logger.Error("failed to download sql file from the product library %s", err.Error())
+			return nil, nil, false, err
+		}
+	}
+	// 最后删除临时目录,不会返回错误
+	defer tf.delTempDir()
+
+	errChan := make(chan error, 1)
+	alreadExecutedSqlfileChan := make(chan string, len(tf.Param.FileNames))
+
+	go func() {
+		if err = tf.Execute(alreadExecutedSqlfileChan, version); err != nil {
+			logger.Error("failed to execute tmysqlparse: %s", err.Error())
+			errChan <- err
+		}
+		close(alreadExecutedSqlfileChan)
+	}()
+
+	logger.Info("start to analyze the parsing result")
+	createDbs, relationDbs, dumpAll, err = tf.doParseInchan(alreadExecutedSqlfileChan, version)
+	if err != nil {
+		logger.Error("failed to analyze the parsing result:%s", err.Error())
+		return nil, nil, false, err
+	}
+	logger.Info("createDbs:%v,relationDbs:%v,dumpAll:%v,err:%v", createDbs, relationDbs, dumpAll, err)
+	return createDbs, relationDbs, dumpAll, nil
+}
+
+// doParseInchan RelationDbs do parse relation db
+func (t *TmysqlParse) doParseInchan(alreadExecutedSqlfileCh chan string,
+	mysqlVersion string) (createDbs []string, relationDbs []string, dumpAll bool, err error) {
+	var errs []error
+	c := make(chan struct{}, 10)
+	errChan := make(chan error, 5)
+	wg := &sync.WaitGroup{}
+	stopChan := make(chan struct{})
+
+	for sqlfile := range alreadExecutedSqlfileCh {
+		wg.Add(1)
+		c <- struct{}{}
+		go func(fileName string) {
+			defer wg.Done()
+			cdbs, dbs, dumpAllDbs, err := t.analyzeRelationDbs(fileName, mysqlVersion)
+			logger.Info("createDbs:%v,dbs:%v,dumpAllDbs:%v,err:%v", cdbs, dbs, dumpAllDbs, err)
+			if err != nil {
+				errChan <- err
+			}
+			if dumpAllDbs {
+				dumpAll = true
+				stopChan <- struct{}{}
+			}
+			t.mu.Lock()
+			relationDbs = append(relationDbs, dbs...)
+			createDbs = append(createDbs, cdbs...)
+			t.mu.Unlock()
+			<-c
+		}(sqlfile)
+	}
+
+	go func() {
+		wg.Wait()
+		close(errChan)
+		stopChan <- struct{}{}
+	}()
+
+	for {
+		select {
+		case err := <-errChan:
+			errs = append(errs, err)
+		case <-stopChan:
+			return createDbs, relationDbs, dumpAll, errors.Join(errs...)
+		}
+	}
+}
+
+// analyzeRelationDbs 分析变更sql相关的db
+func (t *TmysqlParse) analyzeRelationDbs(inputfileName, mysqlVersion string) (
+	createDbs []string,
+	relationDbs []string,
+	dumpAll bool,
+	err error) {
+	defer func() {
+		if r := recover(); r != nil {
+			logger.Error("panic error:%v,stack:%s", r, string(debug.Stack()))
+		}
+	}()
+	t.result[inputfileName] = &CheckInfo{}
+	f, err := os.Open(t.getAbsoutputfilePath(inputfileName, mysqlVersion))
+	if err != nil {
+		logger.Error("open file failed %s", err.Error())
+		return nil, nil, false, err
+	}
+	defer f.Close()
+	reader := bufio.NewReader(f)
+	for {
+		line, errx := reader.ReadBytes(byte('\n'))
+		if errx != nil {
+			if errx == io.EOF {
+				break
+			}
+			logger.Error("read Line Error %s", errx.Error())
+			return nil, nil, false, errx
+		}
+		if len(line) == 1 && line[0] == byte('\n') {
+			continue
+		}
+		var res ParseLineQueryBase
+		if err = json.Unmarshal(line, &res); err != nil {
+			logger.Error("json unmasrshal line:%s failed %s", string(line), err.Error())
+			return nil, nil, false, err
+		}
+		// 判断是否有语法错误
+		if res.ErrorCode != 0 {
+			return nil, nil, false, err
+		}
+		if slices.Contains([]string{SQLTypeCreateProcedure, SQLTypeCreateFunction, SQLTypeCreateView, SQLTypeCreateTrigger},
+			res.Command) {
+			return nil, nil, true, nil
+		}
+		if lo.IsEmpty(res.DbName) {
+			continue
+		}
+		// create db not need dump db
+		if slices.Contains([]string{SQLTypeCreateDb}, res.Command) {
+			createDbs = append(createDbs, res.DbName)
+			continue
+		}
+		relationDbs = append(relationDbs, res.DbName)
+
+	}
+	return createDbs, relationDbs, false, nil
+}

dbm-services/mysql/db-simulation/app/syntax/syntax.go

@@ -301,7 +301,10 @@ func (tf *TmysqlParseFile) UploadDdlTblMapFile() (err error) {
 }
 
 func getSQLParseResultFile(fileName, version string) string {
-	return fmt.Sprintf("%s-%s.json", version, fileName)
+	if lo.IsNotEmpty(version) {
+		return fmt.Sprintf("%s-%s.json", version, fileName)
+	}
+	return fileName + ".json"
 }
 
 // getCommand generates the command string for running TmysqlParse

dbm-services/mysql/db-simulation/app/syntax/tmysqlpase_schema.go

@@ -43,6 +43,8 @@ const (
 	SQLTypeDelete = "delete"
 	// SQLTypeUpdate is update sql
 	SQLTypeUpdate = "update"
+	// SQLTypeUseDb is use db sql
+	SQLTypeUseDb = "change_db"
 )
 
 // ColDef mysql column definition