mirror of https://github.com/milvus-io/milvus.git
45 lines
741 B
C++
45 lines
741 B
C++
#pragma once
|
|
|
|
#include <map>
|
|
|
|
#include "tantivy-binding.h"
|
|
#include "rust-binding.h"
|
|
|
|
namespace milvus::tantivy {
|
|
|
|
struct RustHashMap {
|
|
public:
|
|
NO_COPY_OR_ASSIGN(RustHashMap);
|
|
|
|
RustHashMap() {
|
|
ptr_ = create_hashmap();
|
|
}
|
|
|
|
~RustHashMap() {
|
|
if (ptr_ != nullptr) {
|
|
free_hashmap(ptr_);
|
|
}
|
|
}
|
|
|
|
void
|
|
from(const std::map<std::string, std::string>& m) {
|
|
for (const auto& [k, v] : m) {
|
|
set(k, v);
|
|
}
|
|
}
|
|
|
|
void*
|
|
get_pointer() {
|
|
return ptr_;
|
|
}
|
|
|
|
void
|
|
set(const std::string& k, const std::string& v) {
|
|
hashmap_set_value(ptr_, k.c_str(), v.c_str());
|
|
}
|
|
|
|
private:
|
|
void* ptr_ = nullptr;
|
|
};
|
|
} // namespace milvus::tantivy
|