github/kubeshark
1
0
Fork 0
mirror of https://github.com/kubeshark/kubeshark.git synced 2025-09-11 07:19:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fixing acceptance tests (#900)

Browse Source
This commit is contained in:
gadotroee
2022-03-21 14:21:08 +02:00
committed by GitHub
parent 5455220a3a
commit a4d0e250c9
16 changed files with 142 additions and 148 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
ui-common/src/hooks/useWS.tsx
View File

@@ -18,7 +18,9 @@ const useWS = (wsUrl: string) => {
const onMessage = (e) => { setMessage(e) }
const onError = (e) => setError(e)
const onOpen = () => { setisOpen(true) }
const onClose = () => setisOpen(false)
const onClose = () => {
setisOpen(false)
}
const openSocket = () => {
ws.current = new WebSocket(wsUrl)
@@ -36,13 +38,17 @@ const useWS = (wsUrl: string) => {
ws.current.removeEventListener("close", onClose)
}
const sendQuery = (query: string) => {
if (ws.current && (ws.current.readyState === WebSocketReadyState.OPEN)) {
ws.current.send(JSON.stringify({ "query": query, "enableFullEntries": false }));
}
const sendQueryWhenWsOpen = (query) => {
setTimeout(() => {
if (ws?.current?.readyState === WebSocket.OPEN) {
ws.current.send(JSON.stringify({"query": query, "enableFullEntries": false}));
} else {
sendQueryWhenWsOpen(query);
}
}, 500)
}
return { message, error, isOpen, openSocket, closeSocket, sendQuery }
return { message, error, isOpen, openSocket, closeSocket, sendQueryWhenWsOpen }
}
export default useWS
export default useWS