BlockStorage/repertory
1
0
Fork 0
Code Issues 7 Pull Requests Releases 6 Wiki Activity

file db unit tests and fixes

Browse Source
This commit is contained in:
Scott E. Graves 2024-12-18 08:43:42 -06:00
parent c944039759
commit 2df67abffb
1 changed files with 24 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
repertory/librepertory/src/db/impl/rdb_file_db.cpp
View File

@ -91,14 +91,22 @@ auto rdb_file_db::add_directory(const std::string &api_path,
const std::string &source_path) -> api_error {
REPERTORY_USES_FUNCTION_NAME();
return perform_action(
function_name, [&](rocksdb::Transaction *txn) -> rocksdb::Status {
auto res = remove_item(api_path, source_path, txn);
if (not res.ok() && not res.IsNotFound()) {
std::string existing_source_path;
auto res = get_directory_source_path(api_path, existing_source_path);
if (res != api_error::success && res != api_error::directory_not_found) {
return res;
}
res = txn->Put(directory_family_, api_path, source_path);
return perform_action(
function_name, [&](rocksdb::Transaction *txn) -> rocksdb::Status {
if (not existing_source_path.empty()) {
auto res = remove_item(api_path, existing_source_path, txn);
if (not res.ok() && not res.IsNotFound()) {
return res;
}
}
auto res = txn->Put(directory_family_, api_path, source_path);
if (not res.ok()) {
return res;
}
@ -111,12 +119,20 @@ auto rdb_file_db::add_or_update_file(const i_file_db::file_data &data)
-> api_error {
REPERTORY_USES_FUNCTION_NAME();
std::string existing_source_path;
auto res = get_file_source_path(data.api_path, existing_source_path);
if (res != api_error::success && res != api_error::item_not_found) {
return res;
}
return perform_action(
function_name, [&](rocksdb::Transaction *txn) -> rocksdb::Status {
auto res = remove_item(data.api_path, data.source_path, txn);
if (not existing_source_path.empty()) {
auto res = remove_item(data.api_path, existing_source_path, txn);
if (not res.ok() && not res.IsNotFound()) {
return res;
}
}
json json_data = {
{"file_size", data.file_size},
@ -124,7 +140,7 @@ auto rdb_file_db::add_or_update_file(const i_file_db::file_data &data)
{"source_path", data.source_path},
};
res = txn->Put(file_family_, data.api_path, json_data.dump());
auto res = txn->Put(file_family_, data.api_path, json_data.dump());
if (not res.ok()) {
return res;
}