Add copy constructor to ColumnFamilyHandleImpl (#353)

* Add copy constructor for ColumnFamilyHandleImpl Signed-off-by: Yang Zhang <[email protected]>
tikv · Jan 3, 2024 · dcf2f8d · dcf2f8d
1 parent 938c016
commit dcf2f8d
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/db/column_family.cc b/db/column_family.cc
@@ -50,6 +50,14 @@ ColumnFamilyHandleImpl::ColumnFamilyHandleImpl(
   }
 }
 
+ColumnFamilyHandleImpl::ColumnFamilyHandleImpl(
+    const ColumnFamilyHandleImpl& other)
+    : cfd_(other.cfd_), db_(other.db_), mutex_(other.mutex_) {
+  if (cfd_ != nullptr) {
+    cfd_->Ref();
+  }
+}
+
 ColumnFamilyHandleImpl::~ColumnFamilyHandleImpl() {
   if (cfd_ != nullptr) {
 #ifndef ROCKSDB_LITE

diff --git a/db/column_family.h b/db/column_family.h
@@ -162,6 +162,7 @@ class ColumnFamilyHandleImpl : public ColumnFamilyHandle {
   // create while holding the mutex
   ColumnFamilyHandleImpl(
       ColumnFamilyData* cfd, DBImpl* db, InstrumentedMutex* mutex);
+  ColumnFamilyHandleImpl(const ColumnFamilyHandleImpl& other);
   // destroy without mutex
   virtual ~ColumnFamilyHandleImpl();
   virtual ColumnFamilyData* cfd() const { return cfd_; }