Support more check about merge zone arguments (vesoft-inc#500)

Co-authored-by: yaphet <[email protected]>
yixinglu · Jan 13, 2022 · a7d5c29 · a7d5c29
1 parent 96f763b
commit a7d5c29
Show file tree

Hide file tree

Showing 3 changed files with 28 additions and 0 deletions.
diff --git a/src/meta/processors/zone/MergeZoneProcessor.cpp b/src/meta/processors/zone/MergeZoneProcessor.cpp
@@ -169,6 +169,17 @@ void MergeZoneProcessor::process(const cpp2::MergeZoneReq& req) {
   // Write the merged zone into meta service
   auto key = MetaKeyUtils::zoneKey(zoneName);
   std::vector<HostAddr> zoneHosts;
+  auto valueRet = doGet(key);
+  if (nebula::ok(valueRet)) {
+    auto it = std::find(zones.begin(), zones.end(), zoneName);
+    if (it == zones.end()) {
+      LOG(ERROR) << "The target zone should exist in merge zone list";
+      handleErrorCode(nebula::cpp2::ErrorCode::E_INVALID_PARM);
+      onFinished();
+      return;
+    }
+  }
+
   for (auto& zone : zones) {
     auto zoneKey = MetaKeyUtils::zoneKey(zone);
     auto zoneValueRet = doGet(std::move(zoneKey));

diff --git a/src/meta/test/MetaClientTest.cpp b/src/meta/test/MetaClientTest.cpp
@@ -2278,6 +2278,13 @@ TEST(MetaClientTest, MergeZoneTest) {
     ASSERT_EQ("default_zone_127.0.0.1_8988", zones[2].get_zone_name());
     ASSERT_EQ("default_zone_127.0.0.1_8989", zones[3].get_zone_name());
   }
+  {
+    auto result = client
+                      ->mergeZone({"default_zone_127.0.0.1_8986", "default_zone_127.0.0.1_8987"},
+                                  "default_zone_127.0.0.1_8988")
+                      .get();
+    EXPECT_FALSE(result.ok());
+  }
   {
     // Merge zones is empty
     auto result = client->mergeZone({}, "new_zone").get();

diff --git a/src/meta/test/ProcessorTest.cpp b/src/meta/test/ProcessorTest.cpp
@@ -3762,6 +3762,16 @@ TEST(ProcessorTest, MergeZoneTest) {
     ASSERT_EQ("default_zone_127.0.0.1_8988", zones[2].get_zone_name());
     ASSERT_EQ("default_zone_127.0.0.1_8989", zones[3].get_zone_name());
   }
+  {
+    cpp2::MergeZoneReq req;
+    req.zones_ref() = {"default_zone_127.0.0.1_8986", "default_zone_127.0.0.1_8987"};
+    req.zone_name_ref() = "default_zone_127.0.0.1_8988";
+    auto* processor = MergeZoneProcessor::instance(kv.get());
+    auto f = processor->getFuture();
+    processor->process(req);
+    auto resp = std::move(f).get();
+    ASSERT_EQ(nebula::cpp2::ErrorCode::E_INVALID_PARM, resp.get_code());
+  }
   {
     // Merge an empty zone list
     cpp2::MergeZoneReq req;