Update goss.py

Add timeout parameter for connection

Author: craig8

gridappsd-python-lib/gridappsd/goss.py

@@ -70,6 +70,7 @@ class GRIDAPPSD_ENV_ENUM(Enum):
     GRIDAPPSD_ADDRESS = "GRIDAPPSD_ADDRESS"
     GRIDAPPSD_PORT = "GRIDAPPSD_PORT"
     GRIDAPPSD_PASS = "GRIDAPPSD_PASSWORD"
+    GRIDAPPSD_HEARTBEAT = "GRIDAPPSD_HEARTBEAT"
 
 
 class TimeoutError(Exception):
@@ -119,6 +120,8 @@ def __init__(self,
 
         if not self.__user__ or not self.__pass__:
             raise ValueError("Invalid username/password specified.")
+
+        self._heartbeat = int(os.environ.get(GRIDAPPSD_ENV_ENUM.GRIDAPPSD_HEARTBEAT.value, 4000))
         self._conn = None
         self._ids = set()
         self._topic_set = set()
@@ -306,7 +309,7 @@ def _make_connection(self):
                 # send request to token topic
                 tokenTopic = "/topic/pnnl.goss.token.topic"
 
-                tmpConn = Connection([(self.stomp_address, self.stomp_port)])
+                tmpConn = Connection([(self.stomp_address, self.stomp_port)], heartbeats=(self._heartbeat, self._heartbeat))
                 if self._override_thread_fc is not None:
                     tmpConn.transport.override_threading(self._override_thread_fc)
                 tmpConn.connect(self.__user__, self.__pass__, wait=True)
@@ -349,7 +352,7 @@ def on_disconnect(self, header, message):
                     sleep(1)
                     iter += 1
 
-            self._conn = Connection([(self.stomp_address, self.stomp_port)])
+            self._conn = Connection([(self.stomp_address, self.stomp_port)], heartbeats=(self._heartbeat, self._heartbeat))
             if self._override_thread_fc is not None:
                 self._conn.transport.override_threading(self._override_thread_fc)
             try: