feat: string cmd setbit

src/base_cmd.h

@@ -34,6 +34,7 @@ const std::string kCmdNameStrlen = "strlen";
 const std::string kCmdNameSetex = "setex";
 const std::string kCmdNamePsetex = "psetex";
 const std::string kCmdNameSetnx = "setnx";
+const std::string kCmdNameSetBit = "setbit";
 const std::string kCmdNameGetBit = "getbit";
 
 // multi

src/cmd_kv.cc

@@ -499,4 +499,67 @@ void GetBitCmd::DoCmd(PClient* client) {
   return;
 }
 
+SetBitCmd::SetBitCmd(const std::string& name, int16_t arity)
+    : BaseCmd(name, arity, CmdFlagsWrite, AclCategoryWrite | AclCategoryString) {}
+
+bool SetBitCmd::DoInitial(PClient* client) {
+  client->SetKey(client->argv_[1]);
+  return true;
+}
+
+void SetBitCmd::DoCmd(PClient* client) {
+  PObject* value = nullptr;
+  PError err = PSTORE.GetValueByType(client->Key(), value, PType_string);
+  if (err == PError_notExist) {
+    value = PSTORE.SetValue(client->Key(), PObject::CreateString(""));
+    err = PError_ok;
+  }
+
+  if (err != PError_ok) {
+    client->AppendInteger(0);
+    return;
+  }
+
+  long offset = 0;
+  long on = 0;
+  if (!Strtol(client->argv_[2].c_str(), client->argv_[2].size(), &offset) ||
+      !Strtol(client->argv_[3].c_str(), client->argv_[3].size(), &on)) {
+    client->SetRes(CmdRes::kInvalidInt);
+    return;
+  }
+
+  if (offset < 0 || offset > kStringMaxBytes) {
+    client->AppendInteger(0);
+    return;
+  }
+
+  PString* pStringPtr = value->CastString();
+  if (!pStringPtr) {
+    client->AppendInteger(0);
+    return;
+  }
+
+  PString& newVal = *pStringPtr;
+
+  size_t bytes = offset / 8;
+  size_t bits = offset % 8;
+
+  if (bytes + 1 > newVal.size()) {
+    newVal.resize(bytes + 1, '\0');
+  }
+
+  const char oldByte = newVal[bytes];
+  char& byte = newVal[bytes];
+  if (on) {
+    byte |= (0x1 << bits);
+  } else {
+    byte &= ~(0x1 << bits);
+  }
+
+  value->Reset(new PString(newVal));
+  value->encoding = PEncode_raw;
+  client->AppendInteger((oldByte & (0x1 << bits)) ? 1 : 0);
+  return;
+}
+
 }  // namespace pikiwidb

src/cmd_kv.h

@@ -160,6 +160,17 @@ class IncrbyCmd : public BaseCmd {
   void DoCmd(PClient *client) override;
 };
 
+class SetBitCmd : public BaseCmd {
+ public:
+  SetBitCmd(const std::string &name, int16_t arity);
+
+ protected:
+  bool DoInitial(PClient *client) override;
+
+ private:
+  void DoCmd(PClient *client) override;
+};
+
 class GetBitCmd : public BaseCmd {
  public:
   GetBitCmd(const std::string &name, int16_t arity);

src/cmd_table_manager.cc

@@ -62,6 +62,8 @@ void CmdTableManager::InitCmdTable() {
   cmds_->insert(std::make_pair(kCmdNamePsetex, std::move(psetexPtr)));
   std::unique_ptr<BaseCmd> setnxPtr = std::make_unique<SetnxCmd>(kCmdNameSetnx, 3);
   cmds_->insert(std::make_pair(kCmdNameSetnx, std::move(setnxPtr)));
+  std::unique_ptr<BaseCmd> setbitPtr = std::make_unique<SetBitCmd>(kCmdNameSetBit, 4);
+  cmds_->insert(std::make_pair(kCmdNameSetBit, std::move(setbitPtr)));
   std::unique_ptr<BaseCmd> getbitPtr = std::make_unique<GetBitCmd>(kCmdNameGetBit, 3);
   cmds_->insert(std::make_pair(kCmdNameGetBit, std::move(getbitPtr)));
 }