feat: add close function for Client (#8)

* feat: add close function for Client

* style: format python code (pep8)

* Update client.py

Co-authored-by: github-actions <41898282+github-actions[bot]@users.noreply.github.com>

Author: hezean

demo.py

@@ -12,21 +12,20 @@
                        sync_mode='pooling',
                        refresh_interval=3)
 
-    client = fp.Client('server-b8b7c58417680d9e76b1b8454326f357296b5003',
-                       # Server Side SDK Key for your project and environment
-                       config)
-
-    # create one user
-    # key is for percentage rollout, normally use userId as key
-    user = fp.User('00001', {'userId': '00001'})
-
-    # Toggle you want to use
-    TOGGLE_KEY = 'feature_toggle02'
-
-    # get toggle result for this user
-    is_open = client.value(TOGGLE_KEY, user, default=False)
-    print('feature for this user is: ' + str(is_open))
-
-    is_open_detail = client.value_detail(TOGGLE_KEY, user, default=False)
-    print('detail: ' + str(is_open_detail.reason))
-    print('rule index: ' + str(is_open_detail.rule_index))
+    # Server Side SDK Key for your project and environment
+    SDK_KEY = 'server-b8b7c58417680d9e76b1b8454326f357296b5003'
+    with fp.Client(SDK_KEY, config) as client:
+        # create one user
+        # key is for percentage rollout, normally use userId as key
+        user = fp.User('00001', {'userId': '00001'})
+
+        # Toggle you want to use
+        TOGGLE_KEY = 'feature_toggle02'
+
+        # get toggle result for this user
+        is_open = client.value(TOGGLE_KEY, user, default=False)
+        print('feature for this user is: ' + str(is_open))
+
+        is_open_detail = client.value_detail(TOGGLE_KEY, user, default=False)
+        print('detail: ' + str(is_open_detail.reason))
+        print('rule index: ' + str(is_open_detail.rule_index))

featureprobe/client.py

@@ -12,6 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+import logging
 import time
 from typing import Any
 
@@ -24,17 +25,42 @@
 
 
 class Client:
+    __logger = logging.getLogger('FeatureProbe')
+
     def __init__(self, sdk_key: str, config: Config = Config()):
         if empty_str(sdk_key):
             raise ValueError('sdk key must not be blank')
         context = Context(sdk_key, config)
         self._event_processor = config.event_processor_creator(context)
         self._data_repo = config.data_repository_creator(context)
-        config.synchronizer_creator(context, self._data_repo).sync()
+        self._synchronizer = config.synchronizer_creator(
+            context, self._data_repo)
+        self._synchronizer.sync()
+
+    def __enter__(self):
+        return self
+
+    def __exit__(self, exc_type, exc_val, exc_tb):
+        """Alias for :func:`~featureprobe.Client.close`
+
+        Usage::
+
+          >>> import featureprobe as fp
+          >>> with fp.Client('key_000') as client:
+          >>>     ...
+          >>> # client will be closed here
+        """
+        self.close()
 
     def flush(self):
         self._event_processor.flush()
 
+    def close(self):
+        Client.__logger.info('Closing FeatureProbe Client')
+        self._synchronizer.close()
+        self.flush()
+        self._data_repo.close()
+
     def value(self, toggle_key: str, user: User, default) -> Any:
         toggle = self._data_repo.get_toggle(toggle_key)
         segments = self._data_repo.get_all_segment()