mirror of
https://gitee.com/milvus-io/milvus.git
synced 2025-12-31 16:05:30 +08:00
75 lines
2.3 KiB
C++
75 lines
2.3 KiB
C++
/*******************************************************************************
|
|
* Copyright 上海赜睿信息科技有限公司(Zilliz) - All Rights Reserved
|
|
* Unauthorized copying of this file, via any medium is strictly prohibited.
|
|
* Proprietary and confidential.
|
|
******************************************************************************/
|
|
#pragma once
|
|
|
|
#include "utils/Error.h"
|
|
|
|
#include <string>
|
|
#include <vector>
|
|
#include <unordered_map>
|
|
|
|
namespace rocksdb {
|
|
class DB;
|
|
}
|
|
|
|
namespace zilliz {
|
|
namespace vecwise {
|
|
namespace server {
|
|
|
|
class IVecIdMapper {
|
|
public:
|
|
static IVecIdMapper* GetInstance();
|
|
|
|
virtual ~IVecIdMapper(){}
|
|
|
|
virtual ServerError Put(const std::string& nid, const std::string& sid) = 0;
|
|
virtual ServerError Put(const std::vector<std::string>& nid, const std::vector<std::string>& sid) = 0;
|
|
|
|
virtual ServerError Get(const std::string& nid, std::string& sid) const = 0;
|
|
//NOTE: the 'sid' will be cleared at begin of the function
|
|
virtual ServerError Get(const std::vector<std::string>& nid, std::vector<std::string>& sid) const = 0;
|
|
|
|
virtual ServerError Delete(const std::string& nid) = 0;
|
|
};
|
|
|
|
class SimpleIdMapper : public IVecIdMapper{
|
|
public:
|
|
SimpleIdMapper();
|
|
~SimpleIdMapper();
|
|
|
|
ServerError Put(const std::string& nid, const std::string& sid) override;
|
|
ServerError Put(const std::vector<std::string>& nid, const std::vector<std::string>& sid) override;
|
|
|
|
ServerError Get(const std::string& nid, std::string& sid) const override;
|
|
ServerError Get(const std::vector<std::string>& nid, std::vector<std::string>& sid) const override;
|
|
|
|
ServerError Delete(const std::string& nid) override;
|
|
|
|
private:
|
|
std::unordered_map<std::string, std::string> ids_;
|
|
};
|
|
|
|
class RocksIdMapper : public IVecIdMapper{
|
|
public:
|
|
RocksIdMapper();
|
|
~RocksIdMapper();
|
|
|
|
ServerError Put(const std::string& nid, const std::string& sid) override;
|
|
ServerError Put(const std::vector<std::string>& nid, const std::vector<std::string>& sid) override;
|
|
|
|
ServerError Get(const std::string& nid, std::string& sid) const override;
|
|
ServerError Get(const std::vector<std::string>& nid, std::vector<std::string>& sid) const override;
|
|
|
|
ServerError Delete(const std::string& nid) override;
|
|
|
|
private:
|
|
rocksdb::DB* db_;
|
|
};
|
|
|
|
}
|
|
}
|
|
}
|