diff --git a/dtmgrpc/type.go b/dtmgrpc/type.go
index 8ff2d23..a2d3042 100644
--- a/dtmgrpc/type.go
+++ b/dtmgrpc/type.go
@@ -94,3 +94,8 @@ func Result2Error(res interface{}, err error) error {
 	}
 	return e
 }
+
+// SetCurrentDBType set the current db type
+func SetCurrentDBType(dbType string) {
+	dtmcli.SetCurrentDBType(dbType)
+}