File size: 1,005 Bytes
edb7d3e | 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 | import { useState, useEffect, useRef } from 'react'
import { createWebSocket } from '../utils/api.js'
/**
* Custom hook for WebSocket connection
* @param {string} jobId - The job ID to connect to
* @returns {{jobData: object|null, isConnected: boolean, error: string|null}}
*/
export function useWebSocket(jobId) {
const [jobData, setJobData] = useState(null)
const [isConnected, setIsConnected] = useState(false)
const [error, setError] = useState(null)
const wsRef = useRef(null)
useEffect(() => {
if (!jobId) return
setError(null)
setIsConnected(false)
wsRef.current = createWebSocket(
jobId,
(data) => {
setJobData(data)
setIsConnected(true)
setError(null)
},
() => {
setIsConnected(false)
// Don't set error on normal close
}
)
return () => {
if (wsRef.current) {
wsRef.current.close()
}
}
}, [jobId])
return {
jobData,
isConnected,
error,
}
}
|