This commit is contained in:
song_xiao_lin
2024-05-09 15:06:21 +08:00
parent 0090c60141
commit f881ad6ce2
+10 -6
View File
@@ -38,15 +38,15 @@ export const Contro: React.FC<ControProps> = () => {
wsc.onWSCMessage((message) => { wsc.onWSCMessage((message) => {
if (message.action === ActionType.STATUS) { if (message.action === ActionType.STATUS) {
setConnected(message.connected);
if (message.connected === false) { if (message.connected === false) {
localStorage.setItem("yakit-connect", ""); handleConnectFail();
setAutoFindFailedReason("Yakit WebSocket Controller Connect Fail");
} else { } else {
localStorage.setItem( localStorage.setItem(
"yakit-connect", "yakit-connect",
JSON.stringify({ connected: true, port: message.port }) JSON.stringify({ connected: true, port: message.port })
); );
setEnginePort(message.port + "");
setConnected(true);
setAutoFindFailedReason(""); setAutoFindFailedReason("");
} }
} }
@@ -57,9 +57,7 @@ export const Contro: React.FC<ControProps> = () => {
const ws = new WebSocket(`ws://127.0.0.1:${port}`); const ws = new WebSocket(`ws://127.0.0.1:${port}`);
ws.onclose = (e: CloseEvent) => { ws.onclose = (e: CloseEvent) => {
if (enginePortRef.current) { if (enginePortRef.current) {
setConnected(false); handleConnectFail();
setAutoFindFailedReason("Yakit WebSocket Controller Connect Fail");
localStorage.setItem("yakit-connect", "");
return; return;
} }
if (e.reason !== `FoundYakitWebSocketController` && port + 1 <= max) { if (e.reason !== `FoundYakitWebSocketController` && port + 1 <= max) {
@@ -81,6 +79,12 @@ export const Contro: React.FC<ControProps> = () => {
}; };
}; };
const handleConnectFail = () => {
setConnected(false);
setAutoFindFailedReason("Yakit WebSocket Controller Connect Fail");
localStorage.setItem("yakit-connect", "");
};
const connectPort = (port: number) => { const connectPort = (port: number) => {
setAutoFindFailedReason(""); setAutoFindFailedReason("");
wsc.connect(port); wsc.connect(port);