Feature add has undo log table

pkg/constant/undo.go

@@ -23,4 +23,8 @@ const (
 	// UndoLogTableName Todo get from config
 	UndoLogTableName = DefaultTransactionUndoLogTable
 	DeleteUndoLogSql = DeleteFrom + UndoLogTableName + " WHERE " + UndoLogBranchXid + " = ? AND " + UndoLogXid + " = ?"
+	// CheckUndoLogTableExistSql check undo log if exist
+	CheckUndoLogTableExistSql = "SELECT 1 FROM " + UndoLogTableName + " LIMIT 1"
 )
+
+const ErrCodeTableNotExist = "1146"

pkg/datasource/sql/undo/base/undo.go

@@ -112,6 +112,22 @@ func (m *BaseUndoLogManager) DBType() types.DBType {
 	panic("implement me")
 }
 
+// HasUndoLogTable check undo log table if exist
+func (m *BaseUndoLogManager) HasUndoLogTable(ctx context.Context, conn *sql.Conn) (res bool, err error) {
+	_, err = conn.QueryContext(ctx, constant.CheckUndoLogTableExistSql)
+	// 1146 mysql table not exist fault code
+	if err != nil && strings.Contains(err.Error(), constant.ErrCodeTableNotExist) {
+		return false, nil
+	}
+
+	if err != nil {
+		log.Errorf("[HasUndoLogTable] query sql fail, err: %v", err)
+		return
+	}
+
+	return true, nil
+}
+
 // getBatchDeleteUndoLogSql build batch delete undo log
 func (m *BaseUndoLogManager) getBatchDeleteUndoLogSql(xid []string, branchID []int64) (string, error) {
 	if len(xid) == 0 || len(branchID) == 0 {

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

@@ -67,3 +67,8 @@ func (m *undoLogManager) RunUndo(xid string, branchID int64, conn *sql.Conn) err
 func (m *undoLogManager) DBType() types.DBType {
 	return types.DBTypeMySQL
 }
+
+// HasUndoLogTable check undo log table if exist
+func (m *undoLogManager) HasUndoLogTable(ctx context.Context, conn *sql.Conn) (bool, error) {
+	return m.Base.HasUndoLogTable(ctx, conn)
+}

pkg/datasource/sql/undo/undo.go

@@ -61,6 +61,8 @@ type UndoLogManager interface {
 	RunUndo(xid string, branchID int64, conn *sql.Conn) error
 	// DBType
 	DBType() types.DBType
+	// HasUndoLogTable
+	HasUndoLogTable(ctx context.Context, conn *sql.Conn) (bool, error)
 }
 
 // GetUndoLogManager

pkg/datasource/sql/undo_test.go

@@ -71,3 +71,28 @@ func TestDeleteUndoLogs(t *testing.T) {
 		testDeleteUndoLogs()
 	})
 }
+
+// TestHasUndoLogTable
+func TestHasUndoLogTable(t *testing.T) {
+	// local test can annotation t.SkipNow()
+	t.SkipNow()
+
+	testHasUndoLogTable := func() {
+		db, err := sql.Open(SeataMySQLDriver, "root:12345678@tcp(127.0.0.1:3306)/seata_order?multiStatements=true")
+		assert.Nil(t, err)
+
+		ctx := context.Background()
+		sqlConn, err := db.Conn(ctx)
+		assert.Nil(t, err)
+
+		undoLogManager := new(base.BaseUndoLogManager)
+
+		res, err := undoLogManager.HasUndoLogTable(ctx, sqlConn)
+		assert.Nil(t, err)
+		assert.True(t, res)
+	}
+
+	t.Run("test_has_undo_log_table", func(t *testing.T) {
+		testHasUndoLogTable()
+	})
+}