File size: 2,246 Bytes
babeaf6 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 |
/**
* src/hooks/useWebsocket.js
* Connect web socket. Send message to sever.
*
* created by Lynchee on 7/16/23
*/
import { useRef, useCallback } from 'react';
const useWebsocket = (token, onOpen, onMessage, selectedModel) => {
const socketRef = useRef(null);
// initialize web socket and connect to server.
const connectSocket = useCallback(() => {
if (!socketRef.current) {
const clientId = Math.floor(Math.random() * 1010000);
const ws_scheme = window.location.protocol === "https:" ? "wss" : "ws";
// const ws_path = ws_scheme + '://realchar.ai:8000/ws/' + clientId;
// Get the current host value
var currentHost = window.location.host;
// Split the host into IP and port number
var parts = currentHost.split(':');
// Extract the IP address and port number
var ipAddress = parts[0];
var currentPort = parts[1];
// Define the new port number
var newPort = '8000';
// Generate the new host value with the same IP but different port
var newHost = ipAddress + ':' + newPort;
const ws_path = ws_scheme + '://' + newHost + `/ws/${clientId}?llm_model=${selectedModel}&token=${token}`;
socketRef.current = new WebSocket(ws_path);
const socket = socketRef.current;
socket.binaryType = 'arraybuffer';
socket.onopen = onOpen;
socket.onmessage = onMessage;
socket.onerror = (error) => {
console.log(`WebSocket Error: ${error}`);
};
socket.onclose = (event) => {
console.log("Socket closed");
};
}
}, [onOpen, onMessage]);
// send message to server
const send = (data) => {
console.log("message sent to server");
if (socketRef.current && socketRef.current.readyState === WebSocket.OPEN) {
socketRef.current.send(data);
}
};
const closeSocket = () => {
socketRef.current.close();
socketRef.current = null;
}
return { socketRef, send, connectSocket, closeSocket };
};
export default useWebsocket;
|