Skip to content

Commit

Permalink
feat(dbm-services): 模拟执行只备份导入变更SQL相关库 TencentBlueKing#8318
Browse files Browse the repository at this point in the history
  • Loading branch information
ymakedaq committed Dec 9, 2024
1 parent 87423dc commit 4f9a7f5
Show file tree
Hide file tree
Showing 18 changed files with 634 additions and 227 deletions.
1 change: 0 additions & 1 deletion dbm-services/mysql/db-simulation/app/keyworld/mysql56.go
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,6 @@ var Keywords56 = []Keyword{
{"ALGORITHM", false},
{"ALL", false},
{"ALTER", false},
{"ANALYSE", false},
{"ANALYZE", false},
{"AND", false},
{"ANY", false},
Expand Down
1 change: 0 additions & 1 deletion dbm-services/mysql/db-simulation/app/keyworld/mysql57.go
Original file line number Diff line number Diff line change
Expand Up @@ -23,7 +23,6 @@ var Keywords57 = []Keyword{
{"ALL", false},
{"ALTER", true},
{"ALWAYS", false},
{"ANALYSE", false},
{"ANALYZE", false},
{"AND", false},
{"ANY", false},
Expand Down
1 change: 0 additions & 1 deletion dbm-services/mysql/db-simulation/app/keyworld/mysql80.go
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,6 @@ var Keywords80 = []Keyword{
{"ALL", false},
{"ALTER", true},
{"ALWAYS", false},
{"ANALYSE", false},
{"ANALYZE", false},
{"AND", false},
{"ANY", false},
Expand Down
174 changes: 174 additions & 0 deletions dbm-services/mysql/db-simulation/app/syntax/parse_relation_db.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,174 @@
/*
* 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
}

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)
if err != nil {
errChan <- err
}
if dumpAllDbs {
dumpAll = true
stopChan <- struct{}{}
}
if len(dbs) > 0 {
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
}
5 changes: 4 additions & 1 deletion dbm-services/mysql/db-simulation/app/syntax/syntax.go
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
Loading

0 comments on commit 4f9a7f5

Please sign in to comment.