test/milvus
1
0
Fork 0
mirror of https://gitee.com/milvus-io/milvus.git synced 2024-11-29 18:38:44 +08:00
Code Issues Actions 11 Packages Projects Releases Wiki Activity

feat(db): init sql orm unique ptr

Browse Source 
Former-commit-id: 41a4a96958a81482d2700a4bc3973d8d91d678c0
This commit is contained in:
Xu Peng 2019-04-17 20:28:16 +08:00
parent 2aa2720dbd
commit 0c8b4b04e8
2 changed files with 19 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
cpp/src/db/DBMetaImpl.cpp
View File

@ -15,6 +15,19 @@ namespace meta {
using namespace sqlite_orm;
inline auto StoragePrototype(const std::string& path) {
return make_storage(path,
make_table("Groups",
make_column("id", &GroupSchema::id, primary_key()),
make_column("group_id", &GroupSchema::group_id, unique()),
make_column("dimension", &GroupSchema::dimension),
make_column("files_cnt", &GroupSchema::files_cnt, default_value(0))));
}
using ConnectorT = decltype(StoragePrototype(""));
static std::unique_ptr<ConnectorT> ConnectorPtr;
long GetFileSize(const std::string& filename)
{
struct stat stat_buf;
@ -32,14 +45,9 @@ Status DBMetaImpl::initialize() {
assert(boost::filesystem::create_directory(_options.path));
}
auto db = make_storage(_options.path + "/meta.sqlite",
make_table("Groups",
make_column("id", &GroupSchema::id, primary_key()),
make_column("group_id", &GroupSchema::group_id, unique()),
make_column("dimension", &GroupSchema::dimension),
make_column("files_cnt", &GroupSchema::files_cnt, default_value(0))));
ConnectorPtr = std::make_unique<ConnectorT>(StoragePrototype(_options.path+"/meta.sqlite"));
db.sync_schema();
ConnectorPtr->sync_schema();
return Status::OK();
}
@ -47,19 +55,13 @@ Status DBMetaImpl::initialize() {
Status DBMetaImpl::add_group(const GroupOptions& options_,
const std::string& group_id_,
GroupSchema& group_info_) {
auto db = make_storage(_options.path + "/meta.sqlite",
make_table("Groups",
make_column("id", &GroupSchema::id, primary_key()),
make_column("group_id", &GroupSchema::group_id, unique()),
make_column("dimension", &GroupSchema::dimension),
make_column("files_cnt", &GroupSchema::files_cnt, default_value(0))));
group_info_.dimension = options_.dimension;
group_info_.group_id = group_id_;
group_info_.files_cnt = 0;
group_info_.id = -1;
try {
auto id = db.insert(group_info_);
auto id = ConnectorPtr->insert(group_info_);
std::cout << "id=" << id << std::endl;
group_info_.id = id;
} catch(std::system_error& e) {
@ -69,7 +71,7 @@ Status DBMetaImpl::add_group(const GroupOptions& options_,
}
Status DBMetaImpl::get_group(const std::string& group_id_, GroupSchema& group_info_) {
//PXU TODO
return Status::OK();
}

2
cpp/src/db/DBMetaImpl.h
View File

@ -9,6 +9,8 @@ namespace vecwise {
namespace engine {
namespace meta {
auto StoragePrototype(const std::string& path);
class DBMetaImpl : public Meta {
public:
DBMetaImpl(const DBMetaOptions& options_);