mirror of
https://gitee.com/milvus-io/milvus.git
synced 2025-12-29 06:55:27 +08:00
da55622042
* write error Signed-off-by: groot <yihua.mo@zilliz.com> * out of storage Signed-off-by: groot <yihua.mo@zilliz.com> * clang format Signed-off-by: groot <yihua.mo@zilliz.com> * fix ut Signed-off-by: groot <yihua.mo@zilliz.com> * fix #1955 Signed-off-by: groot <yihua.mo@zilliz.com> * refine code Signed-off-by: groot <yihua.mo@zilliz.com> * compact threashold Signed-off-by: groot <yihua.mo@zilliz.com> * changelog Signed-off-by: groot <yihua.mo@zilliz.com> * search by id for hnsw/pq/annoy Signed-off-by: groot <yihua.mo@zilliz.com> * fix python test Signed-off-by: yhmo <yihua.mo@zilliz.com> * file cleanup timeout config Signed-off-by: yhmo <yihua.mo@zilliz.com> * typo Signed-off-by: yhmo <yihua.mo@zilliz.com> * changelog Signed-off-by: yhmo <yihua.mo@zilliz.com>
623 lines
20 KiB
C++
623 lines
20 KiB
C++
// Copyright (C) 2019-2020 Zilliz. All rights reserved.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance
|
|
// with the License. You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software distributed under the License
|
|
// is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
|
|
// or implied. See the License for the specific language governing permissions and limitations under the License.
|
|
|
|
#pragma once
|
|
|
|
#include <functional>
|
|
#include <mutex>
|
|
#include <string>
|
|
#include <unordered_map>
|
|
#include <vector>
|
|
|
|
#include "config/ConfigNode.h"
|
|
#include "utils/Status.h"
|
|
|
|
namespace milvus {
|
|
namespace server {
|
|
|
|
using ConfigCallBackF = std::function<Status(const std::string&)>;
|
|
|
|
#define CONFIG_CHECK(func) \
|
|
do { \
|
|
Status s = func; \
|
|
if (!s.ok()) { \
|
|
return s; \
|
|
} \
|
|
} while (false)
|
|
|
|
extern const char* CONFIG_NODE_DELIMITER;
|
|
extern const char* CONFIG_VERSION;
|
|
|
|
/* server config */
|
|
extern const char* CONFIG_SERVER;
|
|
extern const char* CONFIG_SERVER_ADDRESS;
|
|
extern const char* CONFIG_SERVER_ADDRESS_DEFAULT;
|
|
extern const char* CONFIG_SERVER_PORT;
|
|
extern const char* CONFIG_SERVER_PORT_DEFAULT;
|
|
extern const char* CONFIG_SERVER_DEPLOY_MODE;
|
|
extern const char* CONFIG_SERVER_DEPLOY_MODE_DEFAULT;
|
|
extern const char* CONFIG_SERVER_TIME_ZONE;
|
|
extern const char* CONFIG_SERVER_TIME_ZONE_DEFAULT;
|
|
extern const char* CONFIG_SERVER_WEB_ENABLE;
|
|
extern const char* CONFIG_SERVER_WEB_ENABLE_DEFAULT;
|
|
extern const char* CONFIG_SERVER_WEB_PORT;
|
|
extern const char* CONFIG_SERVER_WEB_PORT_DEFAULT;
|
|
|
|
/* db config */
|
|
extern const char* CONFIG_DB;
|
|
extern const char* CONFIG_DB_BACKEND_URL;
|
|
extern const char* CONFIG_DB_BACKEND_URL_DEFAULT;
|
|
extern const char* CONFIG_DB_ARCHIVE_DISK_THRESHOLD;
|
|
extern const char* CONFIG_DB_ARCHIVE_DISK_THRESHOLD_DEFAULT;
|
|
extern const char* CONFIG_DB_ARCHIVE_DAYS_THRESHOLD;
|
|
extern const char* CONFIG_DB_ARCHIVE_DAYS_THRESHOLD_DEFAULT;
|
|
extern const char* CONFIG_DB_PRELOAD_COLLECTION;
|
|
extern const char* CONFIG_DB_PRELOAD_COLLECTION_DEFAULT;
|
|
extern const char* CONFIG_DB_AUTO_FLUSH_INTERVAL;
|
|
extern const char* CONFIG_DB_AUTO_FLUSH_INTERVAL_DEFAULT;
|
|
|
|
/* storage config */
|
|
extern const char* CONFIG_STORAGE;
|
|
extern const char* CONFIG_STORAGE_PRIMARY_PATH;
|
|
extern const char* CONFIG_STORAGE_PRIMARY_PATH_DEFAULT;
|
|
extern const char* CONFIG_STORAGE_SECONDARY_PATH;
|
|
extern const char* CONFIG_STORAGE_SECONDARY_PATH_DEFAULT;
|
|
extern const char* CONFIG_STORAGE_FILE_CLEANUP_TIMEOUT;
|
|
extern const char* CONFIG_STORAGE_S3_ENABLE;
|
|
extern const char* CONFIG_STORAGE_S3_ENABLE_DEFAULT;
|
|
extern const char* CONFIG_STORAGE_S3_ADDRESS;
|
|
extern const char* CONFIG_STORAGE_S3_ADDRESS_DEFAULT;
|
|
extern const char* CONFIG_STORAGE_S3_PORT;
|
|
extern const char* CONFIG_STORAGE_S3_PORT_DEFAULT;
|
|
extern const char* CONFIG_STORAGE_S3_ACCESS_KEY;
|
|
extern const char* CONFIG_STORAGE_S3_ACCESS_KEY_DEFAULT;
|
|
extern const char* CONFIG_STORAGE_S3_SECRET_KEY;
|
|
extern const char* CONFIG_STORAGE_S3_SECRET_KEY_DEFAULT;
|
|
extern const char* CONFIG_STORAGE_S3_BUCKET;
|
|
extern const char* CONFIG_STORAGE_S3_BUCKET_DEFAULT;
|
|
|
|
/* cache config */
|
|
extern const char* CONFIG_CACHE;
|
|
extern const char* CONFIG_CACHE_CPU_CACHE_CAPACITY;
|
|
extern const char* CONFIG_CACHE_CPU_CACHE_CAPACITY_DEFAULT;
|
|
extern const char* CONFIG_CACHE_CPU_CACHE_THRESHOLD;
|
|
extern const char* CONFIG_CACHE_CPU_CACHE_THRESHOLD_DEFAULT;
|
|
extern const char* CONFIG_CACHE_INSERT_BUFFER_SIZE;
|
|
extern const char* CONFIG_CACHE_INSERT_BUFFER_SIZE_DEFAULT;
|
|
extern const char* CONFIG_CACHE_CACHE_INSERT_DATA;
|
|
extern const char* CONFIG_CACHE_CACHE_INSERT_DATA_DEFAULT;
|
|
|
|
/* metric config */
|
|
extern const char* CONFIG_METRIC;
|
|
extern const char* CONFIG_METRIC_ENABLE_MONITOR;
|
|
extern const char* CONFIG_METRIC_ENABLE_MONITOR_DEFAULT;
|
|
extern const char* CONFIG_METRIC_ADDRESS;
|
|
extern const char* CONFIG_METRIC_ADDRESS_DEFAULT;
|
|
extern const char* CONFIG_METRIC_PORT;
|
|
extern const char* CONFIG_METRIC_PORT_DEFAULT;
|
|
|
|
/* engine config */
|
|
extern const char* CONFIG_ENGINE;
|
|
extern const char* CONFIG_ENGINE_USE_BLAS_THRESHOLD;
|
|
extern const char* CONFIG_ENGINE_USE_BLAS_THRESHOLD_DEFAULT;
|
|
extern const char* CONFIG_ENGINE_OMP_THREAD_NUM;
|
|
extern const char* CONFIG_ENGINE_OMP_THREAD_NUM_DEFAULT;
|
|
extern const char* CONFIG_ENGINE_SIMD_TYPE;
|
|
extern const char* CONFIG_ENGINE_SIMD_TYPE_DEFAULT;
|
|
extern const char* CONFIG_ENGINE_GPU_SEARCH_THRESHOLD;
|
|
extern const char* CONFIG_ENGINE_GPU_SEARCH_THRESHOLD_DEFAULT;
|
|
|
|
/* gpu resource config */
|
|
extern const char* CONFIG_GPU_RESOURCE;
|
|
extern const char* CONFIG_GPU_RESOURCE_ENABLE;
|
|
#ifdef MILVUS_GPU_VERSION
|
|
extern const char* CONFIG_GPU_RESOURCE_ENABLE_DEFAULT;
|
|
#else
|
|
extern const char* CONFIG_GPU_RESOURCE_ENABLE_DEFAULT;
|
|
#endif
|
|
extern const char* CONFIG_GPU_RESOURCE_CACHE_CAPACITY;
|
|
extern const char* CONFIG_GPU_RESOURCE_CACHE_CAPACITY_DEFAULT;
|
|
extern const char* CONFIG_GPU_RESOURCE_CACHE_THRESHOLD;
|
|
extern const char* CONFIG_GPU_RESOURCE_CACHE_THRESHOLD_DEFAULT;
|
|
extern const char* CONFIG_GPU_RESOURCE_DELIMITER;
|
|
extern const char* CONFIG_GPU_RESOURCE_SEARCH_RESOURCES;
|
|
extern const char* CONFIG_GPU_RESOURCE_SEARCH_RESOURCES_DEFAULT;
|
|
extern const char* CONFIG_GPU_RESOURCE_BUILD_INDEX_RESOURCES;
|
|
extern const char* CONFIG_GPU_RESOURCE_BUILD_INDEX_RESOURCES_DEFAULT;
|
|
|
|
/* tracing config */
|
|
extern const char* CONFIG_TRACING;
|
|
extern const char* CONFIG_TRACING_JSON_CONFIG_PATH;
|
|
|
|
/* wal config */
|
|
extern const char* CONFIG_WAL;
|
|
extern const char* CONFIG_WAL_ENABLE;
|
|
extern const char* CONFIG_WAL_ENABLE_DEFAULT;
|
|
extern const char* CONFIG_WAL_RECOVERY_ERROR_IGNORE;
|
|
extern const char* CONFIG_WAL_RECOVERY_ERROR_IGNORE_DEFAULT;
|
|
extern const char* CONFIG_WAL_BUFFER_SIZE;
|
|
extern const char* CONFIG_WAL_BUFFER_SIZE_DEFAULT;
|
|
extern const int64_t CONFIG_WAL_BUFFER_SIZE_MAX;
|
|
extern const int64_t CONFIG_WAL_BUFFER_SIZE_MIN;
|
|
extern const char* CONFIG_WAL_WAL_PATH;
|
|
extern const char* CONFIG_WAL_WAL_PATH_DEFAULT;
|
|
|
|
/* logs config */
|
|
extern const char* CONFIG_LOGS;
|
|
extern const char* CONFIG_LOGS_TRACE_ENABLE;
|
|
extern const char* CONFIG_LOGS_TRACE_ENABLE_DEFAULT;
|
|
extern const char* CONFIG_LOGS_DEBUG_ENABLE;
|
|
extern const char* CONFIG_LOGS_DEBUG_ENABLE_DEFAULT;
|
|
extern const char* CONFIG_LOGS_INFO_ENABLE;
|
|
extern const char* CONFIG_LOGS_INFO_ENABLE_DEFAULT;
|
|
extern const char* CONFIG_LOGS_WARNING_ENABLE;
|
|
extern const char* CONFIG_LOGS_WARNING_ENABLE_DEFAULT;
|
|
extern const char* CONFIG_LOGS_ERROR_ENABLE;
|
|
extern const char* CONFIG_LOGS_ERROR_ENABLE_DEFAULT;
|
|
extern const char* CONFIG_LOGS_FATAL_ENABLE;
|
|
extern const char* CONFIG_LOGS_FATAL_ENABLE_DEFAULT;
|
|
extern const char* CONFIG_LOGS_PATH;
|
|
extern const char* CONFIG_LOGS_MAX_LOG_FILE_SIZE;
|
|
extern const char* CONFIG_LOGS_MAX_LOG_FILE_SIZE_DEFAULT;
|
|
extern const int64_t CONFIG_LOGS_MAX_LOG_FILE_SIZE_MAX;
|
|
extern const int64_t CONFIG_LOGS_MAX_LOG_FILE_SIZE_MIN;
|
|
extern const char* CONFIG_LOGS_DELETE_EXCEEDS;
|
|
extern const char* CONFIG_LOGS_DELETE_EXCEEDS_DEFAULT;
|
|
extern const int64_t CONFIG_LOGS_DELETE_EXCEEDS_MAX;
|
|
extern const int64_t CONFIG_LOGS_DELETE_EXCEEDS_MIN;
|
|
|
|
class Config {
|
|
private:
|
|
Config();
|
|
|
|
public:
|
|
static Config&
|
|
GetInstance();
|
|
Status
|
|
LoadConfigFile(const std::string& filename);
|
|
Status
|
|
ValidateConfig();
|
|
Status
|
|
ResetDefaultConfig();
|
|
void
|
|
GetConfigJsonStr(std::string& result, int64_t indent = -1);
|
|
Status
|
|
ProcessConfigCli(std::string& result, const std::string& cmd);
|
|
|
|
Status
|
|
GenUniqueIdentityID(const std::string& identity, std::string& uid);
|
|
|
|
Status
|
|
RegisterCallBack(const std::string& node, const std::string& sub_node, const std::string& key,
|
|
ConfigCallBackF& callback);
|
|
|
|
Status
|
|
CancelCallBack(const std::string& node, const std::string& sub_node, const std::string& key);
|
|
|
|
private:
|
|
ConfigNode&
|
|
GetConfigRoot();
|
|
ConfigNode&
|
|
GetConfigNode(const std::string& name);
|
|
bool
|
|
ConfigNodeValid(const std::string& parent_key, const std::string& child_key);
|
|
Status
|
|
GetConfigValueInMem(const std::string& parent_key, const std::string& child_key, std::string& value);
|
|
Status
|
|
SetConfigValueInMem(const std::string& parent_key, const std::string& child_key, const std::string& value);
|
|
Status
|
|
GetConfigCli(std::string& value, const std::string& parent_key, const std::string& child_key);
|
|
Status
|
|
SetConfigCli(const std::string& parent_key, const std::string& child_key, const std::string& value);
|
|
|
|
Status
|
|
UpdateFileConfigFromMem(const std::string& parent_key, const std::string& child_key);
|
|
|
|
///////////////////////////////////////////////////////////////////////////
|
|
Status
|
|
CheckConfigVersion(const std::string& value);
|
|
|
|
/* server config */
|
|
Status
|
|
CheckServerConfigAddress(const std::string& value);
|
|
Status
|
|
CheckServerConfigPort(const std::string& value);
|
|
Status
|
|
CheckServerConfigDeployMode(const std::string& value);
|
|
Status
|
|
CheckServerConfigTimeZone(const std::string& value);
|
|
Status
|
|
CheckServerConfigWebEnable(const std::string& value);
|
|
Status
|
|
CheckServerConfigWebPort(const std::string& value);
|
|
|
|
/* db config */
|
|
Status
|
|
CheckDBConfigBackendUrl(const std::string& value);
|
|
Status
|
|
CheckDBConfigPreloadCollection(const std::string& value);
|
|
Status
|
|
CheckDBConfigArchiveDiskThreshold(const std::string& value);
|
|
Status
|
|
CheckDBConfigArchiveDaysThreshold(const std::string& value);
|
|
Status
|
|
CheckDBConfigAutoFlushInterval(const std::string& value);
|
|
|
|
/* storage config */
|
|
Status
|
|
CheckStorageConfigPrimaryPath(const std::string& value);
|
|
Status
|
|
CheckStorageConfigSecondaryPath(const std::string& value);
|
|
Status
|
|
CheckStorageConfigFileCleanupTimeout(const std::string& value);
|
|
Status
|
|
CheckStorageConfigS3Enable(const std::string& value);
|
|
Status
|
|
CheckStorageConfigS3Address(const std::string& value);
|
|
Status
|
|
CheckStorageConfigS3Port(const std::string& value);
|
|
Status
|
|
CheckStorageConfigS3AccessKey(const std::string& value);
|
|
Status
|
|
CheckStorageConfigS3SecretKey(const std::string& value);
|
|
Status
|
|
CheckStorageConfigS3Bucket(const std::string& value);
|
|
|
|
/* metric config */
|
|
Status
|
|
CheckMetricConfigEnableMonitor(const std::string& value);
|
|
Status
|
|
CheckMetricConfigAddress(const std::string& value);
|
|
Status
|
|
CheckMetricConfigPort(const std::string& value);
|
|
|
|
/* cache config */
|
|
Status
|
|
CheckCacheConfigCpuCacheCapacity(const std::string& value);
|
|
Status
|
|
CheckCacheConfigCpuCacheThreshold(const std::string& value);
|
|
Status
|
|
CheckCacheConfigInsertBufferSize(const std::string& value);
|
|
Status
|
|
CheckCacheConfigCacheInsertData(const std::string& value);
|
|
|
|
/* engine config */
|
|
Status
|
|
CheckEngineConfigUseBlasThreshold(const std::string& value);
|
|
Status
|
|
CheckEngineConfigOmpThreadNum(const std::string& value);
|
|
Status
|
|
CheckEngineConfigSimdType(const std::string& value);
|
|
|
|
#ifdef MILVUS_GPU_VERSION
|
|
Status
|
|
CheckEngineConfigGpuSearchThreshold(const std::string& value);
|
|
|
|
/* gpu resource config */
|
|
Status
|
|
CheckGpuResourceConfigEnable(const std::string& value);
|
|
Status
|
|
CheckGpuResourceConfigCacheCapacity(const std::string& value);
|
|
Status
|
|
CheckGpuResourceConfigCacheThreshold(const std::string& value);
|
|
Status
|
|
CheckGpuResourceConfigSearchResources(const std::vector<std::string>& value);
|
|
Status
|
|
CheckGpuResourceConfigBuildIndexResources(const std::vector<std::string>& value);
|
|
#endif
|
|
|
|
/* tracing config */
|
|
Status
|
|
CheckTracingConfigJsonConfigPath(const std::string& value);
|
|
|
|
/* wal config */
|
|
Status
|
|
CheckWalConfigEnable(const std::string& value);
|
|
Status
|
|
CheckWalConfigRecoveryErrorIgnore(const std::string& value);
|
|
Status
|
|
CheckWalConfigBufferSize(const std::string& value);
|
|
Status
|
|
CheckWalConfigWalPath(const std::string& value);
|
|
|
|
/* logs config */
|
|
Status
|
|
CheckLogsTraceEnable(const std::string& value);
|
|
Status
|
|
CheckLogsDebugEnable(const std::string& value);
|
|
Status
|
|
CheckLogsInfoEnable(const std::string& value);
|
|
Status
|
|
CheckLogsWarningEnable(const std::string& value);
|
|
Status
|
|
CheckLogsErrorEnable(const std::string& value);
|
|
Status
|
|
CheckLogsFatalEnable(const std::string& value);
|
|
Status
|
|
CheckLogsPath(const std::string& value);
|
|
Status
|
|
CheckLogsMaxLogFileSize(const std::string& value);
|
|
Status
|
|
CheckLogsDeleteExceeds(const std::string& value);
|
|
|
|
std::string
|
|
GetConfigStr(const std::string& parent_key, const std::string& child_key, const std::string& default_value = "");
|
|
std::string
|
|
GetConfigSequenceStr(const std::string& parent_key, const std::string& child_key, const std::string& delim = ",",
|
|
const std::string& default_value = "");
|
|
Status
|
|
GetConfigVersion(std::string& value);
|
|
|
|
Status
|
|
ExecCallBacks(const std::string& node, const std::string& sub_node, const std::string& value);
|
|
|
|
public:
|
|
/* server config */
|
|
Status
|
|
GetServerConfigAddress(std::string& value);
|
|
Status
|
|
GetServerConfigPort(std::string& value);
|
|
Status
|
|
GetServerConfigDeployMode(std::string& value);
|
|
Status
|
|
GetServerConfigTimeZone(std::string& value);
|
|
Status
|
|
GetServerConfigWebEnable(bool& value);
|
|
Status
|
|
GetServerConfigWebPort(std::string& value);
|
|
|
|
/* db config */
|
|
Status
|
|
GetDBConfigBackendUrl(std::string& value);
|
|
Status
|
|
GetDBConfigArchiveDiskThreshold(int64_t& value);
|
|
Status
|
|
GetDBConfigArchiveDaysThreshold(int64_t& value);
|
|
Status
|
|
GetDBConfigPreloadCollection(std::string& value);
|
|
Status
|
|
GetDBConfigAutoFlushInterval(int64_t& value);
|
|
|
|
/* storage config */
|
|
Status
|
|
GetStorageConfigPrimaryPath(std::string& value);
|
|
Status
|
|
GetStorageConfigSecondaryPath(std::string& value);
|
|
Status
|
|
GetStorageConfigFileCleanupTimeup(int64_t& value);
|
|
Status
|
|
GetStorageConfigS3Enable(bool& value);
|
|
Status
|
|
GetStorageConfigS3Address(std::string& value);
|
|
Status
|
|
GetStorageConfigS3Port(std::string& value);
|
|
Status
|
|
GetStorageConfigS3AccessKey(std::string& value);
|
|
Status
|
|
GetStorageConfigS3SecretKey(std::string& value);
|
|
Status
|
|
GetStorageConfigS3Bucket(std::string& value);
|
|
|
|
/* metric config */
|
|
Status
|
|
GetMetricConfigEnableMonitor(bool& value);
|
|
Status
|
|
GetMetricConfigAddress(std::string& value);
|
|
Status
|
|
GetMetricConfigPort(std::string& value);
|
|
|
|
/* cache config */
|
|
Status
|
|
GetCacheConfigCpuCacheCapacity(int64_t& value);
|
|
Status
|
|
GetCacheConfigCpuCacheThreshold(float& value);
|
|
Status
|
|
GetCacheConfigInsertBufferSize(int64_t& value);
|
|
Status
|
|
GetCacheConfigCacheInsertData(bool& value);
|
|
|
|
/* engine config */
|
|
Status
|
|
GetEngineConfigUseBlasThreshold(int64_t& value);
|
|
Status
|
|
GetEngineConfigOmpThreadNum(int64_t& value);
|
|
Status
|
|
GetEngineConfigSimdType(std::string& value);
|
|
|
|
#ifdef MILVUS_GPU_VERSION
|
|
Status
|
|
GetEngineConfigGpuSearchThreshold(int64_t& value);
|
|
|
|
/* gpu resource config */
|
|
Status
|
|
GetGpuResourceConfigEnable(bool& value);
|
|
Status
|
|
GetGpuResourceConfigCacheCapacity(int64_t& value);
|
|
Status
|
|
GetGpuResourceConfigCacheThreshold(float& value);
|
|
Status
|
|
GetGpuResourceConfigSearchResources(std::vector<int64_t>& value);
|
|
Status
|
|
GetGpuResourceConfigBuildIndexResources(std::vector<int64_t>& value);
|
|
#endif
|
|
|
|
/* tracing config */
|
|
Status
|
|
GetTracingConfigJsonConfigPath(std::string& value);
|
|
|
|
/* wal config */
|
|
Status
|
|
GetWalConfigEnable(bool& value);
|
|
Status
|
|
GetWalConfigRecoveryErrorIgnore(bool& value);
|
|
Status
|
|
GetWalConfigBufferSize(int64_t& value);
|
|
Status
|
|
GetWalConfigWalPath(std::string& value);
|
|
|
|
/* logs config */
|
|
Status
|
|
GetLogsTraceEnable(bool& value);
|
|
Status
|
|
GetLogsDebugEnable(bool& value);
|
|
Status
|
|
GetLogsInfoEnable(bool& value);
|
|
Status
|
|
GetLogsWarningEnable(bool& value);
|
|
Status
|
|
GetLogsErrorEnable(bool& value);
|
|
Status
|
|
GetLogsFatalEnable(bool& value);
|
|
Status
|
|
GetLogsPath(std::string& value);
|
|
Status
|
|
GetLogsMaxLogFileSize(int64_t& value);
|
|
Status
|
|
GetLogsDeleteExceeds(int64_t& value);
|
|
|
|
Status
|
|
GetServerRestartRequired(bool& required);
|
|
|
|
public:
|
|
/* server config */
|
|
Status
|
|
SetServerConfigAddress(const std::string& value);
|
|
Status
|
|
SetServerConfigPort(const std::string& value);
|
|
Status
|
|
SetServerConfigDeployMode(const std::string& value);
|
|
Status
|
|
SetServerConfigTimeZone(const std::string& value);
|
|
Status
|
|
SetServerConfigWebEnable(const std::string& value);
|
|
Status
|
|
SetServerConfigWebPort(const std::string& value);
|
|
|
|
/* db config */
|
|
Status
|
|
SetDBConfigBackendUrl(const std::string& value);
|
|
Status
|
|
SetDBConfigPreloadCollection(const std::string& value);
|
|
Status
|
|
SetDBConfigArchiveDiskThreshold(const std::string& value);
|
|
Status
|
|
SetDBConfigArchiveDaysThreshold(const std::string& value);
|
|
Status
|
|
SetDBConfigAutoFlushInterval(const std::string& value);
|
|
|
|
/* storage config */
|
|
Status
|
|
SetStorageConfigPrimaryPath(const std::string& value);
|
|
Status
|
|
SetStorageConfigSecondaryPath(const std::string& value);
|
|
Status
|
|
SetStorageConfigFileCleanupTimeout(const std::string& value);
|
|
Status
|
|
SetStorageConfigS3Enable(const std::string& value);
|
|
Status
|
|
SetStorageConfigS3Address(const std::string& value);
|
|
Status
|
|
SetStorageConfigS3Port(const std::string& value);
|
|
Status
|
|
SetStorageConfigS3AccessKey(const std::string& value);
|
|
Status
|
|
SetStorageConfigS3SecretKey(const std::string& value);
|
|
Status
|
|
SetStorageConfigS3Bucket(const std::string& value);
|
|
|
|
/* metric config */
|
|
Status
|
|
SetMetricConfigEnableMonitor(const std::string& value);
|
|
Status
|
|
SetMetricConfigAddress(const std::string& value);
|
|
Status
|
|
SetMetricConfigPort(const std::string& value);
|
|
|
|
/* cache config */
|
|
Status
|
|
SetCacheConfigCpuCacheCapacity(const std::string& value);
|
|
Status
|
|
SetCacheConfigCpuCacheThreshold(const std::string& value);
|
|
Status
|
|
SetCacheConfigInsertBufferSize(const std::string& value);
|
|
Status
|
|
SetCacheConfigCacheInsertData(const std::string& value);
|
|
|
|
/* engine config */
|
|
Status
|
|
SetEngineConfigUseBlasThreshold(const std::string& value);
|
|
Status
|
|
SetEngineConfigOmpThreadNum(const std::string& value);
|
|
Status
|
|
SetEngineConfigSimdType(const std::string& value);
|
|
|
|
/* tracing config */
|
|
Status
|
|
SetTracingConfigJsonConfigPath(const std::string& value);
|
|
|
|
/* wal config */
|
|
Status
|
|
SetWalConfigEnable(const std::string& value);
|
|
Status
|
|
SetWalConfigRecoveryErrorIgnore(const std::string& value);
|
|
Status
|
|
SetWalConfigBufferSize(const std::string& value);
|
|
Status
|
|
SetWalConfigWalPath(const std::string& value);
|
|
|
|
/* logs config */
|
|
Status
|
|
SetLogsTraceEnable(const std::string& value);
|
|
Status
|
|
SetLogsDebugEnable(const std::string& value);
|
|
Status
|
|
SetLogsInfoEnable(const std::string& value);
|
|
Status
|
|
SetLogsWarningEnable(const std::string& value);
|
|
Status
|
|
SetLogsErrorEnable(const std::string& value);
|
|
Status
|
|
SetLogsFatalEnable(const std::string& value);
|
|
Status
|
|
SetLogsPath(const std::string& value);
|
|
Status
|
|
SetLogsMaxLogFileSize(const std::string& value);
|
|
Status
|
|
SetLogsDeleteExceeds(const std::string& value);
|
|
|
|
#ifdef MILVUS_GPU_VERSION
|
|
Status
|
|
SetEngineConfigGpuSearchThreshold(const std::string& value);
|
|
|
|
/* gpu resource config */
|
|
Status
|
|
SetGpuResourceConfigEnable(const std::string& value);
|
|
Status
|
|
SetGpuResourceConfigCacheCapacity(const std::string& value);
|
|
Status
|
|
SetGpuResourceConfigCacheThreshold(const std::string& value);
|
|
Status
|
|
SetGpuResourceConfigSearchResources(const std::string& value);
|
|
Status
|
|
SetGpuResourceConfigBuildIndexResources(const std::string& value);
|
|
#endif
|
|
|
|
private:
|
|
bool restart_required_ = false;
|
|
std::string config_file_;
|
|
std::unordered_map<std::string, std::unordered_map<std::string, std::string>> config_map_;
|
|
std::unordered_map<std::string, std::unordered_map<std::string, ConfigCallBackF>> config_callback_;
|
|
std::mutex mutex_;
|
|
};
|
|
|
|
} // namespace server
|
|
} // namespace milvus
|