LBRYCommunity/lbry-desktop
1
0
Fork 0
Code Issues 596 Pull requests 6 Projects Releases 238 Packages Wiki Activity

Reconnect when comment websocket loses connection

Browse source
This commit is contained in:
Sean Yesmunt 2021-04-14 00:02:32 -04:00
parent 989126c603
commit 942c85c9f1
1 changed files with 7 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
ui/redux/actions/websocket.js
View file

@ -6,9 +6,10 @@ const NOTIFICATION_WS_URL = 'wss://api.lbry.com/subscribe?auth_token=';
const COMMENT_WS_URL = `wss://comments.lbry.com/api/v2/live-chat/subscribe?subscription_id=`; const COMMENT_WS_URL = `wss://comments.lbry.com/api/v2/live-chat/subscribe?subscription_id=`;
let sockets = {}; let sockets = {};
let closingSockets = {};
let retryCount = 0; let retryCount = 0;
export const doSocketConnect = (url, retryOnDisconnect, cb) => { export const doSocketConnect = (url, cb) => {
function connectToSocket() { function connectToSocket() {
if (sockets[url] !== undefined && sockets[url] !== null) { if (sockets[url] !== undefined && sockets[url] !== null) {
sockets[url].close(); sockets[url].close();
@ -35,11 +36,11 @@ export const doSocketConnect = (url, retryOnDisconnect, cb) => {
sockets[url].onclose = () => { sockets[url].onclose = () => {
console.log('\n Disconnected from WS\n\n'); // eslint-disable-line console.log('\n Disconnected from WS\n\n'); // eslint-disable-line
if (retryOnDisconnect) { if (!closingSockets[url]) {
retryCount += 1; retryCount += 1;
connectToSocket(); connectToSocket();
} else { } else {
sockets[url] = null; closingSockets[url] = null;
} }
}; };
}, timeToWait); }, timeToWait);
@ -50,6 +51,7 @@ export const doSocketConnect = (url, retryOnDisconnect, cb) => {
export const doSocketDisconnect = (url) => (dispatch) => { export const doSocketDisconnect = (url) => (dispatch) => {
if (sockets[url] !== undefined && sockets[url] !== null) { if (sockets[url] !== undefined && sockets[url] !== null) {
closingSockets[url] = true;
sockets[url].close(); sockets[url].close();
sockets[url] = null; sockets[url] = null;
@ -68,7 +70,7 @@ export const doNotificationSocketConnect = (enableNotifications) => (dispatch) =
const url = `${NOTIFICATION_WS_URL}${authToken}`; const url = `${NOTIFICATION_WS_URL}${authToken}`;
doSocketConnect(url, true, (data) => { doSocketConnect(url, (data) => {
switch (data.type) { switch (data.type) {
case 'pending_notifications': case 'pending_notifications':
if (enableNotifications) { if (enableNotifications) {
@ -88,7 +90,7 @@ export const doNotificationSocketConnect = (enableNotifications) => (dispatch) =
export const doCommentSocketConnect = (uri, claimId) => (dispatch) => { export const doCommentSocketConnect = (uri, claimId) => (dispatch) => {
const url = `${COMMENT_WS_URL}${claimId}`; const url = `${COMMENT_WS_URL}${claimId}`;
doSocketConnect(url, false, (response) => { doSocketConnect(url, (response) => {
if (response.type === 'delta') { if (response.type === 'delta') {
const newComment = response.data.comment; const newComment = response.data.comment;
dispatch({ dispatch({