feature:build undo log by insert target SQL

pkg/datasource/sql/types/meta.go

@@ -19,6 +19,8 @@ package types
 
 import (
 	"reflect"
+
+	"github.com/pkg/errors"
 )
 
 // ColumnMeta
@@ -117,3 +119,31 @@ func (m TableMeta) GetPrimaryKeyOnlyName() []string {
 	}
 	return keys
 }
+
+// GetPrimaryKeyType get PK database type
+func (m TableMeta) GetPrimaryKeyType() (int32, error) {
+	for _, index := range m.Indexs {
+		if index.IType == IndexTypePrimaryKey {
+			for i := range index.Columns {
+				return index.Columns[i].DatabaseType, nil
+			}
+		}
+	}
+	return 0, errors.New("get primary key type error")
+}
+
+// GetPrimaryKeyTypeStrMap get all PK type to map
+func (m TableMeta) GetPrimaryKeyTypeStrMap() (map[string]string, error) {
+	pkMap := make(map[string]string)
+	for _, index := range m.Indexs {
+		if index.IType == IndexTypePrimaryKey {
+			for i := range index.Columns {
+				pkMap[index.ColumnName] = index.Columns[i].DatabaseTypeString
+			}
+		}
+	}
+	if len(pkMap) == 0 {
+		return nil, errors.New("get primary key type error")
+	}
+	return pkMap, nil
+}

pkg/datasource/sql/types/meta_test.go

@@ -0,0 +1,110 @@
+/*
+ * 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 types
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestTableMeta_GetPrimaryKeyTypeStrMap(t *testing.T) {
+	type fields struct {
+		TableName   string
+		Columns     map[string]ColumnMeta
+		Indexs      map[string]IndexMeta
+		ColumnNames []string
+	}
+
+	tests := []struct {
+		name   string
+		fields fields
+		want   map[string]string
+	}{
+		{name: "test-1", fields: fields{TableName: "test", Indexs: map[string]IndexMeta{
+			"id": {
+				Name:       "id",
+				ColumnName: "id",
+				IType:      IndexTypePrimaryKey,
+				Columns: []ColumnMeta{
+					{
+						ColumnName:         "id",
+						DatabaseTypeString: "BIGINT",
+					},
+				},
+			},
+		}}, want: map[string]string{
+			"id": "BIGINT",
+		}},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			m := TableMeta{
+				TableName:   tt.fields.TableName,
+				Columns:     tt.fields.Columns,
+				Indexs:      tt.fields.Indexs,
+				ColumnNames: tt.fields.ColumnNames,
+			}
+			got, err := m.GetPrimaryKeyTypeStrMap()
+			assert.Nil(t, err)
+			assert.Equal(t, tt.want, got)
+		})
+	}
+}
+
+func TestTableMeta_GetPrimaryKeyType(t *testing.T) {
+	type fields struct {
+		TableName   string
+		Columns     map[string]ColumnMeta
+		Indexs      map[string]IndexMeta
+		ColumnNames []string
+	}
+	tests := []struct {
+		name   string
+		fields fields
+		want   int32
+	}{
+		{name: "test-1", fields: fields{TableName: "test", Indexs: map[string]IndexMeta{
+			"id": {
+				Name:       "id",
+				ColumnName: "id",
+				IType:      IndexTypePrimaryKey,
+				Columns: []ColumnMeta{
+					{
+						ColumnName:         "id",
+						DatabaseTypeString: "BIGINT",
+						DatabaseType:       GetSqlDataType("BIGINT"),
+					},
+				},
+			},
+		}}, want: GetSqlDataType("BIGINT")},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			m := TableMeta{
+				TableName:   tt.fields.TableName,
+				Columns:     tt.fields.Columns,
+				Indexs:      tt.fields.Indexs,
+				ColumnNames: tt.fields.ColumnNames,
+			}
+			got, err := m.GetPrimaryKeyType()
+			assert.Nil(t, err)
+			assert.Equalf(t, tt.want, got, "GetPrimaryKeyType()")
+		})
+	}
+}