mirror of
https://gitee.com/milvus-io/milvus.git
synced 2026-01-01 08:28:10 +08:00
61 lines
1.5 KiB
C++
61 lines
1.5 KiB
C++
#pragma once
|
|
|
|
#include "db/meta/Meta.h"
|
|
#include "MemTable.h"
|
|
#include "db/Status.h"
|
|
#include "MemManager.h"
|
|
|
|
#include <map>
|
|
#include <string>
|
|
#include <ctime>
|
|
#include <memory>
|
|
#include <mutex>
|
|
|
|
|
|
namespace zilliz {
|
|
namespace milvus {
|
|
namespace engine {
|
|
|
|
class MemManagerImpl : public MemManager {
|
|
public:
|
|
using MetaPtr = meta::Meta::Ptr;
|
|
using Ptr = std::shared_ptr<MemManagerImpl>;
|
|
using MemTablePtr = typename MemTable::Ptr;
|
|
|
|
MemManagerImpl(const std::shared_ptr<meta::Meta> &meta, const Options &options)
|
|
: meta_(meta), options_(options) {}
|
|
|
|
Status InsertVectors(const std::string &table_id,
|
|
size_t n, const float *vectors, IDNumbers &vector_ids) override;
|
|
|
|
Status Serialize(std::set<std::string> &table_ids) override;
|
|
|
|
Status EraseMemVector(const std::string &table_id) override;
|
|
|
|
size_t GetCurrentMutableMem() override;
|
|
|
|
size_t GetCurrentImmutableMem() override;
|
|
|
|
size_t GetCurrentMem() override;
|
|
|
|
private:
|
|
MemTablePtr GetMemByTable(const std::string &table_id);
|
|
|
|
Status InsertVectorsNoLock(const std::string &table_id,
|
|
size_t n, const float *vectors, IDNumbers &vector_ids);
|
|
Status ToImmutable();
|
|
|
|
using MemIdMap = std::map<std::string, MemTablePtr>;
|
|
using MemList = std::vector<MemTablePtr>;
|
|
MemIdMap mem_id_map_;
|
|
MemList immu_mem_list_;
|
|
MetaPtr meta_;
|
|
Options options_;
|
|
std::mutex mutex_;
|
|
std::mutex serialization_mtx_;
|
|
}; // NewMemManager
|
|
|
|
|
|
} // namespace engine
|
|
} // namespace milvus
|
|
} // namespace zilliz
|