feat: support to show databases and tables

console/atest-ui/src/views/DataManager.vue

@@ -46,7 +46,7 @@ const ormDataHandler = (data) => {
     const result = []
     const cols = new Set()
 
-    data.items.forEach(e => {
+    data.Rows.forEach(e => {
         const obj = {}
         e.data.forEach(item => {
             obj[item.key] = item.value
@@ -65,7 +65,7 @@ const ormDataHandler = (data) => {
 
 const keyValueDataHandler = (data) => {
     queryResult.value = []
-    data.data.forEach(e => {
+    data.Pairs.forEach(e => {
         const obj = {}
         obj['key'] = e.key
         obj['value'] = e.value

pkg/runner/monitor/monitor.pb.go

@@ -1,7 +1,7 @@
 // Code generated by protoc-gen-go. DO NOT EDIT.
 // versions:
 // 	protoc-gen-go v1.28.1
-// 	protoc        v4.25.1
+// 	protoc        v4.22.2
 // source: pkg/runner/monitor/monitor.proto
 
 package monitor

pkg/runner/monitor/monitor_grpc.pb.go

@@ -1,7 +1,7 @@
 // Code generated by protoc-gen-go-grpc. DO NOT EDIT.
 // versions:
 // - protoc-gen-go-grpc v1.2.0
-// - protoc             v4.25.1
+// - protoc             v4.22.2
 // source: pkg/runner/monitor/monitor.proto
 
 package monitor

pkg/server/server.pb.gw.go

@@ -35,7 +35,11 @@ func request_Runner_Run_0(ctx context.Context, marshaler runtime.Marshaler, clie
 	var protoReq TestTask
 	var metadata runtime.ServerMetadata
 
-	if err := marshaler.NewDecoder(req.Body).Decode(&protoReq); err != nil && err != io.EOF {
+	newReader, berr := utilities.IOReaderFactory(req.Body)
+	if berr != nil {
+		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", berr)
+	}
+	if err := marshaler.NewDecoder(newReader()).Decode(&protoReq); err != nil && err != io.EOF {
 		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", err)
 	}
 
@@ -48,7 +52,11 @@ func local_request_Runner_Run_0(ctx context.Context, marshaler runtime.Marshaler
 	var protoReq TestTask
 	var metadata runtime.ServerMetadata
 
-	if err := marshaler.NewDecoder(req.Body).Decode(&protoReq); err != nil && err != io.EOF {
+	newReader, berr := utilities.IOReaderFactory(req.Body)
+	if berr != nil {
+		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", berr)
+	}
+	if err := marshaler.NewDecoder(newReader()).Decode(&protoReq); err != nil && err != io.EOF {
 		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", err)
 	}
 
@@ -122,7 +130,11 @@ func request_Runner_CreateTestSuite_0(ctx context.Context, marshaler runtime.Mar
 	var protoReq TestSuiteIdentity
 	var metadata runtime.ServerMetadata
 
-	if err := marshaler.NewDecoder(req.Body).Decode(&protoReq); err != nil && err != io.EOF {
+	newReader, berr := utilities.IOReaderFactory(req.Body)
+	if berr != nil {
+		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", berr)
+	}
+	if err := marshaler.NewDecoder(newReader()).Decode(&protoReq); err != nil && err != io.EOF {
 		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", err)
 	}
 
@@ -135,7 +147,11 @@ func local_request_Runner_CreateTestSuite_0(ctx context.Context, marshaler runti
 	var protoReq TestSuiteIdentity
 	var metadata runtime.ServerMetadata
 
-	if err := marshaler.NewDecoder(req.Body).Decode(&protoReq); err != nil && err != io.EOF {
+	newReader, berr := utilities.IOReaderFactory(req.Body)
+	if berr != nil {
+		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", berr)
+	}
+	if err := marshaler.NewDecoder(newReader()).Decode(&protoReq); err != nil && err != io.EOF {
 		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", err)
 	}
 
@@ -148,7 +164,11 @@ func request_Runner_ImportTestSuite_0(ctx context.Context, marshaler runtime.Mar
 	var protoReq TestSuiteSource
 	var metadata runtime.ServerMetadata
 
-	if err := marshaler.NewDecoder(req.Body).Decode(&protoReq); err != nil && err != io.EOF {
+	newReader, berr := utilities.IOReaderFactory(req.Body)
+	if berr != nil {
+		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", berr)
+	}
+	if err := marshaler.NewDecoder(newReader()).Decode(&protoReq); err != nil && err != io.EOF {
 		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", err)
 	}
 
@@ -161,7 +181,11 @@ func local_request_Runner_ImportTestSuite_0(ctx context.Context, marshaler runti
 	var protoReq TestSuiteSource
 	var metadata runtime.ServerMetadata
 
-	if err := marshaler.NewDecoder(req.Body).Decode(&protoReq); err != nil && err != io.EOF {
+	newReader, berr := utilities.IOReaderFactory(req.Body)
+	if berr != nil {
+		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", berr)
+	}
+	if err := marshaler.NewDecoder(newReader()).Decode(&protoReq); err != nil && err != io.EOF {
 		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", err)
 	}
 
@@ -171,7 +195,7 @@ func local_request_Runner_ImportTestSuite_0(ctx context.Context, marshaler runti
 }
 
 var (
-	filter_Runner_GetTestSuite_0 = &utilities.DoubleArray{Encoding: map[string]int{"name": 0}, Base: []int{1, 1, 0}, Check: []int{0, 1, 2}}
+	filter_Runner_GetTestSuite_0 = &utilities.DoubleArray{Encoding: map[string]int{"name": 0}, Base: []int{1, 2, 0, 0}, Check: []int{0, 1, 2, 2}}
 )
 
 func request_Runner_GetTestSuite_0(ctx context.Context, marshaler runtime.Marshaler, client RunnerClient, req *http.Request, pathParams map[string]string) (proto.Message, runtime.ServerMetadata, error) {
@@ -244,7 +268,11 @@ func request_Runner_UpdateTestSuite_0(ctx context.Context, marshaler runtime.Mar
 	var protoReq TestSuite
 	var metadata runtime.ServerMetadata
 
-	if err := marshaler.NewDecoder(req.Body).Decode(&protoReq); err != nil && err != io.EOF {
+	newReader, berr := utilities.IOReaderFactory(req.Body)
+	if berr != nil {
+		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", berr)
+	}
+	if err := marshaler.NewDecoder(newReader()).Decode(&protoReq); err != nil && err != io.EOF {
 		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", err)
 	}
 
@@ -274,7 +302,11 @@ func local_request_Runner_UpdateTestSuite_0(ctx context.Context, marshaler runti
 	var protoReq TestSuite
 	var metadata runtime.ServerMetadata
 
-	if err := marshaler.NewDecoder(req.Body).Decode(&protoReq); err != nil && err != io.EOF {
+	newReader, berr := utilities.IOReaderFactory(req.Body)
+	if berr != nil {
+		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", berr)
+	}
+	if err := marshaler.NewDecoder(newReader()).Decode(&protoReq); err != nil && err != io.EOF {
 		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", err)
 	}
 
@@ -301,7 +333,7 @@ func local_request_Runner_UpdateTestSuite_0(ctx context.Context, marshaler runti
 }
 
 var (
-	filter_Runner_DeleteTestSuite_0 = &utilities.DoubleArray{Encoding: map[string]int{"name": 0}, Base: []int{1, 1, 0}, Check: []int{0, 1, 2}}
+	filter_Runner_DeleteTestSuite_0 = &utilities.DoubleArray{Encoding: map[string]int{"name": 0}, Base: []int{1, 2, 0, 0}, Check: []int{0, 1, 2, 2}}
 )
 
 func request_Runner_DeleteTestSuite_0(ctx context.Context, marshaler runtime.Marshaler, client RunnerClient, req *http.Request, pathParams map[string]string) (proto.Message, runtime.ServerMetadata, error) {
@@ -374,7 +406,11 @@ func request_Runner_DuplicateTestSuite_0(ctx context.Context, marshaler runtime.
 	var protoReq TestSuiteDuplicate
 	var metadata runtime.ServerMetadata
 
-	if err := marshaler.NewDecoder(req.Body).Decode(&protoReq); err != nil && err != io.EOF {
+	newReader, berr := utilities.IOReaderFactory(req.Body)
+	if berr != nil {
+		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", berr)
+	}
+	if err := marshaler.NewDecoder(newReader()).Decode(&protoReq); err != nil && err != io.EOF {
 		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", err)
 	}
 
@@ -404,7 +440,11 @@ func local_request_Runner_DuplicateTestSuite_0(ctx context.Context, marshaler ru
 	var protoReq TestSuiteDuplicate
 	var metadata runtime.ServerMetadata
 
-	if err := marshaler.NewDecoder(req.Body).Decode(&protoReq); err != nil && err != io.EOF {
+	newReader, berr := utilities.IOReaderFactory(req.Body)
+	if berr != nil {
+		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", berr)
+	}
+	if err := marshaler.NewDecoder(newReader()).Decode(&protoReq); err != nil && err != io.EOF {
 		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", err)
 	}
 
@@ -434,7 +474,11 @@ func request_Runner_RenameTestSuite_0(ctx context.Context, marshaler runtime.Mar
 	var protoReq TestSuiteDuplicate
 	var metadata runtime.ServerMetadata
 
-	if err := marshaler.NewDecoder(req.Body).Decode(&protoReq); err != nil && err != io.EOF {
+	newReader, berr := utilities.IOReaderFactory(req.Body)
+	if berr != nil {
+		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", berr)
+	}
+	if err := marshaler.NewDecoder(newReader()).Decode(&protoReq); err != nil && err != io.EOF {
 		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", err)
 	}
 
@@ -464,7 +508,11 @@ func local_request_Runner_RenameTestSuite_0(ctx context.Context, marshaler runti
 	var protoReq TestSuiteDuplicate
 	var metadata runtime.ServerMetadata
 
-	if err := marshaler.NewDecoder(req.Body).Decode(&protoReq); err != nil && err != io.EOF {
+	newReader, berr := utilities.IOReaderFactory(req.Body)
+	if berr != nil {
+		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", berr)
+	}
+	if err := marshaler.NewDecoder(newReader()).Decode(&protoReq); err != nil && err != io.EOF {
 		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", err)
 	}
 
@@ -491,7 +539,7 @@ func local_request_Runner_RenameTestSuite_0(ctx context.Context, marshaler runti
 }
 
 var (
-	filter_Runner_GetTestSuiteYaml_0 = &utilities.DoubleArray{Encoding: map[string]int{"name": 0}, Base: []int{1, 1, 0}, Check: []int{0, 1, 2}}
+	filter_Runner_GetTestSuiteYaml_0 = &utilities.DoubleArray{Encoding: map[string]int{"name": 0}, Base: []int{1, 2, 0, 0}, Check: []int{0, 1, 2, 2}}
 )
 
 func request_Runner_GetTestSuiteYaml_0(ctx context.Context, marshaler runtime.Marshaler, client RunnerClient, req *http.Request, pathParams map[string]string) (proto.Message, runtime.ServerMetadata, error) {
@@ -561,7 +609,7 @@ func local_request_Runner_GetTestSuiteYaml_0(ctx context.Context, marshaler runt
 }
 
 var (
-	filter_Runner_ListTestCase_0 = &utilities.DoubleArray{Encoding: map[string]int{"name": 0}, Base: []int{1, 1, 0}, Check: []int{0, 1, 2}}
+	filter_Runner_ListTestCase_0 = &utilities.DoubleArray{Encoding: map[string]int{"name": 0}, Base: []int{1, 2, 0, 0}, Check: []int{0, 1, 2, 2}}
 )
 
 func request_Runner_ListTestCase_0(ctx context.Context, marshaler runtime.Marshaler, client RunnerClient, req *http.Request, pathParams map[string]string) (proto.Message, runtime.ServerMetadata, error) {
@@ -634,7 +682,11 @@ func request_Runner_RunTestCase_0(ctx context.Context, marshaler runtime.Marshal
 	var protoReq TestCaseIdentity
 	var metadata runtime.ServerMetadata
 
-	if err := marshaler.NewDecoder(req.Body).Decode(&protoReq); err != nil && err != io.EOF {
+	newReader, berr := utilities.IOReaderFactory(req.Body)
+	if berr != nil {
+		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", berr)
+	}
+	if err := marshaler.NewDecoder(newReader()).Decode(&protoReq); err != nil && err != io.EOF {
 		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", err)
 	}
 
@@ -674,7 +726,11 @@ func local_request_Runner_RunTestCase_0(ctx context.Context, marshaler runtime.M
 	var protoReq TestCaseIdentity
 	var metadata runtime.ServerMetadata
 
-	if err := marshaler.NewDecoder(req.Body).Decode(&protoReq); err != nil && err != io.EOF {
+	newReader, berr := utilities.IOReaderFactory(req.Body)
+	if berr != nil {
+		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", berr)
+	}
+	if err := marshaler.NewDecoder(newReader()).Decode(&protoReq); err != nil && err != io.EOF {
 		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", err)
 	}
 
@@ -754,7 +810,7 @@ func request_Runner_BatchRun_0(ctx context.Context, marshaler runtime.Marshaler,
 }
 
 var (
-	filter_Runner_GetTestCase_0 = &utilities.DoubleArray{Encoding: map[string]int{"suite": 0, "testcase": 1}, Base: []int{1, 1, 2, 0, 0}, Check: []int{0, 1, 1, 2, 3}}
+	filter_Runner_GetTestCase_0 = &utilities.DoubleArray{Encoding: map[string]int{"suite": 0, "testcase": 1}, Base: []int{1, 2, 4, 0, 0, 0, 0}, Check: []int{0, 1, 1, 2, 2, 3, 3}}
 )
 
 func request_Runner_GetTestCase_0(ctx context.Context, marshaler runtime.Marshaler, client RunnerClient, req *http.Request, pathParams map[string]string) (proto.Message, runtime.ServerMetadata, error) {
@@ -847,7 +903,11 @@ func request_Runner_CreateTestCase_0(ctx context.Context, marshaler runtime.Mars
 	var protoReq TestCaseWithSuite
 	var metadata runtime.ServerMetadata
 
-	if err := marshaler.NewDecoder(req.Body).Decode(&protoReq); err != nil && err != io.EOF {
+	newReader, berr := utilities.IOReaderFactory(req.Body)
+	if berr != nil {
+		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", berr)
+	}
+	if err := marshaler.NewDecoder(newReader()).Decode(&protoReq); err != nil && err != io.EOF {
 		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", err)
 	}
 
@@ -877,7 +937,11 @@ func local_request_Runner_CreateTestCase_0(ctx context.Context, marshaler runtim
 	var protoReq TestCaseWithSuite
 	var metadata runtime.ServerMetadata
 
-	if err := marshaler.NewDecoder(req.Body).Decode(&protoReq); err != nil && err != io.EOF {
+	newReader, berr := utilities.IOReaderFactory(req.Body)
+	if berr != nil {
+		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", berr)
+	}
+	if err := marshaler.NewDecoder(newReader()).Decode(&protoReq); err != nil && err != io.EOF {
 		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", err)
 	}
 
@@ -907,7 +971,11 @@ func request_Runner_UpdateTestCase_0(ctx context.Context, marshaler runtime.Mars
 	var protoReq TestCaseWithSuite
 	var metadata runtime.ServerMetadata
 
-	if err := marshaler.NewDecoder(req.Body).Decode(&protoReq); err != nil && err != io.EOF {
+	newReader, berr := utilities.IOReaderFactory(req.Body)
+	if berr != nil {
+		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", berr)
+	}
+	if err := marshaler.NewDecoder(newReader()).Decode(&protoReq); err != nil && err != io.EOF {
 		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", err)
 	}
 
@@ -947,7 +1015,11 @@ func local_request_Runner_UpdateTestCase_0(ctx context.Context, marshaler runtim
 	var protoReq TestCaseWithSuite
 	var metadata runtime.ServerMetadata
 
-	if err := marshaler.NewDecoder(req.Body).Decode(&protoReq); err != nil && err != io.EOF {
+	newReader, berr := utilities.IOReaderFactory(req.Body)
+	if berr != nil {
+		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", berr)
+	}
+	if err := marshaler.NewDecoder(newReader()).Decode(&protoReq); err != nil && err != io.EOF {
 		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", err)
 	}
 
@@ -984,7 +1056,7 @@ func local_request_Runner_UpdateTestCase_0(ctx context.Context, marshaler runtim
 }
 
 var (
-	filter_Runner_DeleteTestCase_0 = &utilities.DoubleArray{Encoding: map[string]int{"suite": 0, "testcase": 1}, Base: []int{1, 1, 2, 0, 0}, Check: []int{0, 1, 1, 2, 3}}
+	filter_Runner_DeleteTestCase_0 = &utilities.DoubleArray{Encoding: map[string]int{"suite": 0, "testcase": 1}, Base: []int{1, 2, 4, 0, 0, 0, 0}, Check: []int{0, 1, 1, 2, 2, 3, 3}}
 )
 
 func request_Runner_DeleteTestCase_0(ctx context.Context, marshaler runtime.Marshaler, client RunnerClient, req *http.Request, pathParams map[string]string) (proto.Message, runtime.ServerMetadata, error) {
@@ -1077,7 +1149,11 @@ func request_Runner_DuplicateTestCase_0(ctx context.Context, marshaler runtime.M
 	var protoReq TestCaseDuplicate
 	var metadata runtime.ServerMetadata
 
-	if err := marshaler.NewDecoder(req.Body).Decode(&protoReq); err != nil && err != io.EOF {
+	newReader, berr := utilities.IOReaderFactory(req.Body)
+	if berr != nil {
+		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", berr)
+	}
+	if err := marshaler.NewDecoder(newReader()).Decode(&protoReq); err != nil && err != io.EOF {
 		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", err)
 	}
 
@@ -1117,7 +1193,11 @@ func local_request_Runner_DuplicateTestCase_0(ctx context.Context, marshaler run
 	var protoReq TestCaseDuplicate
 	var metadata runtime.ServerMetadata
 
-	if err := marshaler.NewDecoder(req.Body).Decode(&protoReq); err != nil && err != io.EOF {
+	newReader, berr := utilities.IOReaderFactory(req.Body)
+	if berr != nil {
+		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", berr)
+	}
+	if err := marshaler.NewDecoder(newReader()).Decode(&protoReq); err != nil && err != io.EOF {
 		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", err)
 	}
 
@@ -1157,7 +1237,11 @@ func request_Runner_RenameTestCase_0(ctx context.Context, marshaler runtime.Mars
 	var protoReq TestCaseDuplicate
 	var metadata runtime.ServerMetadata
 
-	if err := marshaler.NewDecoder(req.Body).Decode(&protoReq); err != nil && err != io.EOF {
+	newReader, berr := utilities.IOReaderFactory(req.Body)
+	if berr != nil {
+		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", berr)
+	}
+	if err := marshaler.NewDecoder(newReader()).Decode(&protoReq); err != nil && err != io.EOF {
 		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", err)
 	}
 
@@ -1197,7 +1281,11 @@ func local_request_Runner_RenameTestCase_0(ctx context.Context, marshaler runtim
 	var protoReq TestCaseDuplicate
 	var metadata runtime.ServerMetadata
 
-	if err := marshaler.NewDecoder(req.Body).Decode(&protoReq); err != nil && err != io.EOF {
+	newReader, berr := utilities.IOReaderFactory(req.Body)
+	if berr != nil {
+		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", berr)
+	}
+	if err := marshaler.NewDecoder(newReader()).Decode(&protoReq); err != nil && err != io.EOF {
 		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", err)
 	}
 
@@ -1288,7 +1376,7 @@ func local_request_Runner_GetHistorySuites_0(ctx context.Context, marshaler runt
 }
 
 var (
-	filter_Runner_GetHistoryTestCaseWithResult_0 = &utilities.DoubleArray{Encoding: map[string]int{"ID": 0}, Base: []int{1, 1, 0}, Check: []int{0, 1, 2}}
+	filter_Runner_GetHistoryTestCaseWithResult_0 = &utilities.DoubleArray{Encoding: map[string]int{"ID": 0}, Base: []int{1, 2, 0, 0}, Check: []int{0, 1, 2, 2}}
 )
 
 func request_Runner_GetHistoryTestCaseWithResult_0(ctx context.Context, marshaler runtime.Marshaler, client RunnerClient, req *http.Request, pathParams map[string]string) (proto.Message, runtime.ServerMetadata, error) {
@@ -1358,7 +1446,7 @@ func local_request_Runner_GetHistoryTestCaseWithResult_0(ctx context.Context, ma
 }
 
 var (
-	filter_Runner_GetHistoryTestCase_0 = &utilities.DoubleArray{Encoding: map[string]int{"ID": 0}, Base: []int{1, 1, 0}, Check: []int{0, 1, 2}}
+	filter_Runner_GetHistoryTestCase_0 = &utilities.DoubleArray{Encoding: map[string]int{"ID": 0}, Base: []int{1, 2, 0, 0}, Check: []int{0, 1, 2, 2}}
 )
 
 func request_Runner_GetHistoryTestCase_0(ctx context.Context, marshaler runtime.Marshaler, client RunnerClient, req *http.Request, pathParams map[string]string) (proto.Message, runtime.ServerMetadata, error) {
@@ -1428,7 +1516,7 @@ func local_request_Runner_GetHistoryTestCase_0(ctx context.Context, marshaler ru
 }
 
 var (
-	filter_Runner_DeleteHistoryTestCase_0 = &utilities.DoubleArray{Encoding: map[string]int{"ID": 0}, Base: []int{1, 1, 0}, Check: []int{0, 1, 2}}
+	filter_Runner_DeleteHistoryTestCase_0 = &utilities.DoubleArray{Encoding: map[string]int{"ID": 0}, Base: []int{1, 2, 0, 0}, Check: []int{0, 1, 2, 2}}
 )
 
 func request_Runner_DeleteHistoryTestCase_0(ctx context.Context, marshaler runtime.Marshaler, client RunnerClient, req *http.Request, pathParams map[string]string) (proto.Message, runtime.ServerMetadata, error) {
@@ -1498,7 +1586,7 @@ func local_request_Runner_DeleteHistoryTestCase_0(ctx context.Context, marshaler
 }
 
 var (
-	filter_Runner_DeleteAllHistoryTestCase_0 = &utilities.DoubleArray{Encoding: map[string]int{"suiteName": 0, "caseName": 1}, Base: []int{1, 1, 2, 0, 0}, Check: []int{0, 1, 1, 2, 3}}
+	filter_Runner_DeleteAllHistoryTestCase_0 = &utilities.DoubleArray{Encoding: map[string]int{"suiteName": 0, "caseName": 1}, Base: []int{1, 2, 4, 0, 0, 0, 0}, Check: []int{0, 1, 1, 2, 2, 3, 3}}
 )
 
 func request_Runner_DeleteAllHistoryTestCase_0(ctx context.Context, marshaler runtime.Marshaler, client RunnerClient, req *http.Request, pathParams map[string]string) (proto.Message, runtime.ServerMetadata, error) {
@@ -1588,7 +1676,7 @@ func local_request_Runner_DeleteAllHistoryTestCase_0(ctx context.Context, marsha
 }
 
 var (
-	filter_Runner_GetTestCaseAllHistory_0 = &utilities.DoubleArray{Encoding: map[string]int{"suiteName": 0, "name": 1}, Base: []int{1, 1, 2, 0, 0}, Check: []int{0, 1, 1, 2, 3}}
+	filter_Runner_GetTestCaseAllHistory_0 = &utilities.DoubleArray{Encoding: map[string]int{"suiteName": 0, "name": 1}, Base: []int{1, 2, 4, 0, 0, 0, 0}, Check: []int{0, 1, 1, 2, 2, 3, 3}}
 )
 
 func request_Runner_GetTestCaseAllHistory_0(ctx context.Context, marshaler runtime.Marshaler, client RunnerClient, req *http.Request, pathParams map[string]string) (proto.Message, runtime.ServerMetadata, error) {
@@ -1699,7 +1787,11 @@ func request_Runner_GenerateCode_0(ctx context.Context, marshaler runtime.Marsha
 	var protoReq CodeGenerateRequest
 	var metadata runtime.ServerMetadata
 
-	if err := marshaler.NewDecoder(req.Body).Decode(&protoReq); err != nil && err != io.EOF {
+	newReader, berr := utilities.IOReaderFactory(req.Body)
+	if berr != nil {
+		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", berr)
+	}
+	if err := marshaler.NewDecoder(newReader()).Decode(&protoReq); err != nil && err != io.EOF {
 		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", err)
 	}
 
@@ -1712,7 +1804,11 @@ func local_request_Runner_GenerateCode_0(ctx context.Context, marshaler runtime.
 	var protoReq CodeGenerateRequest
 	var metadata runtime.ServerMetadata
 
-	if err := marshaler.NewDecoder(req.Body).Decode(&protoReq); err != nil && err != io.EOF {
+	newReader, berr := utilities.IOReaderFactory(req.Body)
+	if berr != nil {
+		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", berr)
+	}
+	if err := marshaler.NewDecoder(newReader()).Decode(&protoReq); err != nil && err != io.EOF {
 		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", err)
 	}
 
@@ -1725,7 +1821,11 @@ func request_Runner_HistoryGenerateCode_0(ctx context.Context, marshaler runtime
 	var protoReq CodeGenerateRequest
 	var metadata runtime.ServerMetadata
 
-	if err := marshaler.NewDecoder(req.Body).Decode(&protoReq); err != nil && err != io.EOF {
+	newReader, berr := utilities.IOReaderFactory(req.Body)
+	if berr != nil {
+		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", berr)
+	}
+	if err := marshaler.NewDecoder(newReader()).Decode(&protoReq); err != nil && err != io.EOF {
 		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", err)
 	}
 
@@ -1738,7 +1838,11 @@ func local_request_Runner_HistoryGenerateCode_0(ctx context.Context, marshaler r
 	var protoReq CodeGenerateRequest
 	var metadata runtime.ServerMetadata
 
-	if err := marshaler.NewDecoder(req.Body).Decode(&protoReq); err != nil && err != io.EOF {
+	newReader, berr := utilities.IOReaderFactory(req.Body)
+	if berr != nil {
+		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", berr)
+	}
+	if err := marshaler.NewDecoder(newReader()).Decode(&protoReq); err != nil && err != io.EOF {
 		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", err)
 	}
 
@@ -1769,7 +1873,11 @@ func request_Runner_ConvertTestSuite_0(ctx context.Context, marshaler runtime.Ma
 	var protoReq CodeGenerateRequest
 	var metadata runtime.ServerMetadata
 
-	if err := marshaler.NewDecoder(req.Body).Decode(&protoReq); err != nil && err != io.EOF {
+	newReader, berr := utilities.IOReaderFactory(req.Body)
+	if berr != nil {
+		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", berr)
+	}
+	if err := marshaler.NewDecoder(newReader()).Decode(&protoReq); err != nil && err != io.EOF {
 		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", err)
 	}
 
@@ -1782,7 +1890,11 @@ func local_request_Runner_ConvertTestSuite_0(ctx context.Context, marshaler runt
 	var protoReq CodeGenerateRequest
 	var metadata runtime.ServerMetadata
 
-	if err := marshaler.NewDecoder(req.Body).Decode(&protoReq); err != nil && err != io.EOF {
+	newReader, berr := utilities.IOReaderFactory(req.Body)
+	if berr != nil {
+		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", berr)
+	}
+	if err := marshaler.NewDecoder(newReader()).Decode(&protoReq); err != nil && err != io.EOF {
 		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", err)
 	}
 
@@ -1928,7 +2040,11 @@ func request_Runner_DownloadResponseFile_0(ctx context.Context, marshaler runtim
 	var protoReq TestCase
 	var metadata runtime.ServerMetadata
 
-	if err := marshaler.NewDecoder(req.Body).Decode(&protoReq); err != nil && err != io.EOF {
+	newReader, berr := utilities.IOReaderFactory(req.Body)
+	if berr != nil {
+		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", berr)
+	}
+	if err := marshaler.NewDecoder(newReader()).Decode(&protoReq); err != nil && err != io.EOF {
 		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", err)
 	}
 
@@ -1958,7 +2074,11 @@ func local_request_Runner_DownloadResponseFile_0(ctx context.Context, marshaler
 	var protoReq TestCase
 	var metadata runtime.ServerMetadata
 
-	if err := marshaler.NewDecoder(req.Body).Decode(&protoReq); err != nil && err != io.EOF {
+	newReader, berr := utilities.IOReaderFactory(req.Body)
+	if berr != nil {
+		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", berr)
+	}
+	if err := marshaler.NewDecoder(newReader()).Decode(&protoReq); err != nil && err != io.EOF {
 		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", err)
 	}
 
@@ -2024,7 +2144,11 @@ func request_Runner_CreateStore_0(ctx context.Context, marshaler runtime.Marshal
 	var protoReq Store
 	var metadata runtime.ServerMetadata
 
-	if err := marshaler.NewDecoder(req.Body).Decode(&protoReq); err != nil && err != io.EOF {
+	newReader, berr := utilities.IOReaderFactory(req.Body)
+	if berr != nil {
+		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", berr)
+	}
+	if err := marshaler.NewDecoder(newReader()).Decode(&protoReq); err != nil && err != io.EOF {
 		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", err)
 	}
 
@@ -2037,7 +2161,11 @@ func local_request_Runner_CreateStore_0(ctx context.Context, marshaler runtime.M
 	var protoReq Store
 	var metadata runtime.ServerMetadata
 
-	if err := marshaler.NewDecoder(req.Body).Decode(&protoReq); err != nil && err != io.EOF {
+	newReader, berr := utilities.IOReaderFactory(req.Body)
+	if berr != nil {
+		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", berr)
+	}
+	if err := marshaler.NewDecoder(newReader()).Decode(&protoReq); err != nil && err != io.EOF {
 		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", err)
 	}
 
@@ -2050,7 +2178,11 @@ func request_Runner_UpdateStore_0(ctx context.Context, marshaler runtime.Marshal
 	var protoReq Store
 	var metadata runtime.ServerMetadata
 
-	if err := marshaler.NewDecoder(req.Body).Decode(&protoReq); err != nil && err != io.EOF {
+	newReader, berr := utilities.IOReaderFactory(req.Body)
+	if berr != nil {
+		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", berr)
+	}
+	if err := marshaler.NewDecoder(newReader()).Decode(&protoReq); err != nil && err != io.EOF {
 		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", err)
 	}
 
@@ -2080,7 +2212,11 @@ func local_request_Runner_UpdateStore_0(ctx context.Context, marshaler runtime.M
 	var protoReq Store
 	var metadata runtime.ServerMetadata
 
-	if err := marshaler.NewDecoder(req.Body).Decode(&protoReq); err != nil && err != io.EOF {
+	newReader, berr := utilities.IOReaderFactory(req.Body)
+	if berr != nil {
+		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", berr)
+	}
+	if err := marshaler.NewDecoder(newReader()).Decode(&protoReq); err != nil && err != io.EOF {
 		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", err)
 	}
 
@@ -2107,7 +2243,7 @@ func local_request_Runner_UpdateStore_0(ctx context.Context, marshaler runtime.M
 }
 
 var (
-	filter_Runner_DeleteStore_0 = &utilities.DoubleArray{Encoding: map[string]int{"name": 0}, Base: []int{1, 1, 0}, Check: []int{0, 1, 2}}
+	filter_Runner_DeleteStore_0 = &utilities.DoubleArray{Encoding: map[string]int{"name": 0}, Base: []int{1, 2, 0, 0}, Check: []int{0, 1, 2, 2}}
 )
 
 func request_Runner_DeleteStore_0(ctx context.Context, marshaler runtime.Marshaler, client RunnerClient, req *http.Request, pathParams map[string]string) (proto.Message, runtime.ServerMetadata, error) {
@@ -2180,7 +2316,11 @@ func request_Runner_VerifyStore_0(ctx context.Context, marshaler runtime.Marshal
 	var protoReq SimpleQuery
 	var metadata runtime.ServerMetadata
 
-	if err := marshaler.NewDecoder(req.Body).Decode(&protoReq); err != nil && err != io.EOF {
+	newReader, berr := utilities.IOReaderFactory(req.Body)
+	if berr != nil {
+		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", berr)
+	}
+	if err := marshaler.NewDecoder(newReader()).Decode(&protoReq); err != nil && err != io.EOF {
 		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", err)
 	}
 
@@ -2193,7 +2333,11 @@ func local_request_Runner_VerifyStore_0(ctx context.Context, marshaler runtime.M
 	var protoReq SimpleQuery
 	var metadata runtime.ServerMetadata
 
-	if err := marshaler.NewDecoder(req.Body).Decode(&protoReq); err != nil && err != io.EOF {
+	newReader, berr := utilities.IOReaderFactory(req.Body)
+	if berr != nil {
+		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", berr)
+	}
+	if err := marshaler.NewDecoder(newReader()).Decode(&protoReq); err != nil && err != io.EOF {
 		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", err)
 	}
 
@@ -2224,7 +2368,11 @@ func request_Runner_CreateSecret_0(ctx context.Context, marshaler runtime.Marsha
 	var protoReq Secret
 	var metadata runtime.ServerMetadata
 
-	if err := marshaler.NewDecoder(req.Body).Decode(&protoReq); err != nil && err != io.EOF {
+	newReader, berr := utilities.IOReaderFactory(req.Body)
+	if berr != nil {
+		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", berr)
+	}
+	if err := marshaler.NewDecoder(newReader()).Decode(&protoReq); err != nil && err != io.EOF {
 		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", err)
 	}
 
@@ -2237,7 +2385,11 @@ func local_request_Runner_CreateSecret_0(ctx context.Context, marshaler runtime.
 	var protoReq Secret
 	var metadata runtime.ServerMetadata
 
-	if err := marshaler.NewDecoder(req.Body).Decode(&protoReq); err != nil && err != io.EOF {
+	newReader, berr := utilities.IOReaderFactory(req.Body)
+	if berr != nil {
+		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", berr)
+	}
+	if err := marshaler.NewDecoder(newReader()).Decode(&protoReq); err != nil && err != io.EOF {
 		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", err)
 	}
 
@@ -2247,7 +2399,7 @@ func local_request_Runner_CreateSecret_0(ctx context.Context, marshaler runtime.
 }
 
 var (
-	filter_Runner_DeleteSecret_0 = &utilities.DoubleArray{Encoding: map[string]int{"Name": 0}, Base: []int{1, 1, 0}, Check: []int{0, 1, 2}}
+	filter_Runner_DeleteSecret_0 = &utilities.DoubleArray{Encoding: map[string]int{"Name": 0}, Base: []int{1, 2, 0, 0}, Check: []int{0, 1, 2, 2}}
 )
 
 func request_Runner_DeleteSecret_0(ctx context.Context, marshaler runtime.Marshaler, client RunnerClient, req *http.Request, pathParams map[string]string) (proto.Message, runtime.ServerMetadata, error) {
@@ -2320,7 +2472,11 @@ func request_Runner_UpdateSecret_0(ctx context.Context, marshaler runtime.Marsha
 	var protoReq Secret
 	var metadata runtime.ServerMetadata
 
-	if err := marshaler.NewDecoder(req.Body).Decode(&protoReq); err != nil && err != io.EOF {
+	newReader, berr := utilities.IOReaderFactory(req.Body)
+	if berr != nil {
+		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", berr)
+	}
+	if err := marshaler.NewDecoder(newReader()).Decode(&protoReq); err != nil && err != io.EOF {
 		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", err)
 	}
 
@@ -2350,7 +2506,11 @@ func local_request_Runner_UpdateSecret_0(ctx context.Context, marshaler runtime.
 	var protoReq Secret
 	var metadata runtime.ServerMetadata
 
-	if err := marshaler.NewDecoder(req.Body).Decode(&protoReq); err != nil && err != io.EOF {
+	newReader, berr := utilities.IOReaderFactory(req.Body)
+	if berr != nil {
+		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", berr)
+	}
+	if err := marshaler.NewDecoder(newReader()).Decode(&protoReq); err != nil && err != io.EOF {
 		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", err)
 	}
 
@@ -2380,7 +2540,11 @@ func request_Runner_PProf_0(ctx context.Context, marshaler runtime.Marshaler, cl
 	var protoReq PProfRequest
 	var metadata runtime.ServerMetadata
 
-	if err := marshaler.NewDecoder(req.Body).Decode(&protoReq); err != nil && err != io.EOF {
+	newReader, berr := utilities.IOReaderFactory(req.Body)
+	if berr != nil {
+		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", berr)
+	}
+	if err := marshaler.NewDecoder(newReader()).Decode(&protoReq); err != nil && err != io.EOF {
 		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", err)
 	}
 
@@ -2393,7 +2557,11 @@ func local_request_Runner_PProf_0(ctx context.Context, marshaler runtime.Marshal
 	var protoReq PProfRequest
 	var metadata runtime.ServerMetadata
 
-	if err := marshaler.NewDecoder(req.Body).Decode(&protoReq); err != nil && err != io.EOF {
+	newReader, berr := utilities.IOReaderFactory(req.Body)
+	if berr != nil {
+		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", berr)
+	}
+	if err := marshaler.NewDecoder(newReader()).Decode(&protoReq); err != nil && err != io.EOF {
 		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", err)
 	}
 
@@ -2406,7 +2574,11 @@ func request_RunnerExtension_Run_0(ctx context.Context, marshaler runtime.Marsha
 	var protoReq TestSuiteWithCase
 	var metadata runtime.ServerMetadata
 
-	if err := marshaler.NewDecoder(req.Body).Decode(&protoReq); err != nil && err != io.EOF {
+	newReader, berr := utilities.IOReaderFactory(req.Body)
+	if berr != nil {
+		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", berr)
+	}
+	if err := marshaler.NewDecoder(newReader()).Decode(&protoReq); err != nil && err != io.EOF {
 		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", err)
 	}
 
@@ -2419,7 +2591,11 @@ func local_request_RunnerExtension_Run_0(ctx context.Context, marshaler runtime.
 	var protoReq TestSuiteWithCase
 	var metadata runtime.ServerMetadata
 
-	if err := marshaler.NewDecoder(req.Body).Decode(&protoReq); err != nil && err != io.EOF {
+	newReader, berr := utilities.IOReaderFactory(req.Body)
+	if berr != nil {
+		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", berr)
+	}
+	if err := marshaler.NewDecoder(newReader()).Decode(&protoReq); err != nil && err != io.EOF {
 		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", err)
 	}
 
@@ -2432,7 +2608,11 @@ func request_Mock_Reload_0(ctx context.Context, marshaler runtime.Marshaler, cli
 	var protoReq MockConfig
 	var metadata runtime.ServerMetadata
 
-	if err := marshaler.NewDecoder(req.Body).Decode(&protoReq); err != nil && err != io.EOF {
+	newReader, berr := utilities.IOReaderFactory(req.Body)
+	if berr != nil {
+		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", berr)
+	}
+	if err := marshaler.NewDecoder(newReader()).Decode(&protoReq); err != nil && err != io.EOF {
 		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", err)
 	}
 
@@ -2445,7 +2625,11 @@ func local_request_Mock_Reload_0(ctx context.Context, marshaler runtime.Marshale
 	var protoReq MockConfig
 	var metadata runtime.ServerMetadata
 
-	if err := marshaler.NewDecoder(req.Body).Decode(&protoReq); err != nil && err != io.EOF {
+	newReader, berr := utilities.IOReaderFactory(req.Body)
+	if berr != nil {
+		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", berr)
+	}
+	if err := marshaler.NewDecoder(newReader()).Decode(&protoReq); err != nil && err != io.EOF {
 		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", err)
 	}
 
@@ -2476,7 +2660,11 @@ func request_DataServer_Query_0(ctx context.Context, marshaler runtime.Marshaler
 	var protoReq DataQuery
 	var metadata runtime.ServerMetadata
 
-	if err := marshaler.NewDecoder(req.Body).Decode(&protoReq); err != nil && err != io.EOF {
+	newReader, berr := utilities.IOReaderFactory(req.Body)
+	if berr != nil {
+		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", berr)
+	}
+	if err := marshaler.NewDecoder(newReader()).Decode(&protoReq); err != nil && err != io.EOF {
 		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", err)
 	}
 
@@ -2489,7 +2677,11 @@ func local_request_DataServer_Query_0(ctx context.Context, marshaler runtime.Mar
 	var protoReq DataQuery
 	var metadata runtime.ServerMetadata
 
-	if err := marshaler.NewDecoder(req.Body).Decode(&protoReq); err != nil && err != io.EOF {
+	newReader, berr := utilities.IOReaderFactory(req.Body)
+	if berr != nil {
+		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", berr)
+	}
+	if err := marshaler.NewDecoder(newReader()).Decode(&protoReq); err != nil && err != io.EOF {
 		return nil, metadata, status.Errorf(codes.InvalidArgument, "%v", err)
 	}
 

pkg/server/server.proto

@@ -663,4 +663,11 @@ message DataQuery {
 message DataQueryResult {
     repeated Pair data = 1;
     repeated Pairs items = 2;
+    DataMeta meta = 3;
+}
+
+message DataMeta {
+    repeated string databases = 1;
+    repeated string tables = 2;
+    string currentDatabase = 3;
 }

pkg/server/server_grpc.pb.go

@@ -1,7 +1,7 @@
 // Code generated by protoc-gen-go-grpc. DO NOT EDIT.
 // versions:
 // - protoc-gen-go-grpc v1.2.0
-// - protoc             v4.25.1
+// - protoc             v4.22.2
 // source: pkg/server/server.proto
 
 package server

pkg/testing/loader.go

@@ -27,7 +27,15 @@ type Loader interface {
 	Verify() (readOnly bool, err error)
 	PProf(name string) []byte
 
-	Query(query map[string]string) (result map[string]string, items []map[string]string, err error)
+	Query(query map[string]string) (result DataResult, err error)
+}
+
+type DataResult struct {
+	Pairs           map[string]string
+	Rows            []map[string]string
+	Databases       []string
+	Tables          []string
+	CurrentDatabase string
 }
 
 type Writer interface {

pkg/testing/loader_file.go

@@ -515,8 +515,8 @@ func (l *fileLoader) Close() {
 	// not support
 }
 
-func (l *fileLoader) Query(query map[string]string) (result map[string]string, items []map[string]string, err error) {
-	result = map[string]string{
+func (l *fileLoader) Query(query map[string]string) (result DataResult, err error) {
+	result.Pairs = map[string]string{
 		"message": "not support",
 	}
 	return

pkg/testing/loader_non.go

@@ -150,6 +150,6 @@ func (l *nonLoader) Close() {
 	// not support
 }
 
-func (l *nonLoader) Query(query map[string]string) (result map[string]string, items []map[string]string, err error) {
+func (l *nonLoader) Query(query map[string]string) (result DataResult, err error) {
 	return
 }

pkg/testing/remote/grpc_store.go

@@ -17,311 +17,314 @@ limitations under the License.
 package remote
 
 import (
-    context "context"
-    "errors"
-    "time"
+	context "context"
+	"errors"
+	"time"
 
-    "github.com/linuxsuren/api-testing/pkg/logging"
-    server "github.com/linuxsuren/api-testing/pkg/server"
-    "github.com/linuxsuren/api-testing/pkg/testing"
+	"github.com/linuxsuren/api-testing/pkg/logging"
+	server "github.com/linuxsuren/api-testing/pkg/server"
+	"github.com/linuxsuren/api-testing/pkg/testing"
 
-    "google.golang.org/grpc"
+	"google.golang.org/grpc"
 )
 
 var (
-    grpcLogger = logging.DefaultLogger(logging.LogLevelInfo).WithName("grpc")
+	grpcLogger = logging.DefaultLogger(logging.LogLevelInfo).WithName("grpc")
 )
 
 type gRPCLoader struct {
-    store  *testing.Store
-    client LoaderClient
-    ctx    context.Context
-    conn   *grpc.ClientConn
+	store  *testing.Store
+	client LoaderClient
+	ctx    context.Context
+	conn   *grpc.ClientConn
 }
 
 func NewGRPCloaderFromStore() testing.StoreWriterFactory {
-    return &gRPCLoader{}
+	return &gRPCLoader{}
 }
 
 func (g *gRPCLoader) NewInstance(store testing.Store) (writer testing.Writer, err error) {
-    address := store.Kind.URL
+	address := store.Kind.URL
 
-    var conn *grpc.ClientConn
-    if conn, err = grpc.Dial(address, grpc.WithInsecure()); err == nil {
-        writer = &gRPCLoader{
-            store:  &store,
-            ctx:    WithStoreContext(context.Background(), &store),
-            client: NewLoaderClient(conn),
-            conn:   conn,
-        }
-    }
-    return
+	var conn *grpc.ClientConn
+	if conn, err = grpc.Dial(address, grpc.WithInsecure()); err == nil {
+		writer = &gRPCLoader{
+			store:  &store,
+			ctx:    WithStoreContext(context.Background(), &store),
+			client: NewLoaderClient(conn),
+			conn:   conn,
+		}
+	}
+	return
 }
 
 func (g *gRPCLoader) HasMore() bool {
-    // nothing to do
-    return false
+	// nothing to do
+	return false
 }
 
 func (g *gRPCLoader) Load() ([]byte, error) {
-    // nothing to do
-    return nil, nil
+	// nothing to do
+	return nil, nil
 }
 
 func (g *gRPCLoader) Put(path string) error {
-    // nothing to do
-    return nil
+	// nothing to do
+	return nil
 }
 
 func (g *gRPCLoader) GetContext() string {
-    // nothing to do
-    return ""
+	// nothing to do
+	return ""
 }
 
 func (g *gRPCLoader) GetCount() int {
-    // nothing to do
-    return 0
+	// nothing to do
+	return 0
 }
 
 func (g *gRPCLoader) Reset() {
-    // nothing to do
+	// nothing to do
 }
 
 func (g *gRPCLoader) GetTestSuiteYaml(suite string) (testSuiteYaml []byte, err error) {
-    var testSuite testing.TestSuite
-    if testSuite, err = g.GetTestSuite(suite, true); err == nil {
-        testSuiteYaml, err = testing.ToYAML(&testSuite)
-    }
-    return
+	var testSuite testing.TestSuite
+	if testSuite, err = g.GetTestSuite(suite, true); err == nil {
+		testSuiteYaml, err = testing.ToYAML(&testSuite)
+	}
+	return
 }
 
 func (g *gRPCLoader) ListTestCase(suite string) (testcases []testing.TestCase, err error) {
-    var testCases *server.TestCases
-    testCases, err = g.client.ListTestCases(g.ctx, &TestSuite{
-        Name: suite,
-    })
+	var testCases *server.TestCases
+	testCases, err = g.client.ListTestCases(g.ctx, &TestSuite{
+		Name: suite,
+	})
 
-    if err == nil && testCases.Data != nil {
-        for _, item := range testCases.Data {
-            if item.Name == "" {
-                continue
-            }
-            testcases = append(testcases, ConvertToNormalTestCase(item))
-        }
-    }
-    return
+	if err == nil && testCases.Data != nil {
+		for _, item := range testCases.Data {
+			if item.Name == "" {
+				continue
+			}
+			testcases = append(testcases, ConvertToNormalTestCase(item))
+		}
+	}
+	return
 }
 func (g *gRPCLoader) GetTestCase(suite, name string) (testcase testing.TestCase, err error) {
-    var result *server.TestCase
-    result, err = g.client.GetTestCase(g.ctx, &server.TestCase{
-        Name:      name,
-        SuiteName: suite,
-    })
-    if err == nil && result != nil {
-        testcase = ConvertToNormalTestCase(result)
-    }
-    return
+	var result *server.TestCase
+	result, err = g.client.GetTestCase(g.ctx, &server.TestCase{
+		Name:      name,
+		SuiteName: suite,
+	})
+	if err == nil && result != nil {
+		testcase = ConvertToNormalTestCase(result)
+	}
+	return
 }
 
 func (g *gRPCLoader) GetHistoryTestCaseWithResult(id string) (result testing.HistoryTestResult, err error) {
-    var historyTestResult *server.HistoryTestResult
-    historyTestResult, err = g.client.GetHistoryTestCaseWithResult(g.ctx, &server.HistoryTestCase{
-        ID: id,
-    })
-    if err == nil && historyTestResult != nil {
-        result = ConvertToNormalTestCaseResult(historyTestResult)
-    }
-    return
+	var historyTestResult *server.HistoryTestResult
+	historyTestResult, err = g.client.GetHistoryTestCaseWithResult(g.ctx, &server.HistoryTestCase{
+		ID: id,
+	})
+	if err == nil && historyTestResult != nil {
+		result = ConvertToNormalTestCaseResult(historyTestResult)
+	}
+	return
 }
 
 func (g *gRPCLoader) GetHistoryTestCase(id string) (result testing.HistoryTestCase, err error) {
-    var historyTestCase *server.HistoryTestCase
-    historyTestCase, err = g.client.GetHistoryTestCase(g.ctx, &server.HistoryTestCase{
-        ID: id,
-    })
-    if err == nil && historyTestCase != nil {
-        result = ConvertToNormalHistoryTestCase(historyTestCase)
-    }
-    return
+	var historyTestCase *server.HistoryTestCase
+	historyTestCase, err = g.client.GetHistoryTestCase(g.ctx, &server.HistoryTestCase{
+		ID: id,
+	})
+	if err == nil && historyTestCase != nil {
+		result = ConvertToNormalHistoryTestCase(historyTestCase)
+	}
+	return
 }
 
 func (g *gRPCLoader) CreateTestCase(suite string, testcase testing.TestCase) (err error) {
-    payload := ConvertToGRPCTestCase(testcase)
-    payload.SuiteName = suite
-    _, err = g.client.CreateTestCase(g.ctx, payload)
-    return
+	payload := ConvertToGRPCTestCase(testcase)
+	payload.SuiteName = suite
+	_, err = g.client.CreateTestCase(g.ctx, payload)
+	return
 }
 
 func (g *gRPCLoader) UpdateTestCase(suite string, testcase testing.TestCase) (err error) {
-    payload := ConvertToGRPCTestCase(testcase)
-    payload.SuiteName = suite
-    _, err = g.client.UpdateTestCase(g.ctx, payload)
-    return
+	payload := ConvertToGRPCTestCase(testcase)
+	payload.SuiteName = suite
+	_, err = g.client.UpdateTestCase(g.ctx, payload)
+	return
 }
 
 func (g *gRPCLoader) DeleteTestCase(suite, testcase string) (err error) {
-    _, err = g.client.DeleteTestCase(g.ctx, &server.TestCase{
-        Name:      testcase,
-        SuiteName: suite,
-    })
-    return
+	_, err = g.client.DeleteTestCase(g.ctx, &server.TestCase{
+		Name:      testcase,
+		SuiteName: suite,
+	})
+	return
 }
 
 func (g *gRPCLoader) RenameTestCase(suite, oldName, newName string) (err error) {
-    _, err = g.client.RenameTestCase(g.ctx, &server.TestCaseDuplicate{
-        SourceCaseName:  oldName,
-        SourceSuiteName: suite,
-        TargetCaseName:  newName,
-    })
-    return
+	_, err = g.client.RenameTestCase(g.ctx, &server.TestCaseDuplicate{
+		SourceCaseName:  oldName,
+		SourceSuiteName: suite,
+		TargetCaseName:  newName,
+	})
+	return
 }
 
 func (g *gRPCLoader) CreateHistoryTestCase(testcaseResult testing.TestCaseResult, testSuite *testing.TestSuite, historyHeader map[string]string) (err error) {
-    payload := ConvertToGRPCHistoryTestCaseResult(testcaseResult, testSuite, historyHeader)
-    _, err = g.client.CreateTestCaseHistory(g.ctx, payload)
-    return
+	payload := ConvertToGRPCHistoryTestCaseResult(testcaseResult, testSuite, historyHeader)
+	_, err = g.client.CreateTestCaseHistory(g.ctx, payload)
+	return
 }
 
 func (g *gRPCLoader) ListHistoryTestSuite() (suites []testing.HistoryTestSuite, err error) {
-    var items *HistoryTestSuites
-    items, err = g.client.ListHistoryTestSuite(g.ctx, &server.Empty{})
-    if err == nil && items != nil {
-        for _, item := range items.Data {
-            suites = append(suites, *ConvertToNormalHistoryTestSuite(item))
-        }
-    }
-    return
+	var items *HistoryTestSuites
+	items, err = g.client.ListHistoryTestSuite(g.ctx, &server.Empty{})
+	if err == nil && items != nil {
+		for _, item := range items.Data {
+			suites = append(suites, *ConvertToNormalHistoryTestSuite(item))
+		}
+	}
+	return
 }
 
 func (g *gRPCLoader) DeleteHistoryTestCase(id string) (err error) {
-    _, err = g.client.DeleteHistoryTestCase(g.ctx, &server.HistoryTestCase{
-        ID: id,
-    })
-    return
+	_, err = g.client.DeleteHistoryTestCase(g.ctx, &server.HistoryTestCase{
+		ID: id,
+	})
+	return
 }
 
 func (g *gRPCLoader) DeleteAllHistoryTestCase(suite, name string) (err error) {
-    _, err = g.client.DeleteAllHistoryTestCase(g.ctx, &server.HistoryTestCase{
-        SuiteName: suite,
-        CaseName:  name,
-    })
-    return
+	_, err = g.client.DeleteAllHistoryTestCase(g.ctx, &server.HistoryTestCase{
+		SuiteName: suite,
+		CaseName:  name,
+	})
+	return
 }
 
 func (g *gRPCLoader) GetTestCaseAllHistory(suite, name string) (historyTestcases []testing.HistoryTestCase, err error) {
-    var historyTestCases *server.HistoryTestCases
-    historyTestCases, err = g.client.GetTestCaseAllHistory(g.ctx, &server.TestCase{
-        Name:      name,
-        SuiteName: suite,
-    })
-    if err == nil && historyTestCases.Data != nil {
-        for _, item := range historyTestCases.Data {
-            historyTestcases = append(historyTestcases, ConvertToNormalHistoryTestCase(item))
-        }
-    }
-    return
+	var historyTestCases *server.HistoryTestCases
+	historyTestCases, err = g.client.GetTestCaseAllHistory(g.ctx, &server.TestCase{
+		Name:      name,
+		SuiteName: suite,
+	})
+	if err == nil && historyTestCases.Data != nil {
+		for _, item := range historyTestCases.Data {
+			historyTestcases = append(historyTestcases, ConvertToNormalHistoryTestCase(item))
+		}
+	}
+	return
 }
 
 func (g *gRPCLoader) ListTestSuite() (suites []testing.TestSuite, err error) {
-    var items *TestSuites
-    items, err = g.client.ListTestSuite(g.ctx, &server.Empty{})
-    if err == nil && items != nil {
-        for _, item := range items.Data {
-            suites = append(suites, *ConvertToNormalTestSuite(item))
-        }
-    }
-    return
+	var items *TestSuites
+	items, err = g.client.ListTestSuite(g.ctx, &server.Empty{})
+	if err == nil && items != nil {
+		for _, item := range items.Data {
+			suites = append(suites, *ConvertToNormalTestSuite(item))
+		}
+	}
+	return
 }
 
 func (g *gRPCLoader) GetTestSuite(name string, full bool) (suite testing.TestSuite, err error) {
-    var result *TestSuite
-    if result, err = g.client.GetTestSuite(g.ctx,
-        &TestSuite{Name: name, Full: full}); err == nil {
-        suite = *ConvertToNormalTestSuite(result)
-    }
-    return
+	var result *TestSuite
+	if result, err = g.client.GetTestSuite(g.ctx,
+		&TestSuite{Name: name, Full: full}); err == nil {
+		suite = *ConvertToNormalTestSuite(result)
+	}
+	return
 }
 
 func (g *gRPCLoader) CreateSuite(name, api string) (err error) {
-    _, err = g.client.CreateTestSuite(g.ctx, &TestSuite{
-        Name: name,
-        Api:  api,
-    })
-    return
+	_, err = g.client.CreateTestSuite(g.ctx, &TestSuite{
+		Name: name,
+		Api:  api,
+	})
+	return
 }
 
 func (g *gRPCLoader) GetSuite(name string) (reply *testing.TestSuite, _ string, err error) {
-    var suite *TestSuite
-    if suite, err = g.client.GetTestSuite(g.ctx,
-        &TestSuite{Name: name}); err != nil {
-        return
-    }
+	var suite *TestSuite
+	if suite, err = g.client.GetTestSuite(g.ctx,
+		&TestSuite{Name: name}); err != nil {
+		return
+	}
 
-    reply = ConvertToNormalTestSuite(suite)
-    return
+	reply = ConvertToNormalTestSuite(suite)
+	return
 }
 
 func (g *gRPCLoader) UpdateSuite(suite testing.TestSuite) (err error) {
-    _, err = g.client.UpdateTestSuite(g.ctx, ConvertToGRPCTestSuite(&suite))
-    return
+	_, err = g.client.UpdateTestSuite(g.ctx, ConvertToGRPCTestSuite(&suite))
+	return
 }
 
 func (g *gRPCLoader) DeleteSuite(name string) (err error) {
-    _, err = g.client.DeleteTestSuite(g.ctx, &TestSuite{
-        Name: name,
-    })
-    return
+	_, err = g.client.DeleteTestSuite(g.ctx, &TestSuite{
+		Name: name,
+	})
+	return
 }
 
 func (g *gRPCLoader) RenameTestSuite(oldName, newName string) (err error) {
-    _, err = g.client.RenameTestSuite(g.ctx, &server.TestSuiteDuplicate{
-        SourceSuiteName: oldName,
-        TargetSuiteName: newName,
-    })
-    return
+	_, err = g.client.RenameTestSuite(g.ctx, &server.TestSuiteDuplicate{
+		SourceSuiteName: oldName,
+		TargetSuiteName: newName,
+	})
+	return
 }
 
 func (g *gRPCLoader) Verify() (readOnly bool, err error) {
-    // avoid to long to wait the response
-    ctx, cancel := context.WithTimeout(g.ctx, time.Second*5)
-    defer cancel()
+	// avoid to long to wait the response
+	ctx, cancel := context.WithTimeout(g.ctx, time.Second*5)
+	defer cancel()
 
-    var result *server.ExtensionStatus
-    if result, err = g.client.Verify(ctx, &server.Empty{}); err == nil {
-        readOnly = result.ReadOnly
-        if !result.Ready {
-            err = errors.New(result.Message)
-        }
-    }
-    return
+	var result *server.ExtensionStatus
+	if result, err = g.client.Verify(ctx, &server.Empty{}); err == nil {
+		readOnly = result.ReadOnly
+		if !result.Ready {
+			err = errors.New(result.Message)
+		}
+	}
+	return
 }
 
 func (g *gRPCLoader) PProf(name string) []byte {
-    data, err := g.client.PProf(context.Background(), &server.PProfRequest{
-        Name: name,
-    })
-    if err != nil {
-        grpcLogger.Info("failed to get pprof:", "error", err)
-    }
-    return data.Data
+	data, err := g.client.PProf(context.Background(), &server.PProfRequest{
+		Name: name,
+	})
+	if err != nil {
+		grpcLogger.Info("failed to get pprof:", "error", err)
+	}
+	return data.Data
 }
 
-func (g *gRPCLoader) Query(query map[string]string) (result map[string]string, items []map[string]string, err error) {
-    var dataResult *server.DataQueryResult
-    if dataResult, err = g.client.Query(g.ctx, &server.DataQuery{
-        Sql: query["sql"],
-        Key: query["key"],
-    }); err == nil {
-        result = pairToMap(dataResult.Data)
-        for _, item := range dataResult.Items {
-            items = append(items, pairToMap(item.Data))
-        }
-    }
-    return
+func (g *gRPCLoader) Query(query map[string]string) (result testing.DataResult, err error) {
+	var dataResult *server.DataQueryResult
+	if dataResult, err = g.client.Query(g.ctx, &server.DataQuery{
+		Sql: query["sql"],
+		Key: query["key"],
+	}); err == nil {
+		result.Pairs = pairToMap(dataResult.Data)
+		for _, item := range dataResult.Items {
+			result.Rows = append(result.Rows, pairToMap(item.Data))
+		}
+		result.Databases = dataResult.Meta.Databases
+		result.Tables = dataResult.Meta.Tables
+		result.CurrentDatabase = dataResult.Meta.CurrentDatabase
+	}
+	return
 }
 
 func (g *gRPCLoader) Close() {
-    if g.conn != nil {
-        g.conn.Close()
-    }
+	if g.conn != nil {
+		g.conn.Close()
+	}
 }

pkg/testing/remote/loader.pb.go

@@ -1,7 +1,7 @@
 // Code generated by protoc-gen-go. DO NOT EDIT.
 // versions:
 // 	protoc-gen-go v1.28.1
-// 	protoc        v4.25.1
+// 	protoc        v4.22.2
 // source: pkg/testing/remote/loader.proto
 
 package remote

pkg/testing/remote/loader_grpc.pb.go

@@ -1,7 +1,7 @@
 // Code generated by protoc-gen-go-grpc. DO NOT EDIT.
 // versions:
 // - protoc-gen-go-grpc v1.2.0
-// - protoc             v4.25.1
+// - protoc             v4.22.2
 // source: pkg/testing/remote/loader.proto
 
 package remote