fixup! Initialize every connection of http client with session settings

cr8/clients.py

@@ -315,7 +315,7 @@ def _append_sql(host):
 
 
 class HttpClient:
-    def __init__(self, hosts, conn_pool_limit=25, session_settings=None):
+    def __init__(self, hosts, conn_pool_limit=25):
         self.hosts = hosts
         self.urls = itertools.cycle(list(map(_append_sql, hosts)))
         self._connector_params = {
@@ -324,7 +324,6 @@ def __init__(self, hosts, conn_pool_limit=25, session_settings=None):
         }
         self.__session = None
         self.is_cratedb = True
-        self.session_settings = session_settings or {}
 
     @property
     async def _session(self):
@@ -386,4 +385,4 @@ def client(hosts, session_settings=None, concurrency=25):
         if not asyncpg:
             raise ValueError('Cannot use "asyncpg" scheme if asyncpg is not available')
         return AsyncpgClient(hosts, pool_size=concurrency, session_settings=session_settings)
-    return HttpClient(_to_http_hosts(hosts), conn_pool_limit=concurrency, session_settings=session_settings)
+    return HttpClient(_to_http_hosts(hosts), conn_pool_limit=concurrency)

cr8/engine.py

@@ -74,7 +74,7 @@ def __init__(self, hosts, concurrency, sample_mode, session_settings=None):
         self.sampler = get_sampler(sample_mode)
         self.session_settings = session_settings
 
-    def init(self):
+    def set_session_settings(self):
         if isinstance(self.client, HttpClient):
             f = self.client.execute
             for setting, value in self.session_settings.items():

cr8/run_spec.py

@@ -205,7 +205,7 @@ def run_queries(self, queries: Iterable[dict], meta=None, session_settings=None)
                  f'   {mode_desc}: {duration or iterations}')
             )
             with Runner(self.benchmark_hosts, concurrency, self.sample_mode, session_settings) as runner:
-                runner.init()
+                runner.set_session_settings()
                 if warmup > 0:
                     runner.warmup(stmt, warmup, concurrency, args)
                 timed_stats = runner.run(