Merge pull request #471 from splitio/async-client-factory

Added async to factory

Author: chillaq

splitio/__init__.py

@@ -1,3 +1,3 @@
-from splitio.client.factory import get_factory
+from splitio.client.factory import get_factory, get_factory_async
 from splitio.client.key import Key
 from splitio.version import __version__

splitio/api/client.py

@@ -3,6 +3,8 @@
 import requests
 import urllib
 import abc
+import logging
+import json
 
 from splitio.optional.loaders import aiohttp
 from splitio.util.time import get_current_epoch_time_ms
@@ -12,6 +14,8 @@
 AUTH_URL = 'https://auth.split.io/api'
 TELEMETRY_URL = 'https://telemetry.split.io/api'
 
+_LOGGER = logging.getLogger(__name__)
+
 
 HttpResponse = namedtuple('HttpResponse', ['status_code', 'body', 'headers'])
 
@@ -242,13 +246,20 @@ async def get(self, server, path, apikey, query=None, extra_headers=None):  # py
             headers.update(extra_headers)
         start = get_current_epoch_time_ms()
         try:
+            url = _build_url(server, path, self._urls)
+            _LOGGER.debug("GET request: %s", url)
+            _LOGGER.debug("query params: %s", query)
+            _LOGGER.debug("headers: %s", headers)
             async with self._session.get(
-                _build_url(server, path, self._urls),
+                url,
                 params=query,
                 headers=headers,
                 timeout=self._timeout
             ) as response:
                 body = await response.text()
+                _LOGGER.debug("Response:")
+                _LOGGER.debug(response)
+                _LOGGER.debug(body)
                 await self._record_telemetry(response.status, get_current_epoch_time_ms() - start)
                 return HttpResponse(response.status, body, response.headers)
         except aiohttp.ClientError as exc:  # pylint: disable=broad-except
@@ -277,6 +288,12 @@ async def post(self, server, path, apikey, body, query=None, extra_headers=None)
             headers.update(extra_headers)
         start = get_current_epoch_time_ms()
         try:
+            headers['Accept-Encoding'] = 'gzip'
+            _LOGGER.debug("POST request: %s", _build_url(server, path, self._urls))
+            _LOGGER.debug("query params: %s", query)
+            _LOGGER.debug("headers: %s", headers)
+            _LOGGER.debug("payload: ")
+            _LOGGER.debug(str(json.dumps(body)).encode('utf-8'))
             async with self._session.post(
                 _build_url(server, path, self._urls),
                 params=query,
@@ -285,6 +302,9 @@ async def post(self, server, path, apikey, body, query=None, extra_headers=None)
                 timeout=self._timeout
             ) as response:
                 body = await response.text()
+                _LOGGER.debug("Response:")
+                _LOGGER.debug(response)
+                _LOGGER.debug(body)
                 await self._record_telemetry(response.status, get_current_epoch_time_ms() - start)
                 return HttpResponse(response.status, body, response.headers)
         except aiohttp.ClientError as exc:  # pylint: disable=broad-except
@@ -305,3 +325,7 @@ async def _record_telemetry(self, status_code, elapsed):
             await self._telemetry_runtime_producer.record_successful_sync(self._metric_name, get_current_epoch_time_ms())
             return
         await self._telemetry_runtime_producer.record_sync_error(self._metric_name, status_code)
+
+    async def close_session(self):
+        if not self._session.closed:
+            await self._session.close()

splitio/api/telemetry.py

@@ -59,7 +59,7 @@ def record_init(self, configs):
         try:
             response = self._client.post(
                 'telemetry',
-                '/v1/metrics/config',
+                'v1/metrics/config',
                 self._sdk_key,
                 body=configs,
                 extra_headers=self._metadata,
@@ -83,7 +83,7 @@ def record_stats(self, stats):
         try:
             response = self._client.post(
                 'telemetry',
-                '/v1/metrics/usage',
+                'v1/metrics/usage',
                 self._sdk_key,
                 body=stats,
                 extra_headers=self._metadata,
@@ -150,7 +150,7 @@ async def record_init(self, configs):
         try:
             response = await self._client.post(
                 'telemetry',
-                '/v1/metrics/config',
+                'v1/metrics/config',
                 self._sdk_key,
                 body=configs,
                 extra_headers=self._metadata,
@@ -174,7 +174,7 @@ async def record_stats(self, stats):
         try:
             response = await self._client.post(
                 'telemetry',
-                '/v1/metrics/usage',
+                'v1/metrics/usage',
                 self._sdk_key,
                 body=stats,
                 extra_headers=self._metadata,