useWebsocketWithCallback.js

function useWebsocketWithCallback(callback, ...args) {
  const {lastMessage} = useWebsocket(...args);
  const lastMessageRef = useRef();
  lastMessageRef.current = lastMessage;
  useEffect(() => {
    if (lastMessage !== lastMessageRef.current) {
      callback(lastMessage);
    }
  }, [lastMessage]);
}