63 lines
1.6 KiB
TypeScript
63 lines
1.6 KiB
TypeScript
import { NOTIFICATION_API_WEBSOCKET } from 'utils/constants'
|
|
|
|
export class NotificationsWebSocket {
|
|
ws: WebSocket | null = null
|
|
token: string
|
|
publicKey: string
|
|
pingInterval: NodeJS.Timer | null
|
|
retryCount = 0
|
|
maxRetries = 2
|
|
|
|
constructor(token: string, publicKey: string) {
|
|
this.token = token
|
|
this.publicKey = publicKey
|
|
this.pingInterval = null
|
|
}
|
|
|
|
connect() {
|
|
const wsUrl = new URL(NOTIFICATION_API_WEBSOCKET)
|
|
wsUrl.searchParams.append('authorization', this.token)
|
|
wsUrl.searchParams.append('publickey', this.publicKey)
|
|
this.ws = new WebSocket(wsUrl)
|
|
|
|
this.ws.addEventListener('open', () => {
|
|
console.log('Notifications WebSocket opened')
|
|
// Send a ping message to the server every 10 seconds
|
|
const interval = setInterval(() => {
|
|
if (this.ws?.readyState === this.ws?.OPEN) {
|
|
this.ws?.send('ping')
|
|
}
|
|
}, 30000)
|
|
this.pingInterval = interval
|
|
})
|
|
|
|
this.ws.addEventListener('close', (event: CloseEvent) => {
|
|
console.log('Notifications WebSocket closed')
|
|
this.handleClearSocketInterval()
|
|
//1000 close form clinet
|
|
//1008 unauthorized
|
|
if (
|
|
event.code !== 1000 &&
|
|
event.code !== 1008 &&
|
|
this.retryCount < this.maxRetries
|
|
) {
|
|
this.retryCount++
|
|
setTimeout(() => {
|
|
this.connect()
|
|
}, 5000)
|
|
}
|
|
})
|
|
|
|
this.ws.addEventListener('error', (event) => {
|
|
console.log('WebSocket error:', event)
|
|
})
|
|
return this
|
|
}
|
|
handleClearSocketInterval() {
|
|
if (this.pingInterval) {
|
|
clearInterval(this.pingInterval)
|
|
this.pingInterval = null
|
|
}
|
|
}
|
|
}
|