feat: heartbeat check if connection offline

- WIP price message

Signed-off-by: pakintada@gmail.com <Pakin>

src/lib/core/stores/websocketStore.ts

@@ -9,8 +9,11 @@ import { addNotification } from './noti';
 import { permission } from './permissions';
 
 let socket: WebSocket | null = null;
+let reconnectTimeout: any;
 const ENABLE_WS_DEBUG: boolean = false;
 
+export const socketConnectionOfflineCount = writable<number>(0);
+export const socketAlreadySendHeartbeat = writable<number>(0);
 export const socketStore = writable<WebSocket | null>(null);
 
 export function connectToWebsocket() {
@@ -28,6 +31,8 @@ export function connectToWebsocket() {
 				addNotification('INFO:Connected!');
 
 				if (socket) {
+					clearTimeout(reconnectTimeout);
+
 					// send auth message
 
 					let auth_data = get(authStore);
@@ -46,34 +51,43 @@ export function connectToWebsocket() {
 					});
 				}
 
-				// recover messages on connect, flushing
-				// while (get(msgQueue).length) {
-				// 	let queue = get(msgQueue);
-				// 	let current = queue.shift();
-				// 	if (current && socket) {
-				// 		socket.send(current);
-				// 		// set next
-				// 		msgQueue.set(queue);
-				// 	}
-				// }
-
 				// heartbeat 10s
 				setInterval(() => {
-					if (socket) {
+					if (get(socketAlreadySendHeartbeat) > 0) {
+						let heartbeat_may_offline_count = get(socketConnectionOfflineCount);
+
+						socketConnectionOfflineCount.set(heartbeat_may_offline_count + 1);
+					}
+
+					if (get(socketConnectionOfflineCount) > 4) {
+						// counting offline 5 times then try reconnect
+						socketConnectionOfflineCount.set(0);
+						socketAlreadySendHeartbeat.set(0);
+
+						connectToWebsocket();
+
+						return;
+					}
+
+					if (socket != null) {
 						sendMessage({
 							type: 'heartbeat',
 							payload: {}
 						});
+
+						let heartbeat_count = get(socketAlreadySendHeartbeat);
+						socketAlreadySendHeartbeat.set(heartbeat_count + 1);
+					} else {
+						// may send reconnect
+						console.log('check on socket health', socket);
 					}
 				}, 10000);
 
 				if (auth.currentUser && socket == null) {
 					console.log('try reconnect websocket ...');
 					// retry again
-					setTimeout(() => {
-						if (socket == null) {
-							connectToWebsocket();
-						}
+					reconnectTimeout = setTimeout(() => {
+						connectToWebsocket();
 					}, 5000);
 				}
 			});
@@ -89,7 +103,7 @@ export function connectToWebsocket() {
 				if (auth.currentUser && !socket) {
 					console.log('try reconnect websocket ...');
 					// retry again
-					setTimeout(() => connectToWebsocket(), 30000);
+					reconnectTimeout = setTimeout(() => connectToWebsocket(), 5000);
 				}
 			});