root/clash-verge-rev
1
0
Fork 0
mirror of https://github.com/clash-verge-rev/clash-verge-rev.git synced 2026-01-29 17:15:38 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: add a wrapper around sockette w/ error retry (#1219)

Browse Source 
* feat: add a wrapper around sockette w/ error retry

* chore: use import path alias

* perf: reduce retry
This commit is contained in:
Sukka
2024-06-16 18:25:33 +08:00
committed by GitHub
parent 9a04208a11
commit a9149fb92e
2 changed files with 57 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

47
src/components/layout/layout-traffic.tsx
View File

@@ -12,7 +12,7 @@ import { useLogSetup } from "./use-log-setup";
import { useVisibility } from "@/hooks/use-visibility";
import parseTraffic from "@/utils/parse-traffic";
import useSWRSubscription from "swr/subscription";
import Sockette from "sockette";
import { createSockette } from "@/utils/websocket";
interface MemoryUsage {
inuse: number;
@@ -42,24 +42,17 @@ export const LayoutTraffic = () => {
(_key, { next }) => {
const { server = "", secret = "" } = clashInfo!;
let errorCount = 10;
const s = new Sockette(
const s = createSockette(
`ws://${server}/traffic?token=${encodeURIComponent(secret)}`,
{
onmessage(event) {
errorCount = 0; // reset counter
const data = JSON.parse(event.data) as ITrafficItem;
trafficRef.current?.appendData(data);
next(null, data);
},
onerror(event) {
errorCount -= 1;
if (errorCount <= 0) {
this.close();
next(event, { up: 0, down: 0 });
}
this.close();
next(event, { up: 0, down: 0 });
},
}
);
@@ -86,27 +79,23 @@ export const LayoutTraffic = () => {
clashInfo && pageVisible && displayMemory ? "getRealtimeMemory" : null,
(_key, { next }) => {
const { server = "", secret = "" } = clashInfo!;
const ws = new WebSocket(
`ws://${server}/memory?token=${encodeURIComponent(secret)}`
const s = createSockette(
`ws://${server}/memory?token=${encodeURIComponent(secret)}`,
{
onmessage(event) {
const data = JSON.parse(event.data) as MemoryUsage;
next(null, data);
},
onerror(event) {
this.close();
next(event, { inuse: 0 });
},
}
);
let errorCount = 10;
ws.addEventListener("message", (event) => {
errorCount = 0; // reset counter
next(null, JSON.parse(event.data));
});
ws.addEventListener("error", (event) => {
errorCount -= 1;
if (errorCount <= 0) {
ws.close();
next(event, { inuse: 0 });
}
});
return () => {
ws.close();
s.close();
};
},
{