diff --git a/lib/websockets.js b/lib/websockets.js
index 7e79d6c..61b2f69 100644
--- a/lib/websockets.js
+++ b/lib/websockets.js
@@ -48,7 +48,7 @@ getSocketEndpoint = async function(type, baseURL, environment, sign) {
   }
   eventHanlder = function
 */
-Sockets.initSocket = async function(params, eventHandler) {
+Sockets.initSocket = async function(params, eventHandler, onCloseEventHandler) {
   try {
     if ( !params.sign ) params.sign = false;
     if ( !params.endpoint ) params.endpoint = false;
@@ -72,6 +72,9 @@ Sockets.initSocket = async function(params, eventHandler) {
     ws.on('close', () => {
       clearInterval(Sockets.ws[topic].heartbeat)
       console.log(topic + ' websocket closed...')
+      if (onCloseEventHandler) {
+        onCloseEventHandler()
+      }
     })
   } catch (err) {
     console.log(err)