ftr: optimize the datasource init (#447)

* optimize the datasource init

pkg/client/client.go

@@ -20,6 +20,8 @@ package client
 import (
 	"sync"
 
+	"github.com/seata/seata-go/pkg/datasource"
+
 	at "github.com/seata/seata-go/pkg/datasource/sql"
 	"github.com/seata/seata-go/pkg/integration"
 	"github.com/seata/seata-go/pkg/remoting/getty"
@@ -41,11 +43,13 @@ func InitPath(configFilePath string) {
 
 	initRmClient(cfg)
 	initTmClient(cfg)
+	initDatasource(cfg)
 }
 
 var (
-	onceInitTmClient sync.Once
-	onceInitRmClient sync.Once
+	onceInitTmClient   sync.Once
+	onceInitRmClient   sync.Once
+	onceInitDatasource sync.Once
 )
 
 // InitTmClient init client tm client
@@ -80,3 +84,9 @@ func initRmClient(cfg *Config) {
 		at.InitAT(cfg.ClientConfig.UndoConfig, cfg.AsyncWorkerConfig)
 	})
 }
+
+func initDatasource(cfg *Config) {
+	onceInitDatasource.Do(func() {
+		datasource.Init()
+	})
+}

pkg/datasource/init.go

@@ -0,0 +1,24 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements.  See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License.  You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * 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 datasource
+
+import sql2 "github.com/seata/seata-go/pkg/datasource/sql"
+
+func Init() {
+	sql2.Init()
+}

pkg/datasource/sql/conn_at_test.go

@@ -33,6 +33,11 @@ import (
 	"github.com/stretchr/testify/assert"
 )
 
