rgw-admin: restructure replicalog commands

Signed-off-by: Yehuda Sadeh <yehuda@inktank.com>

1 files changed, 117 insertions, 117 deletions

diff --git a/src/rgw/rgw_admin.cc b/src/rgw/rgw_admin.cc
index 52f4af54f99..c163a69f426 100644
--- a/src/rgw/rgw_admin.cc
+++ b/src/rgw/rgw_admin.cc
@@ -100,12 +100,8 @@ void _usage()
   cerr << "  opstate set                set state on an entry (use client_id, op_id, object)\n";
   cerr << "  opstate renew              renew state on an entry (use client_id, op_id, object)\n";
   cerr << "  opstate rm                 remove entry (use client_id, op_id, object)\n";
-  cerr << "  replicamdlog get           get the replica metadata log\n";
-  cerr << "  replicamdlog delete        delete the replica metadata log\n";
-  cerr << "  replicadatalog get         get the replica data log\n";
-  cerr << "  replicadatalog delete      delete the replica data log\n";
-  cerr << "  replicabucketlog get       get the replica bucket log\n";
-  cerr << "  replicabucketlog delete    delete the replica bucket log\n";
+  cerr << "  replicalog get             get replica metadata log entry\n";
+  cerr << "  replicalog delete          delete replica metadata log entry\n";
   cerr << "options:\n";
   cerr << "   --uid=<id>                user id\n";
   cerr << "   --subuser=<name>          subuser name\n";
@@ -140,6 +136,8 @@ void _usage()
   cerr << "   --show-log-sum=<flag>     enable/disable dump of log summation on log show\n";
   cerr << "   --skip-zero-entries       log show only dumps entries that don't have zero value\n";
   cerr << "                             in one of the numeric field\n";
+  cerr << "   --replica-log-type        replica log type (metadata, data, bucket), required for\n";
+  cerr << "                             replica log operations\n";
   cerr << "   --categories=<list>       comma separated list of categories, used in usage show\n";
   cerr << "   --caps=<caps>             list of caps (e.g., \"usage=read, write; user=read\"\n";
   cerr << "   --yes-i-really-mean-it    required for certain operations\n";
@@ -222,14 +220,8 @@ enum {
   OPT_OPSTATE_SET,
   OPT_OPSTATE_RENEW,
   OPT_OPSTATE_RM,
-  OPT_REPLICAMDLOG_GET,
-  OPT_REPLICAMDLOG_CREATE,
-  OPT_REPLICAMDLOG_DELETE,
-  OPT_REPLICADATALOG_GET,
-  OPT_REPLICADATALOG_CREATE,
-  OPT_REPLICADATALOG_DELETE,
-  OPT_REPLICABUCKETLOG_GET,
-  OPT_REPLICABUCKETLOG_DELETE
+  OPT_REPLICALOG_GET,
+  OPT_REPLICALOG_DELETE,
 };
 
 static int get_cmd(const char *cmd, const char *prev_cmd, bool *need_more)
@@ -260,9 +252,7 @@ static int get_cmd(const char *cmd, const char *prev_cmd, bool *need_more)
       strcmp(cmd, "bilog") == 0 ||
       strcmp(cmd, "datalog") == 0 ||
       strcmp(cmd, "opstate") == 0 ||
-      strcmp(cmd, "replicamdlog") == 0 ||
-      strcmp(cmd, "replicadatalog") == 0 ||
-      strcmp(cmd, "replicabucketlog") == 0) {
+      strcmp(cmd, "replicalog") == 0) {
     *need_more = true;
     return 0;
   }
@@ -421,30 +411,34 @@ static int get_cmd(const char *cmd, const char *prev_cmd, bool *need_more)
       return OPT_OPSTATE_RENEW;
     if (strcmp(cmd, "rm") == 0)
       return OPT_OPSTATE_RM;
-  } else if (strcmp(prev_cmd, "replicamdlog") == 0) {
+  } else if (strcmp(prev_cmd, "replicalog") == 0) {
     if (strcmp(cmd, "get") == 0)
-      return OPT_REPLICAMDLOG_GET;
-    if (strcmp(cmd, "create") == 0)
-          return OPT_REPLICAMDLOG_CREATE;
-    if (strcmp(cmd, "delete") == 0)
-          return OPT_REPLICAMDLOG_DELETE;
-  } else if (strcmp(prev_cmd, "replicadatalog") == 0) {
-    if (strcmp(cmd, "get") == 0)
-      return OPT_REPLICADATALOG_GET;
-    if (strcmp(cmd, "create") == 0)
-          return OPT_REPLICADATALOG_CREATE;
-    if (strcmp(cmd, "delete") == 0)
-          return OPT_REPLICADATALOG_DELETE;
-  } else if (strcmp(prev_cmd, "replicabucketlog") == 0) {
-    if (strcmp(cmd, "get") == 0)
-      return OPT_REPLICABUCKETLOG_GET;
+      return OPT_REPLICALOG_GET;
     if (strcmp(cmd, "delete") == 0)
-      return OPT_REPLICABUCKETLOG_DELETE;
+      return OPT_REPLICALOG_DELETE;
   }
 
   return -EINVAL;
 }
 
