add operator= for index (#487)

Signed-off-by: yusheng.ma <yusheng.ma@zilliz.com>
zilliztech · Apr 2, 2024 · 3c46f4c · 3c46f4c
1 parent 2ec2622
commit 3c46f4c
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/include/knowhere/index.h b/include/knowhere/index.h
@@ -43,6 +43,20 @@ class Index {
         node = idx.node;
     }
 
+    Index<T1>&
+    operator=(const Index<T1>& idx) {
+        if (&idx == this) {
+            return *this;
+        }
+        if (idx.node == nullptr) {
+            node = nullptr;
+            return *this;
+        }
+        idx.node->IncRef();
+        node = idx.node;
+        return *this;
+    }
+
     Index(Index<T1>&& idx) {
         if (idx.node == nullptr) {
             node = nullptr;