add UseDriver

dtmgrpc/type.go

@@ -12,6 +12,7 @@ import (
 	"github.com/yedf/dtm/dtmcli"
 	"github.com/yedf/dtm/dtmcli/dtmimp"
 	"github.com/yedf/dtm/dtmgrpc/dtmgimp"
+	"github.com/yedf/dtmdriver"
 	emptypb "google.golang.org/protobuf/types/known/emptypb"
 )
 
@@ -32,3 +33,7 @@ func SetCurrentDBType(dbType string) {
 func GetCurrentDBType() string {
 	return dtmcli.GetCurrentDBType()
 }
+
+func UseDriver(driverName string) error {
+	return dtmdriver.Use(driverName)
+}

dtmgrpc/type_test.go

@@ -24,4 +24,7 @@ func TestType(t *testing.T) {
 	old := GetCurrentDBType()
 	SetCurrentDBType(dtmcli.DBTypeMysql)
 	SetCurrentDBType(old)
+
+	err = UseDriver("default")
+	assert.Nil(t, err)
 }