feat:add pika_migrate_unstable

Author: brother-jin

tools/include/acl.h

@@ -0,0 +1,15 @@
+// Copyright (c) 2015-present, Qihoo, Inc.  All rights reserved.
+// This source code is licensed under the BSD-style license found in the
+// LICENSE file in the root directory of this source tree. An additional grant
+// of patent rights can be found in the PATENTS file in the same directory.
+
+#ifndef INCLUDE_BUILD_VERSION_H_
+#define INCLUDE_BUILD_VERSION_H_
+
+// this variable tells us about the git revision
+extern const char* pika_build_git_sha;
+
+// Date on which the code was compiled:
+extern const char* pika_build_compile_date;
+
+#endif  // INCLUDE_BUILD_VERSION_H_

tools/include/build_version.h

@@ -0,0 +1,66 @@
+#ifndef MIGRATOR_THREAD_H_
+#define MIGRATOR_THREAD_H_
+
+#include <iostream>
+#include <mutex>
+
+#include "storage/storage.h"
+#include "net/include/redis_cli.h"
+
+#include "include/redis_sender.h"
+
+class MigratorThread : public net::Thread {
+ public:
+  MigratorThread(std::shared_ptr<storage::Storage> storage_, std::vector<std::shared_ptr<RedisSender>>  *senders, int type, int thread_num) :
+      storage_(storage_),
+      should_exit_(false),
+      senders_(senders),
+      type_(type),
+      thread_num_(thread_num),
+      thread_index_(0),
+      num_(0) {
+  }
+
+  virtual ~ MigratorThread();
+
+  int64_t num() {
+    std::lock_guard<std::mutex> l(num_mutex_);
+    return num_;
+  }
+
+  void Stop() {
+	  should_exit_ = true;
+  }
+
+ private:
+  void PlusNum() {
+    std::lock_guard<std::mutex> l(num_mutex_);
+    ++num_;
+  }
+
+  void DispatchKey(const std::string &command, const std::string& key = "");
+
+  void MigrateDB();
+  void MigrateStringsDB();
+  void MigrateListsDB();
+  void MigrateHashesDB();
+  void MigrateSetsDB();
+  void MigrateZsetsDB();
+  void MigrateStreamsDB();
+
+  virtual void *ThreadMain();
+
+ private:
+  std::shared_ptr<storage::Storage> storage_;
+  bool should_exit_;
+
+  std::vector<std::shared_ptr<RedisSender>> *senders_;
+  int type_;
+  int thread_num_;
+  int thread_index_;
+
+  int64_t num_;
+  std::mutex num_mutex_;
+};
+
+#endif

tools/include/migrator_thread.h

@@ -0,0 +1,48 @@
+// Copyright (c) 2015-present, Qihoo, Inc.  All rights reserved.
+// This source code is licensed under the BSD-style license found in the
+// LICENSE file in the root directory of this source tree. An additional grant
+// of patent rights can be found in the PATENTS file in the same directory.
+
+// pika ACL command
+#ifndef PIKA_ACL_CMD_H
+#define PIKA_ACL_CMD_H
+
+#include "include/pika_command.h"
+#include "include/pika_server.h"
+
+extern PikaServer* g_pika_server;
+
+class PikaAclCmd : public Cmd {
+ public:
+  PikaAclCmd(const std::string& name, int arity, uint32_t flag)
+      : Cmd(name, arity, flag, static_cast<uint32_t>(AclCategory::ADMIN)) {
+    subCmdName_ = {"cat", "deluser", "dryrun",  "genpass", "getuser", "list", "load",
+                   "log", "save",    "setuser", "users",   "whoami",  "help"};
+  }
+  void Do() override;
+  void Split(const HintKeys& hint_keys) override{};
+  void Merge() override{};
+  Cmd* Clone() override { return new PikaAclCmd(*this); }
+
+ private:
+  void DoInitial() override;
+  void Clear() override {}
+
+  void Cat();
+  void DelUser();
+  void DryRun();
+  void GenPass();
+  void GetUser();
+  void List();
+  void Load();
+  void Log();
+  void Save();
+  void SetUser();
+  void Users();
+  void WhoAmI();
+  void Help();
+
+  std::string subCmd_;
+};
+
+#endif  // PIKA_ACL_CMD_H