From 8a1e72f23de7372b5ce0d8c8e39ee6475c5966e1 Mon Sep 17 00:00:00 2001
From: rick <linuxsuren@users.noreply.github.com>
Date: Tue, 11 Mar 2025 08:29:47 +0800
Subject: [PATCH] feat: enhance DataManager to display query duration and
 result count

---
 console/atest-ui/src/views/DataManager.vue | 6 ++++--
 pkg/server/remote_server.go                | 1 +
 2 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/console/atest-ui/src/views/DataManager.vue b/console/atest-ui/src/views/DataManager.vue
index 6c0a398..7ad659f 100644
--- a/console/atest-ui/src/views/DataManager.vue
+++ b/console/atest-ui/src/views/DataManager.vue
@@ -205,8 +205,10 @@ const executeQuery = async () => {
               </el-form>
           </el-header>
           <el-main>
-              <el-tag type="primary" v-if="queryResult.length > 0">{{ queryResult.length }} rows</el-tag>
-              <el-tag type="primary" v-if="queryDataMeta.duration">{{  queryDataMeta.duration }}</el-tag>
+              <div style="display: flex; gap: 8px;">
+                <el-tag type="primary" v-if="queryResult.length > 0">{{ queryResult.length }} rows</el-tag>
+                <el-tag type="primary" v-if="queryDataMeta.duration">{{  queryDataMeta.duration }}</el-tag>
+              </div>
               <el-table :data="queryResult" stripe v-if="dataFormat === 'table'">
                   <el-table-column v-for="col in columns" :key="col" :prop="col" :label="col" sortable/>
               </el-table>
diff --git a/pkg/server/remote_server.go b/pkg/server/remote_server.go
index 4c2d37b..76bc480 100644
--- a/pkg/server/remote_server.go
+++ b/pkg/server/remote_server.go
@@ -1261,6 +1261,7 @@ func (s *server) Query(ctx context.Context, query *DataQuery) (result *DataQuery
 			CurrentDatabase: dataResult.CurrentDatabase,
 			Databases:       dataResult.Databases,
 			Tables:          dataResult.Tables,
+			Duration:        dataResult.Duration,
 		}
 	}
 	return