fix the key-value context pass

Author: LinuxSuRen

pkg/server/remote_server.go

@@ -1250,12 +1250,7 @@ func (s *server) Query(ctx context.Context, query *DataQuery) (result *DataQuery
 		"key": query.Key,
 	}); err == nil {
 		result = &DataQueryResult{
-			Data: []*Pair{
-				{
-					Key:   "data",
-					Value: fmt.Sprintf("%v", dataResult),
-				},
-			},
+			Data: mapToPair(dataResult),
 		}
 	}
 	return

pkg/testing/remote/grpc_store.go

@@ -19,7 +19,6 @@ package remote
 import (
 	context "context"
 	"errors"
-	"fmt"
 	"time"
 
 	"github.com/linuxsuren/api-testing/pkg/logging"
@@ -309,8 +308,7 @@ func (g *gRPCLoader) PProf(name string) []byte {
 
 func (g *gRPCLoader) Query(query map[string]string) (result map[string]string, err error) {
 	var dataResult *server.DataQueryResult
-	fmt.Println("query", query)
-	if dataResult, err = g.client.Query(context.Background(), &server.DataQuery{
+	if dataResult, err = g.client.Query(g.ctx, &server.DataQuery{
 		Sql: query["sql"],
 		Key: query["key"],
 	}); err == nil {