Merge pull request #661 from LinuxSuRen/feat/elasticsearch

feat: add elasticsearch extension support

Author: LinuxSuRen

console/atest-ui/src/components/Button.vue

@@ -34,8 +34,17 @@ watch(store, (s) => {
             return
         }
     })
-    queryDataMeta.value.currentDatabase = ''
-    sqlQuery.value = ''
+
+    switch (kind.value) {
+        case 'atest-store-elasticsearch':
+        case 'atest-store-etcd':
+            sqlQuery.value = '*'
+            break
+        default:
+            queryDataMeta.value.currentDatabase = ''
+            sqlQuery.value = ''
+    }
+
     executeQuery()
 })
 
@@ -63,13 +72,22 @@ const describeTable = (data: QueryData) => {
         case 'atest-store-cassandra':
             sqlQuery.value = `@describeTable_${queryDataMeta.value.currentDatabase}:${data.label}`
             break
+            break
         default:
             sqlQuery.value = `@describeTable_${data.label}`
     }
     executeQuery()
 }
 const queryTables = () => {
-    sqlQuery.value = ``
+    switch (kind.value) {
+        case 'atest-store-elasticsearch':
+            if (sqlQuery.value === '') {
+                sqlQuery.value = '*'
+            }
+            break
+        default:
+            sqlQuery.value = ``
+    }
     executeQuery()
 }
 watch(kind, (k) => {
@@ -84,6 +102,9 @@ watch(kind, (k) => {
         case 'atest-store-redis':
             queryTip.value = 'Enter key'
             break;
+        case 'atest-store-elasticsearch':
+            queryTip.value = 'field:value OR field:other'
+            break;
     }
 })
 
@@ -153,19 +174,14 @@ const executeQuery = async () => {
     return executeWithQuery(sqlQuery.value)
 }
 const executeWithQuery = async (sql: string) => {
-    switch (kind.value) {
-        case 'atest-store-etcd':
-            sqlQuery.value = '*'
-            break;
-    }
-
     let success = false
     try {
         const data = await API.DataQueryAsync(store.value, kind.value, queryDataMeta.value.currentDatabase, sql);
         switch (kind.value) {
             case 'atest-store-orm':
             case 'atest-store-cassandra':
             case 'atest-store-iotdb':
+            case 'atest-store-elasticsearch':
                 ormDataHandler(data)
                 success = true
                 break;
@@ -196,7 +212,7 @@ const executeWithQuery = async (sql: string) => {
 <template>
     <div>
         <el-container style="height: calc(100vh - 50px);">
-            <el-aside v-if="kind === 'atest-store-orm' || kind === 'atest-store-iotdb' || kind === 'atest-store-cassandra'">
+            <el-aside v-if="kind === 'atest-store-orm' || kind === 'atest-store-iotdb' || kind === 'atest-store-cassandra' || kind === 'atest-store-elasticsearch'">
                 <el-scrollbar>
                     <el-select v-model="queryDataMeta.currentDatabase" placeholder="Select database"
                         @change="queryTables" filterable>

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

@@ -19,7 +19,7 @@ import { SupportedExtension, SupportedExtensions } from "../store";
 describe("SupportedExtensions", () => {
     test('length check', () => {
         const extensions = SupportedExtensions()
-        expect(extensions.length).toBe(8)
+        expect(extensions.length).toBe(9)
     })
 
     for (const extension of SupportedExtensions()) {

console/atest-ui/src/views/__test__/store.spec.ts

@@ -785,21 +785,10 @@ interface QueryObject {
     key: string
 }
 var DataQueryAsync = (store: string, kind: string, currentDatabase: string, query: string) => {
-    const queryObj = {} as QueryObject;
-    switch (kind) {
-        case 'atest-store-orm':
-        case 'atest-store-cassandra':
-        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 queryObj = {
+        'key': currentDatabase,
+        'sql': query,
+    } as QueryObject;
     const requestOptions = {
         method: 'POST',
         headers: {

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

@@ -115,6 +115,11 @@ const storeExtensions = [
             defaultValue: 'atest'
         }],
         link: 'https://github.com/LinuxSuRen/atest-ext-store-mongodb'
+    },
+    {
+        name: 'atest-store-elasticsearch',
+        params: [],
+        link: 'https://github.com/LinuxSuRen/atest-ext-store-elasticsearch'
     }
 ] as Store[]
 

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

@@ -10,6 +10,7 @@ Ports in extensions:
 | Store | [redis](https://github.com/LinuxSuRen/atest-ext-store-redis)             |  |
 | Store | [iotdb](https://github.com/LinuxSuRen/atest-ext-store-iotdb) | |
 | Store | [Cassandra](https://github.com/LinuxSuRen/atest-ext-store-cassandra) | |
+| Store | [Elasticsearch](https://github.com/LinuxSuRen/atest-ext-store-elasticsearch) | |
 | Monitor | [docker-monitor](https://github.com/LinuxSuRen/atest-ext-monitor-docker) |  |
 | Agent | [collector](https://github.com/LinuxSuRen/atest-ext-collector)           |  |
 | Secret | [Vault](https://github.com/LinuxSuRen/api-testing-vault-extension)       | |