King_DuckZ/dindexer
1
0
Fork 0
mirror of https://github.com/KingDuckZ/dindexer.git synced 2025-07-02 14:04:22 +00:00
Code Issues Releases Wiki Activity

Implement tagging by regex.

Browse source
This commit is contained in:
King_DuckZ 2016-07-10 14:00:51 +01:00
parent 9c2f1d7403
commit af7443c48b
2 changed files with 33 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

1
src/backends/redis/backend_redis.cpp
View file

@ -128,6 +128,7 @@ namespace dindb {
}
void BackendRedis::tag_files (const std::vector<std::string>& parRegexes, const std::vector<boost::string_ref>& parTags, GroupIDType parSet) {
dindb::tag_files(m_redis, m_tag_if_in_set, parRegexes, parTags, parSet);
}
void BackendRedis::delete_tags (const std::vector<FileIDType>& parFiles, const std::vector<boost::string_ref>& parTags, GroupIDType parSet) {

32
src/backends/redis/tag.cpp
View file

@ -25,6 +25,18 @@
namespace dindb {
namespace {
std::vector<std::regex> compile_regexes (const std::vector<std::string>& parRegexes) {
std::vector<std::regex> retval;
retval.reserve(parRegexes.size());
for (const auto& str : parRegexes) {
retval.emplace_back(std::regex(
str,
std::regex_constants::optimize | std::regex_constants::nosubs | std::regex_constants::ECMAScript
));
}
assert(retval.size() == parRegexes.size());
return retval;
}
} //unnamed namespace
void tag_files (redis::Command& parRedis, redis::Script& parTagIfInSet, const std::vector<uint64_t>& parFiles, const std::vector<boost::string_ref>& parTags, GroupIDType parSet) {
@ -46,6 +58,26 @@ namespace dindb {
}
void tag_files (redis::Command& parRedis, redis::Script& parTagIfInSet, const std::vector<std::string>& parRegexes, const std::vector<boost::string_ref>& parTags, GroupIDType parSet) {
using dinhelp::lexical_cast;
auto batch = parRedis.make_batch();
const std::string set_key = (parSet != InvalidGroupID ? PROGRAM_NAME ":set:" + lexical_cast<std::string>(parSet) : "");
const auto regexes = compile_regexes(parRegexes);
for (const auto& itm : parRedis.scan(PROGRAM_NAME ":file:*")) {
const auto& file_key = itm;
const auto path = redis::get_string(parRedis.run("HGET", file_key, "path"));
for (const auto& regex : regexes) {
if (not std::regex_search(path, regex))
continue;
for (const auto &tag : parTags) {
std::ostringstream oss;
oss << PROGRAM_NAME ":tag:" << tag;
const std::string tag_key = oss.str();
parTagIfInSet.run(batch, std::make_tuple(tag_key, file_key), std::make_tuple(set_key));
}
}
}
}
void delete_tags (redis::Command& parRedis, redis::Script& parDeleIfInSet, const std::vector<uint64_t>& parFiles, const std::vector<boost::string_ref>& parTags, GroupIDType parSet) {