From eac410ac79f86599ac7d4cb3352412ceac3f7d9b Mon Sep 17 00:00:00 2001
From: Lu Ken <bluewish.ken.lu@live.cn>
Date: Sun, 29 Dec 2024 10:19:08 +0800
Subject: [PATCH] ui: add config for server ping interval (#18)

Signed-off-by: Lu Ken <bluewish.ken.lu@live.cn>
---
 .../src/renderer/src/components/FootBar.vue   | 27 +++++++------------
 .../renderer/src/components/TradingMain.vue   |  7 +++--
 src/GenTradeAgent/src/renderer/src/server.ts  |  1 +
 3 files changed, 14 insertions(+), 21 deletions(-)

diff --git a/src/GenTradeAgent/src/renderer/src/components/FootBar.vue b/src/GenTradeAgent/src/renderer/src/components/FootBar.vue
index 1d60e9d..1fcd177 100644
--- a/src/GenTradeAgent/src/renderer/src/components/FootBar.vue
+++ b/src/GenTradeAgent/src/renderer/src/components/FootBar.vue
@@ -5,28 +5,18 @@
     </div>
     <n-space class="foot-status">
       <span class="status-icon">Version v0.1</span>
-      <n-button
-        v-show="isServerConnected"
-        text
-        style="font-size: 18px"
-        @click="handlerClickNetwork"
-      >
+      <n-button v-show="isServerConnected" text style="font-size: 18px" @click="onClickNetwork">
         <n-icon>
           <PlugConnected20Filled />
         </n-icon>
       </n-button>
-      <n-button
-        v-show="!isServerConnected"
-        text
-        style="font-size: 18px"
-        @click="handlerClickNetwork"
-      >
+      <n-button v-show="!isServerConnected" text style="font-size: 18px" @click="onClickNetwork">
         <n-icon>
           <PlugDisconnected20Regular />
         </n-icon>
       </n-button>
-      <div style="width: 30px; max-width: 30px;">
-        <span style="align-items: center;">{{ serverLatency }}ms</span>
+      <div style="width: 30px; max-width: 30px">
+        <span style="align-items: center">{{ serverLatency }}ms</span>
       </div>
     </n-space>
     <n-modal
@@ -42,6 +32,8 @@
         <n-input v-model:value="serverAddress"></n-input>
         API Key
         <n-input v-model:value="apiKey"></n-input>
+        Ping Interval (seconds)
+        <n-input-number v-model:value="pingInterval"></n-input-number>
       </n-space>
       <template #footer>
         <n-button @click="onPositiveClick">OK</n-button>
@@ -52,13 +44,14 @@
 <script setup lang="ts">
 import { ref } from 'vue'
 import { PlugConnected20Filled, PlugDisconnected20Regular } from '@vicons/fluent'
-import { NIcon, NSpace, NButton, NModal, NInput } from 'naive-ui'
+import { NIcon, NSpace, NButton, NModal, NInput, NInputNumber } from 'naive-ui'
 import { useStore } from '../store'
 import { agentServer } from '@renderer/server'
 
 const showServerModal = ref(false)
 const serverAddress = ref(agentServer.serverAddress)
 const apiKey = ref(agentServer.apiKey)
+const pingInterval = ref(agentServer.pingInterval / 1000)
 
 const store = useStore()
 
@@ -81,7 +74,7 @@ store.watch(
   }
 )
 
-const handlerClickNetwork = () => {
+const onClickNetwork = () => {
   console.log('hehe')
   showServerModal.value = true
 }
@@ -89,9 +82,9 @@ const handlerClickNetwork = () => {
 const onPositiveClick = () => {
   console.log('ok')
   agentServer.serverAddress = serverAddress.value
+  agentServer.pingInterval = pingInterval.value * 1000
   showServerModal.value = false
 }
-
 </script>
 <style lang="scss" scoped>
 .foot-main {
diff --git a/src/GenTradeAgent/src/renderer/src/components/TradingMain.vue b/src/GenTradeAgent/src/renderer/src/components/TradingMain.vue
index 5a1cb0e..ca4237f 100644
--- a/src/GenTradeAgent/src/renderer/src/components/TradingMain.vue
+++ b/src/GenTradeAgent/src/renderer/src/components/TradingMain.vue
@@ -105,12 +105,11 @@ const handleUpdateCurrentInterval = (value: string) => {
 }
 
 agentServer.store = useStore()
-const handlerPingServer = (() => {
+const handlerPingServer = () => {
   agentServer.ping()
-  setTimeout(handlerPingServer, 10000)
-})
+  setTimeout(handlerPingServer, agentServer.pingInterval)
+}
 handlerPingServer()
-
 </script>
 
 <style lang="scss" scoped>
diff --git a/src/GenTradeAgent/src/renderer/src/server.ts b/src/GenTradeAgent/src/renderer/src/server.ts
index 5e4cc97..1283561 100644
--- a/src/GenTradeAgent/src/renderer/src/server.ts
+++ b/src/GenTradeAgent/src/renderer/src/server.ts
@@ -8,6 +8,7 @@ class AgentServer {
   tzName: string = ''
   tzOffset: number = 0
   store: Store<IState> | null = null
+  pingInterval: number = 10000
 
   ping() {
     const tsStart = new Date().getTime()