From d133832e7370e144eeedcee7cff707dba13d1e99 Mon Sep 17 00:00:00 2001 From: lash Date: Wed, 20 Apr 2022 15:20:07 +0000 Subject: [PATCH] Add closers for persistent store backend --- shep/store/file.py | 4 ++++ shep/store/redis.py | 4 ++++ shep/store/rocksdb.py | 5 +++++ 3 files changed, 13 insertions(+) diff --git a/shep/store/file.py b/shep/store/file.py index 840c930..9fdbe61 100644 --- a/shep/store/file.py +++ b/shep/store/file.py @@ -154,3 +154,7 @@ class SimpleFileStoreFactory: if re.match(re_processedname, v): r.append(v) return r + + + def close(self): + pass diff --git a/shep/store/redis.py b/shep/store/redis.py index c818d91..ebf96d1 100644 --- a/shep/store/redis.py +++ b/shep/store/redis.py @@ -96,3 +96,7 @@ class RedisStoreFactory: def add(self, k): k = str(k) return RedisStore(k, self.redis, binary=self.__binary) + + + def close(self): + self.redis.close() diff --git a/shep/store/rocksdb.py b/shep/store/rocksdb.py index d056b42..750c5f4 100644 --- a/shep/store/rocksdb.py +++ b/shep/store/rocksdb.py @@ -119,3 +119,8 @@ class RocksDbStoreFactory: def add(self, k): k = str(k) return RocksDbStore(k, self.db, binary=self.__binary) + + + def close(self): + self.db.close() +