+enum ReplicaLogType {
+  ReplicaLog_Invalid = 0,
+  ReplicaLog_Metadata,
+  ReplicaLog_Data,
+  ReplicaLog_Bucket,
+};
+
+ReplicaLogType get_replicalog_type(const string& name) {
+  if (name == "md" || name == "meta" || name == "metadata")
+    return ReplicaLog_Metadata;
+  if (name == "data")
+    return ReplicaLog_Data;
+  if (name == "bucket")
+    return ReplicaLog_Bucket;
+
+  return ReplicaLog_Invalid;
+}
+
 string escape_str(string& src, char c)
 {
   int pos = 0;
@@ -709,6 +703,8 @@ int main(int argc, char **argv)
   string client_id;
   string op_id;
   string state_str;
+  string replica_log_type_str;
+  ReplicaLogType replica_log_type = ReplicaLog_Invalid;
 
   std::string val;
   std::ostringstream errs;
@@ -834,6 +830,13 @@ int main(int argc, char **argv)
       start_marker = val;
     } else if (ceph_argparse_witharg(args, i, &val, "--end-marker", (char*)NULL)) {
       end_marker = val;
+    } else if (ceph_argparse_witharg(args, i, &val, "--replica-log-type", (char*)NULL)) {
+      replica_log_type_str = val;
+      replica_log_type = get_replicalog_type(replica_log_type_str);
+      if (replica_log_type == ReplicaLog_Invalid) {
+        cerr << "ERROR: invalid replica log type" << std::endl;
+        return EINVAL;
+      }
     } else {
       ++i;
     }
@@ -2093,103 +2096,100 @@ next:
     }
   }
 
