fix bug (#4675)

Co-authored-by: Sophie <84560950+Sophie-Xie@users.noreply.github.com>
vesoft-inc · Sep 26, 2022 · cccc014 · cccc014
1 parent a229752
commit cccc014
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 3 deletions.
diff --git a/src/storage/admin/StatsTask.cpp b/src/storage/admin/StatsTask.cpp
@@ -142,6 +142,8 @@ nebula::cpp2::ErrorCode StatsTask::genSubTask(GraphSpaceID spaceId,
     edgetypeEdges[edge.first] = 0;
   }
 
+  VertexID lastVertexId = "";
+
   // Only stats valid vertex data, no multi version
   // For example
   // Vid  tagId
@@ -166,8 +168,15 @@ nebula::cpp2::ErrorCode StatsTask::genSubTask(GraphSpaceID spaceId,
       tagIter->next();
       continue;
     }
-    spaceVertices++;
-    tagsVertices[tagId] += 1;
+
+    if (vId == lastVertexId) {
+      tagsVertices[tagId] += 1;
+    } else {
+      tagsVertices[tagId] += 1;
+      spaceVertices++;
+      lastVertexId = vId;
+    }
+
     tagIter->next();
     sleepIfScannedSomeRecord(++countToSleep);
   }

diff --git a/src/storage/mutate/AddVerticesProcessor.cpp b/src/storage/mutate/AddVerticesProcessor.cpp
@@ -81,6 +81,9 @@ void AddVerticesProcessor::doProcess(const cpp2::AddVerticesRequest& req) {
         code = nebula::cpp2::ErrorCode::E_INVALID_VID;
         break;
       }
+      if (FLAGS_use_vertex_key) {
+        data.emplace_back(NebulaKeyUtils::vertexKey(spaceVidLen_, partId, vid), "");
+      }
       for (auto& newTag : newTags) {
         auto tagId = newTag.get_tag_id();
         VLOG(3) << "PartitionID: " << partId << ", VertexID: " << vid << ", TagID: " << tagId;
@@ -158,7 +161,9 @@ void AddVerticesProcessor::doProcessWithIndex(const cpp2::AddVerticesRequest& re
         code = nebula::cpp2::ErrorCode::E_INVALID_VID;
         break;
       }
-
+      if (FLAGS_use_vertex_key) {
+        verticeData.emplace_back(NebulaKeyUtils::vertexKey(spaceVidLen_, partId, vid));
+      }
       for (const auto& newTag : newTags) {
         auto tagId = newTag.get_tag_id();
         VLOG(3) << "PartitionID: " << partId << ", VertexID: " << vid << ", TagID: " << tagId;