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() +