-  if (opt_cmd == OPT_REPLICAMDLOG_GET) {
-    if (!specified_shard_id) {
-      cerr << "ERROR: shard-id must be specified for get operation" << std::endl;
-      return EINVAL;
-    }
-
-    RGWReplicaObjectLogger logger(store, pool_name, META_REPLICA_LOG_OBJ_PREFIX);
-    string pos_marker;
-    utime_t time_marker;
-    list<cls_replica_log_progress_marker> markers;
-    int ret = logger.get_bounds(shard_id, pos_marker, time_marker, markers);
-    if (ret < 0)
-      return -ret;
-    encode_json("markers", markers, formatter);
-  }
-
-  if (opt_cmd == OPT_REPLICAMDLOG_DELETE) {
-    if (!specified_shard_id) {
-      cerr << "ERROR: shard-id must be specified for delete operation" << std::endl;
-      return EINVAL;
-    }
-    if (!specified_daemon_id) {
-      cerr << "ERROR: daemon-id must be specified for delete operation" << std::endl;
+  if (opt_cmd == OPT_REPLICALOG_GET || opt_cmd == OPT_REPLICALOG_DELETE) {
+    if (replica_log_type_str.empty()) {
+      cerr << "ERROR: need to specify --replica-log-type=<metadata | data | bucket>" << std::endl;
       return EINVAL;
     }
-    RGWReplicaObjectLogger logger(store, pool_name, META_REPLICA_LOG_OBJ_PREFIX);
-    int ret = logger.delete_bound(shard_id, daemon_id);
-    if (ret < 0)
-      return -ret;
   }
 
-  if (opt_cmd == OPT_REPLICADATALOG_GET) {
-    if (!specified_shard_id) {
-      cerr << "ERROR: shard-id must be specified for get operation" << std::endl;
-      return EINVAL;
-    }
-    RGWReplicaObjectLogger logger(store, pool_name, DATA_REPLICA_LOG_OBJ_PREFIX);
+  if (opt_cmd == OPT_REPLICALOG_GET) {
     string pos_marker;
     utime_t time_marker;
     list<cls_replica_log_progress_marker> markers;
-    int ret = logger.get_bounds(shard_id, pos_marker, time_marker, markers);
-    if (ret < 0)
-      return -ret;
-    encode_json("markers", markers, formatter);
-  }
+    if (replica_log_type == ReplicaLog_Metadata) {
+      if (!specified_shard_id) {
+        cerr << "ERROR: shard-id must be specified for get operation" << std::endl;
+        return EINVAL;
+      }
 
-  if (opt_cmd == OPT_REPLICADATALOG_DELETE) {
-    if (!specified_shard_id) {
-      cerr << "ERROR: shard-id must be specified for delete operation" << std::endl;
-      return EINVAL;
-    }
-    if (!specified_daemon_id) {
-      cerr << "ERROR: daemon-id must be specified for delete operation" << std::endl;
-      return EINVAL;
-    }
-    RGWReplicaObjectLogger logger(store, pool_name, DATA_REPLICA_LOG_OBJ_PREFIX);
-    int ret = logger.delete_bound(shard_id, daemon_id);
-    if (ret < 0)
-      return -ret;
-  }
+      RGWReplicaObjectLogger logger(store, pool_name, META_REPLICA_LOG_OBJ_PREFIX);
+      int ret = logger.get_bounds(shard_id, pos_marker, time_marker, markers);
+      if (ret < 0)
+        return -ret;
+    } else if (replica_log_type == ReplicaLog_Data) {
+      if (!specified_shard_id) {
+        cerr << "ERROR: shard-id must be specified for get operation" << std::endl;
+        return EINVAL;
+      }
+      RGWReplicaObjectLogger logger(store, pool_name, DATA_REPLICA_LOG_OBJ_PREFIX);
+      int ret = logger.get_bounds(shard_id, pos_marker, time_marker, markers);
+      if (ret < 0)
+        return -ret;
+    } else if (replica_log_type == ReplicaLog_Bucket) {
+      if (bucket_name.empty()) {
+        cerr << "ERROR: bucket not specified" << std::endl;
+        return -EINVAL;
+      }
+      int ret = init_bucket(bucket_name, bucket);
+      if (ret < 0) {
+        cerr << "ERROR: could not init bucket: " << cpp_strerror(-ret) << std::endl;
+        return -ret;
+      }
 
-  if (opt_cmd == OPT_REPLICABUCKETLOG_GET) {
-    if (bucket_name.empty()) {
-      cerr << "ERROR: bucket not specified" << std::endl;
-      return -EINVAL;
-    }
-    int ret = init_bucket(bucket_name, bucket);
-    if (ret < 0) {
-      cerr << "ERROR: could not init bucket: " << cpp_strerror(-ret) << std::endl;
-      return -ret;
+      RGWReplicaBucketLogger logger(store);
+      ret = logger.get_bounds(bucket, pos_marker, time_marker, markers);
+      if (ret < 0)
+        return -ret;
+    } else { // shouldn't get here
+      assert(0);
     }
-
-    RGWReplicaBucketLogger logger(store);
-    string pos_marker;
-    utime_t time_marker;
-    list<cls_replica_log_progress_marker> markers;
-    ret = logger.get_bounds(bucket, pos_marker, time_marker, markers);
-    if (ret < 0)
-      return -ret;
     encode_json("markers", markers, formatter);
   }
 
-  if (opt_cmd == OPT_REPLICABUCKETLOG_DELETE) {
-    if (bucket_name.empty()) {
-      cerr << "ERROR: bucket not specified" << std::endl;
-      return -EINVAL;
-    }
-    int ret = init_bucket(bucket_name, bucket);
-    if (ret < 0) {
-      cerr << "ERROR: could not init bucket: " << cpp_strerror(-ret) << std::endl;
-      return -ret;
-    }
+  if (opt_cmd == OPT_REPLICALOG_DELETE) {
+    if (replica_log_type == ReplicaLog_Metadata) {
+      if (!specified_shard_id) {
+        cerr << "ERROR: shard-id must be specified for delete operation" << std::endl;
+        return EINVAL;
+      }
+      if (!specified_daemon_id) {
+        cerr << "ERROR: daemon-id must be specified for delete operation" << std::endl;
+        return EINVAL;
+      }
+      RGWReplicaObjectLogger logger(store, pool_name, META_REPLICA_LOG_OBJ_PREFIX);
+      int ret = logger.delete_bound(shard_id, daemon_id);
+      if (ret < 0)
+        return -ret;
+    } else if (replica_log_type == ReplicaLog_Data) {
+      if (!specified_shard_id) {
+        cerr << "ERROR: shard-id must be specified for delete operation" << std::endl;
+        return EINVAL;
+      }
+      if (!specified_daemon_id) {
+        cerr << "ERROR: daemon-id must be specified for delete operation" << std::endl;
+        return EINVAL;
+      }
+      RGWReplicaObjectLogger logger(store, pool_name, DATA_REPLICA_LOG_OBJ_PREFIX);
+      int ret = logger.delete_bound(shard_id, daemon_id);
+      if (ret < 0)
+        return -ret;
+    } else if (replica_log_type == ReplicaLog_Bucket) {
+      if (bucket_name.empty()) {
+        cerr << "ERROR: bucket not specified" << std::endl;
+        return -EINVAL;
+      }
+      int ret = init_bucket(bucket_name, bucket);
+      if (ret < 0) {
+        cerr << "ERROR: could not init bucket: " << cpp_strerror(-ret) << std::endl;
+        return -ret;
+      }
 
-    RGWReplicaBucketLogger logger(store);
-    ret = logger.delete_bound(bucket, daemon_id);
-    if (ret < 0)
-      return -ret;
+      RGWReplicaBucketLogger logger(store);
+      ret = logger.delete_bound(bucket, daemon_id);
+      if (ret < 0)
+        return -ret;
+    }
   }
   return 0;
 }