From 625cbef48c7b33a66568ce071fa62dd3cf98a05f Mon Sep 17 00:00:00 2001
From: Lu Ken <bluewish.ken.lu@live.cn>
Date: Wed, 1 Jan 2025 14:02:35 +0800
Subject: [PATCH] save the server address into browser's local storage

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

diff --git a/src/GenTradeAgent/src/renderer/src/components/FootBar.vue b/src/GenTradeAgent/src/renderer/src/components/FootBar.vue
index 1fcd177..66bbcda 100644
--- a/src/GenTradeAgent/src/renderer/src/components/FootBar.vue
+++ b/src/GenTradeAgent/src/renderer/src/components/FootBar.vue
@@ -79,11 +79,16 @@ const onClickNetwork = () => {
   showServerModal.value = true
 }
 
+const callbackPing = (latency: number) => {
+  store.commit('updateServerConnection', latency)
+}
+
 const onPositiveClick = () => {
   console.log('ok')
   agentServer.serverAddress = serverAddress.value
   agentServer.pingInterval = pingInterval.value * 1000
   showServerModal.value = false
+  agentServer.ping(callbackPing)
 }
 </script>
 <style lang="scss" scoped>
diff --git a/src/GenTradeAgent/src/renderer/src/components/TradingMain.vue b/src/GenTradeAgent/src/renderer/src/components/TradingMain.vue
index 332d4a4..ea09637 100644
--- a/src/GenTradeAgent/src/renderer/src/components/TradingMain.vue
+++ b/src/GenTradeAgent/src/renderer/src/components/TradingMain.vue
@@ -105,9 +105,12 @@ const handleUpdateCurrentInterval = (value: string) => {
   store.commit('updateCurrentInterval', value)
 }
 
-agentServer.store = useStore()
+const callbackPing = (latency: number) => {
+  store.commit('updateServerConnection', latency)
+}
+
 const handlerPingServer = () => {
-  agentServer.ping()
+  agentServer.ping(callbackPing)
   setTimeout(handlerPingServer, agentServer.pingInterval)
 }
 handlerPingServer()
diff --git a/src/GenTradeAgent/src/renderer/src/server.ts b/src/GenTradeAgent/src/renderer/src/server.ts
index c78c71e..ed1f651 100644
--- a/src/GenTradeAgent/src/renderer/src/server.ts
+++ b/src/GenTradeAgent/src/renderer/src/server.ts
@@ -1,17 +1,25 @@
 import axios, { AxiosError } from 'axios'
-import { IState } from '@renderer/store'
-import { Store } from 'vuex'
 
 class AgentServer {
-  serverAddress: string = 'http://47.100.216.225:8000/api/v1'
-  //serverAddress: string = 'http://127.0.0.1:8000/api/v1'
+  static DEFAULT_SERVER_ADDRESS = 'http://47.100.216.225:8000/api/v1'
   apiKey: string = 'e54d4431-5dab-474e-b71a-0db1fcb9e659'
   tzName: string = ''
   tzOffset: number = 0
-  store: Store<IState> | null = null
   pingInterval: number = 10000
 
-  ping() {
+  get serverAddress() {
+    const retval = localStorage.getItem('serverAddress')
+    if (retval == null) {
+      return AgentServer.DEFAULT_SERVER_ADDRESS
+    }
+    return retval
+  }
+
+  set serverAddress(newVal) {
+    localStorage.setItem('serverAddress', newVal)
+  }
+
+  ping(callback: (lantecy: number) => void) {
     const tsStart = new Date().getTime()
     const address = this.serverAddress + '/public/server_time'
 
@@ -21,16 +29,11 @@ class AgentServer {
         const tsEnd = new Date().getTime()
         this.tzName = response.data['timezone_name']
         this.tzOffset = response.data['timezone_offset']
-
-        if (this.store) {
-          this.store.commit('updateServerConnection', (tsEnd - tsStart) / 2)
-        }
+        callback((tsEnd - tsStart) / 2)
       })
       .catch((err: AxiosError) => {
         console.log(err)
-        if (this.store) {
-          this.store.commit('updateServerConnection', -1)
-        }
+        callback(-1)
       })
   }