mirror of
https://gitee.com/milvus-io/milvus.git
synced 2025-12-31 16:05:30 +08:00
52 lines
1.4 KiB
C++
52 lines
1.4 KiB
C++
#ifndef VECENGINE_DB_H_
|
|
#define VECENGINE_DB_H_
|
|
|
|
#include <string>
|
|
#include "Options.h"
|
|
#include "Meta.h"
|
|
#include "Status.h"
|
|
#include "Types.h"
|
|
|
|
namespace zilliz {
|
|
namespace vecwise {
|
|
namespace engine {
|
|
|
|
class Env;
|
|
|
|
class DB {
|
|
public:
|
|
static void Open(const Options& options, DB** dbptr);
|
|
|
|
virtual Status add_group(meta::GroupSchema& group_info_) = 0;
|
|
virtual Status get_group(meta::GroupSchema& group_info_) = 0;
|
|
virtual Status has_group(const std::string& group_id_, bool& has_or_not_) = 0;
|
|
virtual Status get_group_files(const std::string& group_id_,
|
|
const int date_delta_,
|
|
meta::GroupFilesSchema& group_files_info_) = 0;
|
|
|
|
virtual Status add_vectors(const std::string& group_id_,
|
|
size_t n, const float* vectors, IDNumbers& vector_ids_) = 0;
|
|
|
|
virtual Status search(const std::string& group_id, size_t k, size_t nq,
|
|
const float* vectors, QueryResults& results) = 0;
|
|
|
|
virtual Status search(const std::string& group_id, size_t k, size_t nq,
|
|
const float* vectors, const meta::DatesT& dates, QueryResults& results) = 0;
|
|
|
|
virtual Status drop_all() = 0;
|
|
|
|
virtual Status count(const std::string& group_id, long& result) = 0;
|
|
|
|
DB() = default;
|
|
DB(const DB&) = delete;
|
|
DB& operator=(const DB&) = delete;
|
|
|
|
virtual ~DB();
|
|
}; // DB
|
|
|
|
} // namespace engine
|
|
} // namespace vecwise
|
|
} // namespace zilliz
|
|
|
|
#endif // VECENGINE_DB_H_
|