Support single delete in ldb (#9469)

Summary: Pull Request resolved: https://github.com/facebook/rocksdb/pull/9469

Reviewed By: riversand963

Differential Revision: D33953484

fbshipit-source-id: f4e84a2d9865957d744c7e84ff02ffbb0a62b0a8
This commit is contained in:
yaphet 2022-05-10 16:37:19 -07:00 committed by Facebook GitHub Bot
parent 0d1613aad6
commit 26768edb65
2 changed files with 111 additions and 49 deletions

View File

@ -214,6 +214,10 @@ LDBCommand* LDBCommand::SelectCommand(const ParsedParams& parsed_params) {
} else if (parsed_params.cmd == DeleteCommand::Name()) {
return new DeleteCommand(parsed_params.cmd_params, parsed_params.option_map,
parsed_params.flags);
} else if (parsed_params.cmd == SingleDeleteCommand::Name()) {
return new SingleDeleteCommand(parsed_params.cmd_params,
parsed_params.option_map,
parsed_params.flags);
} else if (parsed_params.cmd == DeleteRangeCommand::Name()) {
return new DeleteRangeCommand(parsed_params.cmd_params,
parsed_params.option_map,
@ -3063,6 +3067,42 @@ void DeleteCommand::DoCommand() {
}
}
SingleDeleteCommand::SingleDeleteCommand(
const std::vector<std::string>& params,
const std::map<std::string, std::string>& options,
const std::vector<std::string>& flags)
: LDBCommand(options, flags, false,
BuildCmdLineOptions({ARG_HEX, ARG_KEY_HEX, ARG_VALUE_HEX})) {
if (params.size() != 1) {
exec_state_ = LDBCommandExecuteResult::Failed(
"KEY must be specified for the single delete command");
} else {
key_ = params.at(0);
if (is_key_hex_) {
key_ = HexToString(key_);
}
}
}
void SingleDeleteCommand::Help(std::string& ret) {
ret.append(" ");
ret.append(SingleDeleteCommand::Name() + " <key>");
ret.append("\n");
}
void SingleDeleteCommand::DoCommand() {
if (!db_) {
assert(GetExecuteState().IsFailed());
return;
}
Status st = db_->SingleDelete(WriteOptions(), GetCfHandle(), key_);
if (st.ok()) {
fprintf(stdout, "OK\n");
} else {
exec_state_ = LDBCommandExecuteResult::Failed(st.ToString());
}
}
DeleteRangeCommand::DeleteRangeCommand(
const std::vector<std::string>& params,
const std::map<std::string, std::string>& options,

View File

@ -24,7 +24,7 @@ class CompactorCommand : public LDBCommand {
static void Help(std::string& ret);
virtual void DoCommand() override;
void DoCommand() override;
private:
bool null_from_;
@ -43,7 +43,7 @@ class DBFileDumperCommand : public LDBCommand {
static void Help(std::string& ret);
virtual void DoCommand() override;
void DoCommand() override;
private:
bool decode_blob_index_;
@ -61,7 +61,7 @@ class DBLiveFilesMetadataDumperCommand : public LDBCommand {
static void Help(std::string& ret);
virtual void DoCommand() override;
void DoCommand() override;
private:
bool sort_by_filename_;
@ -79,7 +79,7 @@ class DBDumperCommand : public LDBCommand {
static void Help(std::string& ret);
virtual void DoCommand() override;
void DoCommand() override;
private:
/**
@ -127,7 +127,7 @@ class InternalDumpCommand : public LDBCommand {
static void Help(std::string& ret);
virtual void DoCommand() override;
void DoCommand() override;
private:
bool has_from_;
@ -160,9 +160,10 @@ class DBLoaderCommand : public LDBCommand {
const std::vector<std::string>& flags);
static void Help(std::string& ret);
virtual void DoCommand() override;
virtual void OverrideBaseOptions() override;
void DoCommand() override;
void OverrideBaseOptions() override;
private:
bool disable_wal_;
@ -183,9 +184,10 @@ class ManifestDumpCommand : public LDBCommand {
const std::vector<std::string>& flags);
static void Help(std::string& ret);
virtual void DoCommand() override;
virtual bool NoDBOpen() override { return true; }
void DoCommand() override;
bool NoDBOpen() override { return true; }
private:
bool verbose_;
@ -263,9 +265,10 @@ class ListColumnFamiliesCommand : public LDBCommand {
const std::vector<std::string>& flags);
static void Help(std::string& ret);
virtual void DoCommand() override;
virtual bool NoDBOpen() override { return true; }
void DoCommand() override;
bool NoDBOpen() override { return true; }
};
class CreateColumnFamilyCommand : public LDBCommand {
@ -277,9 +280,10 @@ class CreateColumnFamilyCommand : public LDBCommand {
const std::vector<std::string>& flags);
static void Help(std::string& ret);
virtual void DoCommand() override;
virtual bool NoDBOpen() override { return false; }
void DoCommand() override;
bool NoDBOpen() override { return false; }
private:
std::string new_cf_name_;
@ -294,9 +298,10 @@ class DropColumnFamilyCommand : public LDBCommand {
const std::vector<std::string>& flags);
static void Help(std::string& ret);
virtual void DoCommand() override;
virtual bool NoDBOpen() override { return false; }
void DoCommand() override;
bool NoDBOpen() override { return false; }
private:
std::string cf_name_to_drop_;
@ -310,11 +315,11 @@ class ReduceDBLevelsCommand : public LDBCommand {
const std::map<std::string, std::string>& options,
const std::vector<std::string>& flags);
virtual void OverrideBaseCFOptions(ColumnFamilyOptions* cf_opts) override;
void OverrideBaseCFOptions(ColumnFamilyOptions* cf_opts) override;
virtual void DoCommand() override;
void DoCommand() override;
virtual bool NoDBOpen() override { return true; }
bool NoDBOpen() override { return true; }
static void Help(std::string& msg);
@ -342,9 +347,9 @@ class ChangeCompactionStyleCommand : public LDBCommand {
const std::map<std::string, std::string>& options,
const std::vector<std::string>& flags);
virtual void OverrideBaseCFOptions(ColumnFamilyOptions* cf_opts) override;
void OverrideBaseCFOptions(ColumnFamilyOptions* cf_opts) override;
virtual void DoCommand() override;
void DoCommand() override;
static void Help(std::string& msg);
@ -364,10 +369,11 @@ class WALDumperCommand : public LDBCommand {
const std::map<std::string, std::string>& options,
const std::vector<std::string>& flags);
virtual bool NoDBOpen() override { return true; }
bool NoDBOpen() override { return true; }
static void Help(std::string& ret);
virtual void DoCommand() override;
void DoCommand() override;
private:
bool print_header_;
@ -389,7 +395,7 @@ class GetCommand : public LDBCommand {
const std::map<std::string, std::string>& options,
const std::vector<std::string>& flags);
virtual void DoCommand() override;
void DoCommand() override;
static void Help(std::string& ret);
@ -405,7 +411,7 @@ class ApproxSizeCommand : public LDBCommand {
const std::map<std::string, std::string>& options,
const std::vector<std::string>& flags);
virtual void DoCommand() override;
void DoCommand() override;
static void Help(std::string& ret);
@ -422,11 +428,11 @@ class BatchPutCommand : public LDBCommand {
const std::map<std::string, std::string>& options,
const std::vector<std::string>& flags);
virtual void DoCommand() override;
void DoCommand() override;
static void Help(std::string& ret);
virtual void OverrideBaseOptions() override;
void OverrideBaseOptions() override;
private:
/**
@ -443,7 +449,7 @@ class ScanCommand : public LDBCommand {
const std::map<std::string, std::string>& options,
const std::vector<std::string>& flags);
virtual void DoCommand() override;
void DoCommand() override;
static void Help(std::string& ret);
@ -464,7 +470,23 @@ class DeleteCommand : public LDBCommand {
const std::map<std::string, std::string>& options,
const std::vector<std::string>& flags);
virtual void DoCommand() override;
void DoCommand() override;
static void Help(std::string& ret);
private:
std::string key_;
};
class SingleDeleteCommand : public LDBCommand {
public:
static std::string Name() { return "singledelete"; }
SingleDeleteCommand(const std::vector<std::string>& params,
const std::map<std::string, std::string>& options,
const std::vector<std::string>& flags);
void DoCommand() override;
static void Help(std::string& ret);
@ -480,7 +502,7 @@ class DeleteRangeCommand : public LDBCommand {
const std::map<std::string, std::string>& options,
const std::vector<std::string>& flags);
virtual void DoCommand() override;
void DoCommand() override;
static void Help(std::string& ret);
@ -497,11 +519,11 @@ class PutCommand : public LDBCommand {
const std::map<std::string, std::string>& options,
const std::vector<std::string>& flags);
virtual void DoCommand() override;
void DoCommand() override;
static void Help(std::string& ret);
virtual void OverrideBaseOptions() override;
void OverrideBaseOptions() override;
private:
std::string key_;
@ -522,7 +544,7 @@ class DBQuerierCommand : public LDBCommand {
static void Help(std::string& ret);
virtual void DoCommand() override;
void DoCommand() override;
private:
static const char* HELP_CMD;
@ -539,9 +561,9 @@ class CheckConsistencyCommand : public LDBCommand {
const std::map<std::string, std::string>& options,
const std::vector<std::string>& flags);
virtual void DoCommand() override;
void DoCommand() override;
virtual bool NoDBOpen() override { return true; }
bool NoDBOpen() override { return true; }
static void Help(std::string& ret);
};
@ -554,7 +576,7 @@ class CheckPointCommand : public LDBCommand {
const std::map<std::string, std::string>& options,
const std::vector<std::string>& flags);
virtual void DoCommand() override;
void DoCommand() override;
static void Help(std::string& ret);
@ -571,11 +593,11 @@ class RepairCommand : public LDBCommand {
const std::map<std::string, std::string>& options,
const std::vector<std::string>& flags);
virtual void DoCommand() override;
void DoCommand() override;
virtual bool NoDBOpen() override { return true; }
bool NoDBOpen() override { return true; }
virtual void OverrideBaseOptions() override;
void OverrideBaseOptions() override;
static void Help(std::string& ret);
@ -615,7 +637,7 @@ class BackupCommand : public BackupEngineCommand {
BackupCommand(const std::vector<std::string>& params,
const std::map<std::string, std::string>& options,
const std::vector<std::string>& flags);
virtual void DoCommand() override;
void DoCommand() override;
static void Help(std::string& ret);
};
@ -625,8 +647,8 @@ class RestoreCommand : public BackupEngineCommand {
RestoreCommand(const std::vector<std::string>& params,
const std::map<std::string, std::string>& options,
const std::vector<std::string>& flags);
virtual void DoCommand() override;
virtual bool NoDBOpen() override { return true; }
void DoCommand() override;
bool NoDBOpen() override { return true; }
static void Help(std::string& ret);
};
@ -638,11 +660,11 @@ class WriteExternalSstFilesCommand : public LDBCommand {
const std::map<std::string, std::string>& options,
const std::vector<std::string>& flags);
virtual void DoCommand() override;
void DoCommand() override;
virtual bool NoDBOpen() override { return false; }
bool NoDBOpen() override { return false; }
virtual void OverrideBaseOptions() override;
void OverrideBaseOptions() override;
static void Help(std::string& ret);
@ -658,11 +680,11 @@ class IngestExternalSstFilesCommand : public LDBCommand {
const std::map<std::string, std::string>& options,
const std::vector<std::string>& flags);
virtual void DoCommand() override;
void DoCommand() override;
virtual bool NoDBOpen() override { return false; }
bool NoDBOpen() override { return false; }
virtual void OverrideBaseOptions() override;
void OverrideBaseOptions() override;
static void Help(std::string& ret);
@ -710,9 +732,9 @@ class UnsafeRemoveSstFileCommand : public LDBCommand {
static void Help(std::string& ret);
virtual void DoCommand() override;
void DoCommand() override;
virtual bool NoDBOpen() override { return true; }
bool NoDBOpen() override { return true; }
private:
uint64_t sst_file_number_;