修改bug

test/v1/sdk.go

@@ -114,7 +114,23 @@ func (toolKit *ToolKit) commonQuery(req *ports.CommonQueryRequest, retInfosMap *
 	return toolKit
 }
 
-func (toolKit *ToolKit) CountWhere(req *ports.CountWhereRequest, retCount *int64) *ToolKit {
+func (toolKit *ToolKit) queryByKeys(req *ports.QueryByKeysRequest, retInfoMap *map[string]any) *ToolKit {
+	statement, infoMap, err := clientInstance.QueryByKeys(req)
+	if err != nil {
+		toolKit.t.Fatal(err)
+	}
+
+	fmt.Println(statement)
+
+	if retInfoMap != nil {
+		*retInfoMap = make(map[string]any)
+		*retInfoMap = infoMap
+	}
+
+	return toolKit
+}
+
+func (toolKit *ToolKit) countWhere(req *ports.CountWhereRequest, retCount *int64) *ToolKit {
 	statement, count, err := clientInstance.CountWhere(req)
 	if err != nil {
 		toolKit.t.Fatal(err)
@@ -129,7 +145,7 @@ func (toolKit *ToolKit) CountWhere(req *ports.CountWhereRequest, retCount *int64
 	return toolKit
 }
 
-func (toolKit *ToolKit) CommonCount(req *ports.CommonCountRequest, retCount *int64) *ToolKit {
+func (toolKit *ToolKit) commonCount(req *ports.CommonCountRequest, retCount *int64) *ToolKit {
 	statement, count, err := clientInstance.CommonCount(req)
 	if err != nil {
 		toolKit.t.Fatal(err)

test/v1/v1_test.go

@@ -39,22 +39,36 @@ func TestInsert(t *testing.T) {
 	now := time.Now().Local()
 	tableNum := rand.New(rand.NewSource(now.Unix())).Intn(10)
 
-	newToolKit(t).autoMigrate(&ports.AutoMigrateRequest{
-		DatabaseID:            "2b78141779ee432295ca371b91c5cac7",
-		TablePrefixWithSchema: tablePrefix,
-		Version:               "v1",
-		TableModelDescribe:    tableModelDescribe,
-	}).insert(&ports.InsertRequest{
-		DatabaseID:            "2b78141779ee432295ca371b91c5cac7",
-		TablePrefixWithSchema: tablePrefix,
-		Version:               "v1",
-		KeyColumns:            []string{"id"},
-		TableRow: map[string]any{
-			"id":        id,
-			"name":      name,
-			"time":      now,
-			"table_num": tableNum,
-		},
-		UserID: "test",
-	})
+	resultMap := make(map[string]any)
+
+	newToolKit(t).
+		autoMigrate(&ports.AutoMigrateRequest{
+			DatabaseID:            "2b78141779ee432295ca371b91c5cac7",
+			TablePrefixWithSchema: tablePrefix,
+			Version:               "v1",
+			TableModelDescribe:    tableModelDescribe,
+		}).
+		insert(&ports.InsertRequest{
+			DatabaseID:            "2b78141779ee432295ca371b91c5cac7",
+			TablePrefixWithSchema: tablePrefix,
+			Version:               "v1",
+			KeyColumns:            []string{"id"},
+			TableRow: map[string]any{
+				"id":        id,
+				"name":      name,
+				"time":      now,
+				"table_num": tableNum,
+			},
+			UserID: "test",
+		}).
+		queryByKeys(&ports.QueryByKeysRequest{
+			DatabaseID:            "2b78141779ee432295ca371b91c5cac7",
+			TablePrefixWithSchema: tablePrefix,
+			Version:               "v1",
+			KeyValues:             map[string]string{"id": id},
+		}, &resultMap).
+		assertEqual(id, resultMap["id"], "ID不一致").
+		assertEqual(name, resultMap["name"], "名称不一致").
+		assertEqual(now.UnixMilli(), resultMap["time"].(time.Time).Local().UnixMilli(), "时间不一致").
+		assertEqual(tableNum, resultMap["table_num"], "表数量不一致")
 }