sql-parser/analyzer.go at main · Edsuns/sql-parser · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
package sqlparser

import (
	"errors"
	"sync"

	"github.com/Edsuns/sql-parser/analyzer"
	"github.com/Edsuns/sql-parser/internal/mysql"
	"github.com/Edsuns/sql-parser/internal/spark"
	"github.com/Edsuns/sql-parser/internal/starrocks"
	"github.com/Edsuns/sql-parser/internal/tidb"
)

type delegateSQLAnalyzer struct {
	once              sync.Once
	mysqlAnalyzer     analyzer.SQLAnalyzer
	sparkAnalyzer     analyzer.SQLAnalyzer
	starrocksAnalyzer analyzer.SQLAnalyzer
	tidbAnalyzer      analyzer.SQLAnalyzer
}

// NewAnalyzer 创建SQL分析器
func NewAnalyzer() analyzer.SQLAnalyzer {
	return &delegateSQLAnalyzer{}
}

func (d *delegateSQLAnalyzer) AnalyzeLineage(req *analyzer.AnalyzeReq) ([]*analyzer.LineageResult, error) {
	impl, err := d.route(req.Type)
	if err != nil {
		return nil, err
	}
	return impl.AnalyzeLineage(req)
}

func (d *delegateSQLAnalyzer) AnalyzeDDL(req *analyzer.AnalyzeReq) ([]*analyzer.DDLResult, error) {
	impl, err := d.route(req.Type)
	if err != nil {
		return nil, err
	}
	return impl.AnalyzeDDL(req)
}

func (d *delegateSQLAnalyzer) MakeCommentModification(req *analyzer.MakeCommentModificationReq) (string, error) {
	impl, err := d.route(req.Type)
	if err != nil {
		return "", err
	}
	return impl.MakeCommentModification(req)
}

func (d *delegateSQLAnalyzer) Split(req *analyzer.SplitReq) ([]string, error) {
	impl, err := d.route(req.Type)
	if err != nil {
		return nil, err
	}
	return impl.Split(req)
}

func (d *delegateSQLAnalyzer) route(t analyzer.EngineType) (analyzer.SQLAnalyzer, error) {
	d.once.Do(func() {
		d.mysqlAnalyzer = mysql.NewSQLAnalyzer()
		d.sparkAnalyzer = spark.NewSQLAnalyzer()
		d.starrocksAnalyzer = starrocks.NewSQLAnalyzer()
		d.tidbAnalyzer = tidb.NewSQLAnalyzer()
	})
	switch t {
	case analyzer.EngineMySQL:
		return d.mysqlAnalyzer, nil
	case analyzer.EngineSpark:
		return d.sparkAnalyzer, nil
	case analyzer.EngineStarRocks:
		return d.starrocksAnalyzer, nil
	case analyzer.EngineTiDB:
		return d.tidbAnalyzer, nil
	default:
		return nil, errors.New("unsupported engine type")
	}
}