+func TestMain(m *testing.M) {
+	Init()
+	m.Run()
+}
+
 func initAtConnTestResource(t *testing.T) (*gomock.Controller, *sql.DB, *mockSQLInterceptor, *mockTxHook) {
 	ctrl := gomock.NewController(t)
 

pkg/datasource/sql/driver.go

@@ -41,7 +41,7 @@ const (
 	SeataXAMySQLDriver = "seata-xa-mysql"
 )
 
-func init() {
+func initDriver() {
 	sql.Register(SeataATMySQLDriver, &seataATDriver{
 		seataDriver: &seataDriver{
 			transType: types.ATMode,

pkg/datasource/sql/exec/at/at_executor.go

@@ -28,7 +28,7 @@ import (
 	"github.com/seata/seata-go/pkg/datasource/sql/types"
 )
 
-func init() {
+func Init() {
 	exec.RegisterATExecutor(types.DBTypeMySQL, func() exec.SQLExecutor { return &AtExecutor{} })
 }
 

pkg/datasource/sql/exec/executor.go

@@ -23,14 +23,8 @@ import (
 
 	"github.com/seata/seata-go/pkg/datasource/sql/parser"
 	"github.com/seata/seata-go/pkg/datasource/sql/types"
-	"github.com/seata/seata-go/pkg/datasource/sql/undo"
-	"github.com/seata/seata-go/pkg/datasource/sql/undo/builder"
 )
 
-func init() {
-	undo.RegisterUndoLogBuilder(types.MultiExecutor, builder.GetMySQLMultiUndoLogBuilder)
-}
-
 var (
 	atExecutors = make(map[types.DBType]func() SQLExecutor)
 	xaExecutors = make(map[types.DBType]func() SQLExecutor)

pkg/datasource/sql/exec/xa/default.go

@@ -22,7 +22,7 @@ import (
 	"github.com/seata/seata-go/pkg/datasource/sql/types"
 )
 
-func init() {
+func Init() {
 	exec.RegisterXAExecutor(types.DBTypeMySQL, func() exec.SQLExecutor {
 		return &XAExecutor{}
 	})

pkg/datasource/sql/hook/logger_hook.go

@@ -21,13 +21,14 @@ import (
 	"context"
 
 	"github.com/seata/seata-go/pkg/datasource/sql/exec"
+
 	"github.com/seata/seata-go/pkg/datasource/sql/types"
 	"github.com/seata/seata-go/pkg/util/log"
 	"go.uber.org/zap"
 )
 
-func init() {
-	exec.RegisterHook(&loggerSQLHook{})
+func NewLoggerSQLHook() exec.SQLHook {
+	return &loggerSQLHook{}
 }
 
 type loggerSQLHook struct{}

pkg/datasource/sql/hook/undo_log_hook.go

@@ -27,8 +27,8 @@ import (
 	"github.com/seata/seata-go/pkg/tm"
 )
 
-func init() {
-	exec.RegisterHook(&undoLogSQLHook{})
+func NewUndoLogSQLHook() exec.SQLHook {
+	return &undoLogSQLHook{}
 }
 
 type undoLogSQLHook struct {

pkg/datasource/sql/plugin.go

@@ -18,12 +18,43 @@
 package sql
 
 import (
-	_ "github.com/seata/seata-go/pkg/datasource/sql/hook"
+	"github.com/seata/seata-go/pkg/datasource/sql/exec"
+	"github.com/seata/seata-go/pkg/datasource/sql/exec/at"
+	"github.com/seata/seata-go/pkg/datasource/sql/exec/xa"
+	"github.com/seata/seata-go/pkg/datasource/sql/hook"
+	"github.com/seata/seata-go/pkg/datasource/sql/types"
+	"github.com/seata/seata-go/pkg/datasource/sql/undo"
+	"github.com/seata/seata-go/pkg/datasource/sql/undo/builder"
+	"github.com/seata/seata-go/pkg/datasource/sql/undo/mysql"
+)
 
-	// mysql 相关插件
-	_ "github.com/seata/seata-go/pkg/datasource/sql/datasource/mysql"
-	_ "github.com/seata/seata-go/pkg/datasource/sql/undo/mysql"
+func Init() {
+	hookRegister()
+	executorRegister()
+	undoInit()
+	initDriver()
+}
 
-	_ "github.com/seata/seata-go/pkg/datasource/sql/exec/at"
-	_ "github.com/seata/seata-go/pkg/datasource/sql/exec/xa"
-)
+func hookRegister() {
+	exec.RegisterHook(hook.NewLoggerSQLHook())
+	exec.RegisterHook(hook.NewUndoLogSQLHook())
+}
+
+func executorRegister() {
+	at.Init()
+	xa.Init()
+}
+
+func undoInit() {
+	mysqlUndoLogInit()
+}
+
+func mysqlUndoLogInit() {
+	mysql.InitUndoLogManager()
+
+	undo.RegisterUndoLogBuilder(types.DeleteExecutor, builder.GetMySQLDeleteUndoLogBuilder)
+	undo.RegisterUndoLogBuilder(types.InsertExecutor, builder.GetMySQLInsertUndoLogBuilder)
+	undo.RegisterUndoLogBuilder(types.InsertOnDuplicateExecutor, builder.GetMySQLInsertOnDuplicateUndoLogBuilder)
+	undo.RegisterUndoLogBuilder(types.MultiExecutor, builder.GetMySQLMultiUndoLogBuilder)
+	undo.RegisterUndoLogBuilder(types.UpdateExecutor, builder.GetMySQLUpdateUndoLogBuilder)
+}

pkg/datasource/sql/undo/builder/mysql_delete_undo_log_builder.go

@@ -33,10 +33,6 @@ import (
 	"github.com/seata/seata-go/pkg/util/log"
 )
 
-func init() {
-	undo.RegisterUndoLogBuilder(types.DeleteExecutor, GetMySQLDeleteUndoLogBuilder)
-}
-
 type MySQLDeleteUndoLogBuilder struct {
 	BasicUndoLogBuilder
 }

pkg/datasource/sql/undo/builder/mysql_insert_undo_log_builder.go

@@ -34,10 +34,6 @@ const (
 	SqlPlaceholder = "?"
 )
 
-func init() {
-	undo.RegisterUndoLogBuilder(types.InsertExecutor, GetMySQLInsertUndoLogBuilder)
-}
-
 type MySQLInsertUndoLogBuilder struct {
 	BasicUndoLogBuilder
 	// InsertResult after insert sql

pkg/datasource/sql/undo/builder/mysql_insertonduplicate_update_undo_log_builder.go

@@ -30,10 +30,6 @@ import (
 	"github.com/seata/seata-go/pkg/util/log"
 )
 
-func init() {
-	undo.RegisterUndoLogBuilder(types.InsertOnDuplicateExecutor, GetMySQLInsertOnDuplicateUndoLogBuilder)
-}
-
 type MySQLInsertOnDuplicateUndoLogBuilder struct {
 	MySQLInsertUndoLogBuilder
 	BeforeSelectSql           string

pkg/datasource/sql/undo/builder/mysql_multi_undo_log_builder.go

@@ -24,10 +24,6 @@ import (
 	"github.com/seata/seata-go/pkg/datasource/sql/undo"
 )
 
-func init() {
-	undo.RegisterUndoLogBuilder(types.MultiExecutor, GetMySQLMultiUndoLogBuilder)
-}
-
 type MySQLMultiUndoLogBuilder struct {
 	BasicUndoLogBuilder
 	beforeImages []*types.RecordImage

pkg/datasource/sql/undo/builder/mysql_update_undo_log_builder.go

@@ -39,10 +39,6 @@ const (
 	OnlyCareUpdateColumns = true
 )
 
-func init() {
-	undo.RegisterUndoLogBuilder(types.UpdateExecutor, GetMySQLUpdateUndoLogBuilder)
-}
-
 type MySQLUpdateUndoLogBuilder struct {
 	BasicUndoLogBuilder
 }

pkg/datasource/sql/undo/mysql/default.go

@@ -23,7 +23,7 @@ import (
 	"github.com/seata/seata-go/pkg/datasource/sql/undo/base"
 )
 
-func init() {
+func InitUndoLogManager() {
 	if err := undo.RegisterUndoLogManager(&undoLogManager{Base: base.NewBaseUndoLogManager()}); err != nil {
 		panic(errors.WithStack(err))
 	}