viewerCounter and websocket JS modules

This commit is contained in:
Alexandre Iooss 2020-10-20 21:29:41 +02:00
parent 01efba3e3f
commit 11231ceb84
No known key found for this signature in database
GPG Key ID: 6C79278F3FCDCC02
4 changed files with 108 additions and 0 deletions

6
.gitignore vendored
View File

@ -17,3 +17,9 @@ pkged.go
# Profiler and test files
*.prof
*.test
# Javascript tools
.eslintrc.js
node_modules
package.json
package-lock.json

18
web/static/js/main.js Normal file
View File

@ -0,0 +1,18 @@
import { GSWebSocket } from "./modules/websocket.js";
import { ViewerCounter } from "./modules/viewerCounter.js";
// Create WebSocket
const s = new GSWebSocket();
s.open(() => {
// FIXME open callback
}, () => {
// FIXME close callback
});
// Create viewer counter
const streamName = "demo"; // FIXME
const viewerCounter = new ViewerCounter(
document.getElementById("connected-people"),
streamName,
);
viewerCounter.regularUpdate(1000); // FIXME

View File

@ -0,0 +1,29 @@
/**
* ViewerCounter show the number of active viewers
*/
export class ViewerCounter {
/**
* @param {HTMLElement} element
* @param {String} streamName
*/
constructor(element, streamName) {
this.element = element;
this.url = "/_stats/" + streamName;
}
/**
* Regulary update counter
*
* @param {Number} updatePeriod
*/
regularUpdate(updatePeriod) {
setInterval(this._refreshViewersCounter, updatePeriod);
}
_refreshViewersCounter() {
fetch(this.url)
.then(response => response.json())
.then((data) => this.element.innerText = data.ConnectedViewers)
.catch(console.log);
}
}

View File

@ -0,0 +1,55 @@
/**
* GsWebSocket to do Ghostream signalling
*/
export class GsWebSocket {
constructor() {
const protocol = (window.location.protocol === "https:") ? "wss://" : "ws://";
this.url = protocol + window.location.host + "/_ws/";
}
_open() {
this.socket = new WebSocket(this.url);
}
/**
* Open websocket.
*
* @param {Function} openCallback Function called when connection is established.
* @param {Function} closeCallback Function called when connection is lost.
*/
open(openCallback, closeCallback) {
this._open();
this.socket.addEventListener("open", (event) => {
console.log("WebSocket opened");
openCallback(event);
});
this.socket.addEventListener("close", (event) => {
console.log("WebSocket closed, retrying connection in 1s...");
setTimeout(this._open, 1000);
closeCallback(event);
});
this.socket.addEventListener("error", (event) => {
console.log("WebSocket errored, retrying connection in 1s...");
setTimeout(this._open, 1000);
closeCallback(event);
});
}
/**
* Exchange WebRTC session description with server.
*
* @param {string} data JSON formated data
* @param {Function} receiveCallback Function called when data is received
*/
exchangeDescription(data, receiveCallback) {
if (this.socket.readyState !== 1) {
console.log("WebSocket not ready to send data");
return;
}
this.socket.send(data);
this.socket.addEventListener("message", (event) => {
console.log("Message from server ", event.data);
receiveCallback(event);
});
}
}