Skip to content
This repository was archived by the owner on May 13, 2024. It is now read-only.

Commit 4ffe50c

Browse files
committed
chore: updated the WS connection
1 parent 4d3cf5e commit 4ffe50c

File tree

1 file changed

+13
-3
lines changed

1 file changed

+13
-3
lines changed

src/hooks/useWs/index.tsx

Lines changed: 13 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@ import {
66
} from '@site/src/configs/websocket/types';
77
import { useCallback, useState } from 'react';
88

9-
const useWS = <T extends TSocketEndpointNames>(name: T) => {
9+
const useWS = <T extends TSocketEndpointNames>(name?: T) => {
1010
const [is_loading, setIsLoading] = useState(false);
1111
const [error, setError] = useState<unknown>();
1212
const [data, setData] = useState<TSocketResponseData<T>>();
@@ -19,10 +19,20 @@ const useWS = <T extends TSocketEndpointNames>(name: T) => {
1919
}, []);
2020

2121
const send = useCallback(
22-
async (data?: Parameters<typeof apiManager.augmentedSend<T>>[1]) => {
22+
async (data?: Parameters<typeof apiManager.augmentedSend<T>>[]) => {
23+
let payload: any = data;
24+
console.log();
25+
26+
if (name) {
27+
payload = { ...payload, name: { [name]: 1 } };
28+
} else {
29+
payload = { ...payload };
30+
}
31+
2332
setIsLoading(true);
33+
2434
try {
25-
const response = await apiManager.augmentedSend(name, data);
35+
const response = await apiManager.augmentedSend(payload);
2636
const key = response['msg_type'] ?? name;
2737
setData(response[key] as TSocketResponseData<T>);
2838
setFullResponse(response);

0 commit comments

Comments
 (0)