Add get_container_name func (#88)

* Add get_container_name func * Add get_container_name func --------- Co-authored-by: 杨赫然 <[email protected]>
haiwen · Oct 30, 2024 · 67595fa · 67595fa
1 parent e971c15
commit 67595fa
Show file tree

Hide file tree

Showing 6 changed files with 22 additions and 0 deletions.
diff --git a/seafobj/backends/alioss.py b/seafobj/backends/alioss.py
@@ -8,6 +8,7 @@ class SeafObjStoreOSS(AbstractObjStore):
     def __init__(self, compressed, oss_conf, crypto=None, cache=None):
         AbstractObjStore.__init__(self, compressed, crypto, cache)
         self.oss_client = SeafOSSClient(oss_conf)
+        self.bucket_name = oss_conf.bucket_name
 
     def read_obj_raw(self, repo_id, version, obj_id):
         real_obj_id = '%s/%s' % (repo_id, obj_id)
@@ -44,3 +45,6 @@ def stat_raw(self, repo_id, obj_id):
         key = '%s/%s' % (repo_id, obj_id)
 
         return self.oss_client.stat_obj(key)
+
+    def get_container_name(self):
+        return self.bucket_name
diff --git a/seafobj/backends/base.py b/seafobj/backends/base.py
@@ -83,3 +83,6 @@ def stat(self, repo_id, verison, obj_id):
 
     def stat_raw(self, repo_id, obj_id):
         raise NotImplementedError
+
+    def get_container_name(self):
+        raise NotImplementedError
diff --git a/seafobj/backends/ceph.py b/seafobj/backends/ceph.py
@@ -72,6 +72,7 @@ class SeafObjStoreCeph(AbstractObjStore):
     def __init__(self, compressed, ceph_conf, crypto=None, cache=None):
         AbstractObjStore.__init__(self, compressed, crypto, cache)
         self.ceph_client = SeafCephClient(ceph_conf)
+        self.pool_name = ceph_conf.pool_name
 
     def read_obj_raw(self, repo_id, version, obj_id):
         data = self.ceph_client.read_object_content(repo_id, obj_id)
@@ -129,3 +130,6 @@ def stat_raw(self, repo_id, obj_id):
             raise
         finally:
             self.ceph_client.ioctx_pool.return_ioctx(ioctx)
+
+    def get_container_name(self):
+        return self.pool_name
diff --git a/seafobj/backends/filesystem.py b/seafobj/backends/filesystem.py
@@ -98,3 +98,6 @@ def stat_raw(self, repo_id, obj_id):
         if not stat_info:
             return -1
         return stat_info.st_size
+
+    def get_container_name(self):
+        return self.obj_dir
diff --git a/seafobj/backends/s3.py b/seafobj/backends/s3.py
@@ -9,6 +9,7 @@ class SeafObjStoreS3(AbstractObjStore):
     def __init__(self, compressed, s3_conf, crypto=None, cache=None):
         AbstractObjStore.__init__(self, compressed, crypto, cache)
         self.s3_client = SeafS3Client(s3_conf)
+        self.bucket_name = s3_conf.bucket_name
 
     def read_obj_raw(self, repo_id, version, obj_id):
         real_obj_id = '%s/%s' % (repo_id, obj_id)
@@ -41,3 +42,6 @@ def remove_obj(self, repo_id, obj_id):
     def stat_raw(self, repo_id, obj_id):
         s3_path = '%s/%s' % (repo_id, obj_id)
         return self.s3_client.stat_obj(s3_path)
+
+    def get_container_name(self):
+        return self.bucket_name
diff --git a/seafobj/backends/swift.py b/seafobj/backends/swift.py
@@ -291,6 +291,7 @@ class SeafObjStoreSwift(AbstractObjStore):
     def __init__(self, compressed, swift_conf, crypto=None, cache=None):
         AbstractObjStore.__init__(self, compressed, crypto, cache)
         self.swift_client = SeafSwiftClient(swift_conf)
+        self.container = swift_conf.container
 
     def read_obj_raw(self, repo_id, version, obj_id):
         real_obj_id = '%s/%s' % (repo_id, obj_id)
@@ -322,3 +323,6 @@ def stat_raw(self, repo_id, obj_id):
         key = '%s/%s' % (repo_id, obj_id)
 
         return self.swift_client.stat_obj(key)
+
+    def get_container_name(self):
+        return self.container