-
-
Notifications
You must be signed in to change notification settings - Fork 241
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Refactor websocket code to
$oh namespace
Signed-off-by: Florian Hotze <[email protected]>
- Loading branch information
1 parent
b611805
commit 89e59d9
Showing
3 changed files
with
124 additions
and
53 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,107 @@ | ||
| import { getAccessToken } from './auth' | ||
|
|
||
| const HEARTBEAT_MESSAGE = `{ | ||
| "type": "WebSocketEvent", | ||
| "topic": "openhab/websocket/heartbeat", | ||
| "payload": "PING", | ||
| "source": "WebSocketTestInstance" | ||
| }` | ||
|
|
||
| const openWSConnections = [] | ||
|
|
||
| function newWSConnection (path, messageCallback, readyCallback, errorCallback, heartbeatCallback, heartbeatInterval) { | ||
| // Create a new WebSocket connection | ||
| const socket = new WebSocket(path + `?access_token=${getAccessToken()}`) | ||
| socket.path = path | ||
|
|
||
| // Handle WebSocket connection opened | ||
| socket.onopen = (event) => { | ||
| socket.setKeepalive(heartbeatInterval) | ||
| if (readyCallback) { | ||
| readyCallback(event) | ||
| } | ||
| } | ||
|
|
||
| // Handle WebSocket message received | ||
| socket.onmessage = (event) => { | ||
| let evt = event.data | ||
| try { | ||
| evt = JSON.parse(event.data) | ||
| } catch (e) { | ||
| console.error('Error while parsing message', e) | ||
| } | ||
| messageCallback(evt) | ||
| } | ||
|
|
||
| // Handle WebSocket error | ||
| socket.onerror = (event) => { | ||
| console.error('WebSocket error', event) | ||
| if (errorCallback) { | ||
| errorCallback(event) | ||
| } | ||
| } | ||
|
|
||
| // WebSocket keep alive | ||
| socket.setKeepalive = (seconds = 5) => { | ||
| console.debug('Setting keepalive interval seconds', seconds) | ||
| socket.clearKeepalive() | ||
| socket.keepaliveTimer = setTimeout(() => { | ||
| console.warn('WebSocket timeout error') | ||
| if (heartbeatCallback) { | ||
| heartbeatCallback() | ||
| } else { | ||
| socket.send(HEARTBEAT_MESSAGE) | ||
| } | ||
| }, (seconds + 2) * 1000) | ||
| } | ||
|
|
||
| socket.clearKeepalive = () => { | ||
| if (socket.keepaliveTimer) clearTimeout(socket.keepaliveTimer) | ||
| delete socket.keepaliveTimer | ||
| } | ||
|
|
||
| // Add the new WebSocket connection to the list | ||
| openWSConnections.push(socket) | ||
| console.debug(`new WS connection: ${socket.path}, ${openWSConnections.length} open connections`) | ||
| console.debug(openWSConnections) | ||
|
|
||
| return socket | ||
| } | ||
|
|
||
| export default { | ||
| /** | ||
| * Connect to the websocket at the given path. | ||
| * | ||
| * @param {string} path path to connect to, e.g. `/ws` | ||
| * @param {fn} messageCallback | ||
| * @param {fn} [readyCallback=null] | ||
| * @param {fn} [errorCallback=null] | ||
| * @param {fn} [heartbeatCallback=null] heartbeat callback to use instead of the default PING/PONG | ||
| * @param {number} [heartbeatInterval=5] heartbeat interval in seconds | ||
| * @return {WebSocket} | ||
| */ | ||
| connect (path, messageCallback, readyCallback = null, errorCallback = null, heartbeatCallback = null, heartbeatInterval = 5) { | ||
| return newWSConnection(path, messageCallback, readyCallback, errorCallback, heartbeatCallback, heartbeatInterval) | ||
| }, | ||
| /** | ||
| * Close the given websocket connection. | ||
| * | ||
| * @param {WebSocket} socket | ||
| * @param {fn} [callback=null] callback to execute on connection close | ||
| */ | ||
| close (socket, callback = null) { | ||
| if (!socket) return | ||
| if (openWSConnections.indexOf(socket) >= 0) { | ||
| openWSConnections.splice(openWSConnections.indexOf(socket), 1) | ||
| } | ||
| console.debug(`WS connection closed: ${socket.path}, ${openWSConnections.length} open connections`) | ||
| console.debug(openWSConnections) | ||
| socket.onclose = (event) => { | ||
| if (callback) { | ||
| callback(event) | ||
| } | ||
| } | ||
| socket.close() | ||
| socket.clearKeepalive() | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters