YuuMJ1997/mihomo-party
1
0
Fork 0
mirror of https://gh.catmak.name/https://github.com/mihomo-party-org/mihomo-party synced 2025-12-27 05:00:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

limit the number of websocket retries

Browse Source
This commit is contained in:
pompurin404 2024-08-10 14:55:34 +08:00
parent 3d0f75242e
commit f1916c1027
No known key found for this signature in database
1 changed files with 33 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

37
src/main/core/mihomoApi.ts
View File

@ -2,12 +2,17 @@ import axios, { AxiosInstance } from 'axios'
import { getAppConfig, getControledMihomoConfig } from '../config' import { getAppConfig, getControledMihomoConfig } from '../config'
import WebSocket from 'ws' import WebSocket from 'ws'
import { window } from '..' import { window } from '..'
import { dialog } from 'electron'
let axiosIns: AxiosInstance = null! let axiosIns: AxiosInstance = null!
let mihomoTrafficWs: WebSocket | null = null let mihomoTrafficWs: WebSocket | null = null
let trafficRetry = 10
let mihomoMemoryWs: WebSocket | null = null let mihomoMemoryWs: WebSocket | null = null
let memoryRetry = 10
let mihomoLogsWs: WebSocket | null = null let mihomoLogsWs: WebSocket | null = null
let logsRetry = 10
let mihomoConnectionsWs: WebSocket | null = null let mihomoConnectionsWs: WebSocket | null = null
let connectionsRetry = 10
export const getAxios = async (force: boolean = false): Promise<AxiosInstance> => { export const getAxios = async (force: boolean = false): Promise<AxiosInstance> => {
if (axiosIns && !force) return axiosIns if (axiosIns && !force) return axiosIns
@ -176,11 +181,17 @@ const mihomoTraffic = (): void => {
mihomoTrafficWs.onmessage = (e): void => { mihomoTrafficWs.onmessage = (e): void => {
const data = e.data as string const data = e.data as string
trafficRetry = 10
window?.webContents.send('mihomoTraffic', JSON.parse(data) as IMihomoTrafficInfo) window?.webContents.send('mihomoTraffic', JSON.parse(data) as IMihomoTrafficInfo)
} }
mihomoTrafficWs.onclose = (): void => { mihomoTrafficWs.onclose = (): void => {
mihomoTraffic() if (trafficRetry) {
trafficRetry--
mihomoTraffic()
} else {
dialog.showErrorBox('External controller traffic error', 'Retry limit reached')
}
} }
mihomoTrafficWs.onerror = (): void => { mihomoTrafficWs.onerror = (): void => {
@ -215,11 +226,17 @@ const mihomoMemory = (): void => {
mihomoMemoryWs.onmessage = (e): void => { mihomoMemoryWs.onmessage = (e): void => {
const data = e.data as string const data = e.data as string
memoryRetry = 10
window?.webContents.send('mihomoMemory', JSON.parse(data) as IMihomoMemoryInfo) window?.webContents.send('mihomoMemory', JSON.parse(data) as IMihomoMemoryInfo)
} }
mihomoMemoryWs.onclose = (): void => { mihomoMemoryWs.onclose = (): void => {
mihomoMemory() if (memoryRetry) {
memoryRetry--
mihomoMemory()
} else {
dialog.showErrorBox('External controller memory error', 'Retry limit reached')
}
} }
mihomoMemoryWs.onerror = (): void => { mihomoMemoryWs.onerror = (): void => {
@ -256,11 +273,17 @@ const mihomoLogs = (): void => {
mihomoLogsWs.onmessage = (e): void => { mihomoLogsWs.onmessage = (e): void => {
const data = e.data as string const data = e.data as string
logsRetry = 10
window?.webContents.send('mihomoLogs', JSON.parse(data) as IMihomoLogInfo) window?.webContents.send('mihomoLogs', JSON.parse(data) as IMihomoLogInfo)
} }
mihomoLogsWs.onclose = (): void => { mihomoLogsWs.onclose = (): void => {
mihomoLogs() if (logsRetry) {
logsRetry--
mihomoLogs()
} else {
dialog.showErrorBox('External controller logs error', 'Retry limit reached')
}
} }
mihomoLogsWs.onerror = (): void => { mihomoLogsWs.onerror = (): void => {
@ -297,11 +320,17 @@ const mihomoConnections = (): void => {
mihomoConnectionsWs.onmessage = (e): void => { mihomoConnectionsWs.onmessage = (e): void => {
const data = e.data as string const data = e.data as string
connectionsRetry = 10
window?.webContents.send('mihomoConnections', JSON.parse(data) as IMihomoConnectionsInfo) window?.webContents.send('mihomoConnections', JSON.parse(data) as IMihomoConnectionsInfo)
} }
mihomoConnectionsWs.onclose = (): void => { mihomoConnectionsWs.onclose = (): void => {
mihomoConnections() if (connectionsRetry) {
connectionsRetry--
mihomoConnections()
} else {
dialog.showErrorBox('External controller connections error', 'Retry limit reached')
}
} }
mihomoConnectionsWs.onerror = (): void => { mihomoConnectionsWs.onerror = (): void => {