feat: support to query next page in elastic (#663)

* feat: support to query next page in elastic

* fix the typescript syntax error

---------

Co-authored-by: rick <[email protected]>

Author: LinuxSuRen

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

@@ -1,6 +1,7 @@
 <script setup lang="ts">
 import { ref, watch } from 'vue'
 import { API } from './net'
+import type { QueryObject } from './net'
 import type { Store } from './store'
 import type { Pair } from './types'
 import { ElMessage } from 'element-plus'
@@ -12,6 +13,10 @@ import { Refresh, Document } from '@element-plus/icons-vue'
 const stores: Ref<Store[]> = ref([])
 const kind = ref('')
 const store = ref('')
+const query = ref({
+    offset: 0,
+    limit: 10
+} as QueryObject)
 const sqlQuery = ref('')
 const queryResult = ref([] as any[])
 const queryResultAsJSON = ref('')
@@ -175,8 +180,12 @@ const executeQuery = async () => {
 }
 const executeWithQuery = async (sql: string) => {
     let success = false
+    query.value.store = store.value
+    query.value.key = queryDataMeta.value.currentDatabase
+    query.value.sql = sql
+
     try {
-        const data = await API.DataQueryAsync(store.value, kind.value, queryDataMeta.value.currentDatabase, sql);
+        const data = await API.DataQueryAsync(query.value);
         switch (kind.value) {
             case 'atest-store-orm':
             case 'atest-store-cassandra':
@@ -207,6 +216,10 @@ const executeWithQuery = async (sql: string) => {
     }
     return success
 }
+const nextPage = () => {
+    query.value.offset += query.value.limit
+    executeQuery()
+}
 </script>
 
 <template>
@@ -234,7 +247,7 @@ const executeWithQuery = async (sql: string) => {
                 </el-scrollbar>
             </el-aside>
             <el-container>
-                <el-header>
+                <el-header style="height: auto">
                     <el-form @submit.prevent="executeQuery">
                         <el-row :gutter="10">
                             <el-col :span="4">
@@ -264,6 +277,21 @@ const executeWithQuery = async (sql: string) => {
                                 </el-select>
                             </el-col>
                         </el-row>
+                        <el-row :gutter="10" v-if="kind === 'atest-store-elasticsearch'">
+                            <el-col :span="10">
+                                <el-input type="number" v-model="query.offset">
+                                    <template #prepend>Offset</template>
+                                </el-input>
+                            </el-col>
+                            <el-col :span="10">
+                                <el-input type="number" v-model="query.limit">
+                                    <template #prepend>Limit</template>
+                                </el-input>
+                            </el-col>
+                            <el-col :span="2">
+                                <el-button type="primary" @click="nextPage">Next</el-button>
+                            </el-col>
+                        </el-row>
                     </el-form>
                 </el-header>
                 <el-main>

console/atest-ui/src/views/net.ts

@@ -1,5 +1,5 @@
 /*
-Copyright 2023-2024 API Testing Authors.
+Copyright 2023-2025 API Testing Authors.
 
 Licensed under the Apache License, Version 2.0 (the "License");
 you may not use this file except in compliance with the License.
@@ -13,7 +13,6 @@ 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.
 */
-import { ca } from 'element-plus/es/locales.mjs'
 import { Cache } from './cache'
 
 async function DefaultResponseProcess(response: any) {
@@ -780,22 +779,23 @@ var SBOM = (callback: (d: any) => void) => {
         .then(callback)
 }
 
-interface QueryObject {
+export interface QueryObject {
+    store: string
     sql: string
     key: string
+    offset: number
+    limit: number
 }
-var DataQueryAsync = (store: string, kind: string, currentDatabase: string, query: string) => {
-    const queryObj = {
-        'key': currentDatabase,
-        'sql': query,
-    } as QueryObject;
+const DataQueryAsync = (query: QueryObject) => {
     const requestOptions = {
         method: 'POST',
         headers: {
-            'X-Store-Name': store,
-            'X-Database': currentDatabase
+            'X-Store-Name': query.store,
+            'X-Database': query.key,
+            'X-Offset': `${query.offset}`,
+            'X-Limit': `${query.limit}`
         },
-        body: JSON.stringify(queryObj)
+        body: JSON.stringify(query)
     }
     return fetch(`/api/v1/data/query`, requestOptions)
         .then(DefaultResponseProcess)

pkg/server/gateway.go

@@ -16,7 +16,7 @@ limitations under the License.
 package server
 
 import (
-	context "context"
+	"context"
 	"net/http"
 
 	"google.golang.org/grpc/metadata"

pkg/server/remote_server.go

@@ -1249,8 +1249,10 @@ func (s *server) Query(ctx context.Context, query *DataQuery) (result *DataQuery
 	defer loader.Close()
 	var dataResult testing.DataResult
 	if dataResult, err = loader.Query(map[string]string{
-		"sql": query.Sql,
-		"key": query.Key,
+		"sql":    query.Sql,
+		"key":    query.Key,
+		"offset": fmt.Sprintf("%d", query.Offset),
+		"limit":  fmt.Sprintf("%d", query.Limit),
 	}); err == nil {
 		result = &DataQueryResult{
 			Data: mapToPair(dataResult.Pairs),