milvus/core/src/server/DBWrapper.h
yudong.cai 30534bee23 MS-653 fix when config check fail, Milvus close without message
Former-commit-id: b9ff8c2228d07b6abe9d72f87e06bee823a05363
2019-10-15 15:59:56 +08:00

65 lines
1.5 KiB
C++

// Licensed to the Apache Software Foundation (ASF) under one
// or more contributor license agreements. See the NOTICE file
// distributed with this work for additional information
// regarding copyright ownership. The ASF licenses this file
// to you 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 <string>
#include "db/DB.h"
#include "utils/Status.h"
namespace milvus {
namespace server {
class DBWrapper {
private:
DBWrapper() = default;
~DBWrapper() = default;
public:
static DBWrapper&
GetInstance() {
static DBWrapper wrapper;
return wrapper;
}
static engine::DBPtr
DB() {
return GetInstance().EngineDB();
}
Status
StartService();
Status
StopService();
engine::DBPtr
EngineDB() {
return db_;
}
private:
Status
PreloadTables(const std::string& preload_tables);
private:
engine::DBPtr db_;
};
} // namespace server
} // namespace milvus