Add connect_timeout parameter

Author: kesmit13

singlestoredb/config.py

@@ -181,6 +181,13 @@
     environ='SINGLESTOREDB_BUFFERED',
 )
 
+register_option(
+    'connect_timeout', 'int', check_int, 10,
+    'The timeout for connecting to the database in seconds. '
+    '(default: 10, min: 1, max: 31536000)',
+    environ='SINGLESTOREDB_CONNECT_TIMEOUT',
+)
+
 #
 # Query results options
 #

singlestoredb/connection.py

@@ -1277,6 +1277,7 @@ def connect(
     program_name: Optional[str] = None,
     conn_attrs: Optional[Dict[str, str]] = None,
     multi_statements: Optional[bool] = None,
+    connect_timeout: Optional[int] = None,
 ) -> Connection:
     """
     Return a SingleStoreDB connection.
@@ -1338,6 +1339,11 @@ def connect(
     conn_attrs: dict, optional
         Additional connection attributes for telemetry. Example:
         {'program_version': "1.0.2", "_connector_name": "dbt connector"}
+    multi_statements: bool, optional
+        Should multiple statements be allowed within a single query?
+    connect_timeout: int, optional
+        The timeout for connecting to the database in seconds.
+        (default: 10, min: 1, max: 31536000)
 
     Examples
     --------

singlestoredb/http/connection.py

@@ -809,6 +809,12 @@ def __init__(self, **kwargs: Any):
             if not ssl_verify_cert:
                 self._sess.verify = False
 
+        if kwargs.get('multi_statements', False):
+            raise self.InterfaceError(
+                0, 'The Data API does not allow multiple '
+                'statements within a query',
+            )
+
         version = kwargs.get('version', 'v2')
         self.driver = kwargs.get('driver', 'https')
 
@@ -846,6 +852,8 @@ def _post(self, path: str, *args: Any, **kwargs: Any) -> requests.Response:
         """
         if self._sess is None:
             raise InterfaceError(errno=2048, msg='Connection is closed.')
+        if 'timeout' not in kwargs:
+            kwargs['timeout'] = get_option('connect_timeout')
         return self._sess.post(urljoin(self._url, path), *args, **kwargs)
 
     def close(self) -> None:

singlestoredb/tests/test_connection.py

@@ -1370,6 +1370,12 @@ def test_multi_statements(self):
                     self.assertEqual([(2,)], list(cur))
                     self.assertIsNone(cur.nextset())
 
+    def test_connect_timeout(self):
+        with s2.connect(database=type(self).dbname, connect_timeout=8) as conn:
+            with conn.cursor() as cur:
+                cur.execute('SELECT 1')
+                self.assertEqual([(1,)], list(cur))
+
     def test_show_accessors(self):
         out = self.conn.show.columns('data')
         assert out.columns == [