From c4eca6cbc121fd79c705f828b55b01fa86311539 Mon Sep 17 00:00:00 2001
From: rick <LinuxSuRen@users.noreply.github.com>
Date: Wed, 26 Mar 2025 11:45:03 +0000
Subject: [PATCH] feat: add iotdb support

---
 console/atest-ui/src/views/DataManager.vue |  4 +++
 console/atest-ui/src/views/net.ts          | 37 ++++++++++++++++++++++
 console/atest-ui/src/views/store.ts        |  5 +++
 pkg/server/remote_server.go                |  3 ++
 4 files changed, 49 insertions(+)

diff --git a/console/atest-ui/src/views/DataManager.vue b/console/atest-ui/src/views/DataManager.vue
index 67678af..4537676 100644
--- a/console/atest-ui/src/views/DataManager.vue
+++ b/console/atest-ui/src/views/DataManager.vue
@@ -116,6 +116,10 @@ const executeQuery = async () => {
             case 'atest-store-orm':
                 ormDataHandler(data)
                 break;
+            case 'atest-store-iotdb':
+                ormDataHandler(data)
+                success = true
+                break;
             case 'atest-store-etcd':
                 keyValueDataHandler(data)
                 break;
diff --git a/console/atest-ui/src/views/net.ts b/console/atest-ui/src/views/net.ts
index ee632eb..4765cb6 100644
--- a/console/atest-ui/src/views/net.ts
+++ b/console/atest-ui/src/views/net.ts
@@ -779,6 +779,43 @@ var SBOM = (callback: (d: any) => void) => {
       .then(callback)
 }
 
+<<<<<<< Updated upstream
+=======
+interface QueryObject {
+  sql: string
+  key: string
+}
+var DataQueryAsync = (store: string, kind: string, currentDatabase: string, query: string) => {
+  const queryObj = {} as QueryObject;
+  switch (kind) {
+      case 'atest-store-orm':
+          queryObj['sql'] = query;
+          queryObj['key'] = currentDatabase;
+          break;
+      case 'atest-store-iotdb':
+          queryObj['sql'] = query;
+          queryObj['key'] = currentDatabase;
+          break;
+      case 'atest-store-etcd':
+          queryObj['key'] = query;
+          break;
+      case 'atest-store-redis':
+          queryObj['key'] = query;
+          break;
+  }
+  const requestOptions = {
+      method: 'POST',
+      headers: {
+          'X-Store-Name': store,
+          'X-Database': currentDatabase
+      },
+      body: JSON.stringify(queryObj)
+  }
+  return fetch(`/api/v1/data/query`, requestOptions)
+      .then(DefaultResponseProcess)
+}
+
+>>>>>>> Stashed changes
 var DataQuery = (store: string, kind: string, currentDatabase: string, query: string, callback: (d: any) => void, errHandler: (d: any) => void) => {
     const queryObj = {}
     switch (kind) {
diff --git a/console/atest-ui/src/views/store.ts b/console/atest-ui/src/views/store.ts
index 8dde35f..55f4be9 100644
--- a/console/atest-ui/src/views/store.ts
+++ b/console/atest-ui/src/views/store.ts
@@ -78,6 +78,11 @@ const storeExtensions = [
         }],
         link: 'https://github.com/LinuxSuRen/atest-ext-store-orm'
     },
+    {
+        name: 'atest-store-iotdb',
+        params: [],
+        link: 'https://github.com/LinuxSuRen/atest-ext-store-iotdb'
+    },
     {
         name: 'atest-store-etcd',
         params: [],
diff --git a/pkg/server/remote_server.go b/pkg/server/remote_server.go
index 4c2d37b..1af4879 100644
--- a/pkg/server/remote_server.go
+++ b/pkg/server/remote_server.go
@@ -1157,6 +1157,9 @@ func (s *server) GetStores(ctx context.Context, in *Empty) (reply *Stores, err e
 		}
 		for _, item := range stores {
 			grpcStore := ToGRPCStore(item)
+			if item.Disabled {
+				continue
+			}
 
 			storeStatus, sErr := s.VerifyStore(ctx, &SimpleQuery{Name: item.Name})
 			grpcStore.Ready = sErr == nil && storeStatus.Ready