feat: Cache Bigtable client (feast-dev#3602)

ysk24ok · zerafachris · commit 0d6a06020db2 · 2024-03-05T20:14:50.000+01:00
* Cache Bigtable client

Signed-off-by: Yusuke Nishioka &lt;yusuke.nishioka.0713@gmail.com&gt;

* Run format

Signed-off-by: Yusuke Nishioka &lt;yusuke.nishioka.0713@gmail.com&gt;

---------

Signed-off-by: Yusuke Nishioka &lt;yusuke.nishioka.0713@gmail.com&gt;
Signed-off-by: zerafachris PERSONAL &lt;zerafachris@gmail.com&gt;
diff --git a/sdk/python/feast/infra/online_stores/bigtable.py b/sdk/python/feast/infra/online_stores/bigtable.py
@@ -335,4 +335,8 @@ def teardown(
     def _get_client(
         self, online_config: BigtableOnlineStoreConfig, admin: bool = False
     ):
-        return bigtable.Client(project=online_config.project_id, admin=admin)
+        if self._client is None:
+            self._client = bigtable.Client(
+                project=online_config.project_id, admin=admin
+            )
+        return self._client
