support balance data reset plan (#342)

vesoft-inc · Dec 24, 2020 · 76def26 · 76def26
1 parent 63c6aa3
commit 76def26
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 2 deletions.
diff --git a/src/common/clients/meta/MetaClient.cpp b/src/common/clients/meta/MetaClient.cpp
@@ -2469,14 +2469,18 @@ MetaClient::getUserRoles(std::string account) {
 
 
 folly::Future<StatusOr<int64_t>> MetaClient::balance(std::vector<HostAddr> hostDel,
-                                                     bool isStop) {
+                                                     bool isStop,
+                                                     bool isReset) {
     cpp2::BalanceReq req;
     if (!hostDel.empty()) {
         req.set_host_del(std::move(hostDel));
     }
     if (isStop) {
         req.set_stop(isStop);
     }
+    if (isReset) {
+        req.set_reset(isReset);
+    }
 
     folly::Promise<StatusOr<int64_t>> promise;
     auto future = promise.getFuture();

diff --git a/src/common/clients/meta/MetaClient.h b/src/common/clients/meta/MetaClient.h
@@ -350,7 +350,7 @@ class MetaClient {
 
     // Operations for admin
     folly::Future<StatusOr<int64_t>>
-    balance(std::vector<HostAddr> hostDel, bool isStop = false);
+    balance(std::vector<HostAddr> hostDel, bool isStop, bool isReset);
 
     folly::Future<StatusOr<std::vector<cpp2::BalanceTask>>>
     showBalance(int64_t balanceId);

diff --git a/src/common/interface/meta.thrift b/src/common/interface/meta.thrift
@@ -55,6 +55,7 @@ enum ErrorCode {
     E_NO_RUNNING_BALANCE_PLAN   = -36,
     E_NO_VALID_HOST             = -37,
     E_CORRUPTTED_BALANCE_PLAN   = -38,
+    E_NO_INVALID_BALANCE_PLAN   = -39,
 
     // Authentication Failure
     E_INVALID_PASSWORD          = -41,
@@ -755,6 +756,7 @@ struct BalanceReq {
     2: optional i64                     id,
     3: optional list<common.HostAddr>   host_del,
     4: optional bool                    stop,
+    5: optional bool                    reset,
 }
 
 enum TaskResult {