mirror of
https://gitee.com/milvus-io/milvus.git
synced 2025-12-31 16:05:30 +08:00
58 lines
2.0 KiB
C++
58 lines
2.0 KiB
C++
/*******************************************************************************
|
|
* Copyright 上海赜睿信息科技有限公司(Zilliz) - All Rights Reserved
|
|
* Unauthorized copying of this file, via any medium is strictly prohibited.
|
|
* Proprietary and confidential.
|
|
******************************************************************************/
|
|
|
|
#include "DBWrapper.h"
|
|
#include "ServerConfig.h"
|
|
#include "utils/CommonUtil.h"
|
|
#include "utils/Log.h"
|
|
|
|
namespace zilliz {
|
|
namespace milvus {
|
|
namespace server {
|
|
|
|
DBWrapper::DBWrapper() {
|
|
zilliz::milvus::engine::Options opt;
|
|
ConfigNode& config = ServerConfig::GetInstance().GetConfig(CONFIG_DB);
|
|
opt.meta.backend_uri = config.GetValue(CONFIG_DB_URL);
|
|
std::string db_path = config.GetValue(CONFIG_DB_PATH);
|
|
opt.meta.path = db_path + "/db";
|
|
int64_t index_size = config.GetInt64Value(CONFIG_DB_INDEX_TRIGGER_SIZE);
|
|
if(index_size > 0) {//ensure larger than zero, unit is MB
|
|
opt.index_trigger_size = (size_t)index_size * engine::ONE_MB;
|
|
}
|
|
ConfigNode& serverConfig = ServerConfig::GetInstance().GetConfig(CONFIG_SERVER);
|
|
opt.mode = serverConfig.GetValue(CONFIG_CLUSTER_MODE, "single");
|
|
// std::cout << "mode = " << opt.mode << std::endl;
|
|
|
|
//set archive config
|
|
engine::ArchiveConf::CriteriaT criterial;
|
|
int64_t disk = config.GetInt64Value(CONFIG_DB_ARCHIVE_DISK, 0);
|
|
int64_t days = config.GetInt64Value(CONFIG_DB_ARCHIVE_DAYS, 0);
|
|
if(disk > 0) {
|
|
criterial[engine::ARCHIVE_CONF_DISK] = disk;
|
|
}
|
|
if(days > 0) {
|
|
criterial[engine::ARCHIVE_CONF_DAYS] = days;
|
|
}
|
|
opt.meta.archive_conf.SetCriterias(criterial);
|
|
|
|
//create db root folder
|
|
CommonUtil::CreateDirectory(opt.meta.path);
|
|
|
|
zilliz::milvus::engine::DB::Open(opt, &db_);
|
|
if(db_ == nullptr) {
|
|
SERVER_LOG_ERROR << "Failed to open db. Provided database uri = " << opt.meta.backend_uri;
|
|
throw ServerException(SERVER_NULL_POINTER, "Failed to open db");
|
|
}
|
|
}
|
|
|
|
DBWrapper::~DBWrapper() {
|
|
delete db_;
|
|
}
|
|
|
|
}
|
|
}
|
|
} |