query-builder: Refactor insert interface
authorSangwan Kwon <sangwan.kwon@samsung.com>
Tue, 4 Feb 2020 09:39:14 +0000 (18:39 +0900)
committer권상완/Security 2Lab(SR)/Engineer/삼성전자 <sangwan.kwon@samsung.com>
Mon, 17 Feb 2020 11:16:16 +0000 (20:16 +0900)
Signed-off-by: Sangwan Kwon <sangwan.kwon@samsung.com>
src/vist/policy/policy-storage.cpp
src/vist/query-builder/crud.hpp
src/vist/query-builder/tests/query-builder.cpp

index 2dd390c..2d0305f 100644 (file)
@@ -124,8 +124,8 @@ void PolicyStorage::define(const std::string& policy, const PolicyValue& ivalue)
 
        PolicyDefinition pd = { policy, ivalue.dump() };
 
-       std::string query = schema::policyDefinition.insert(&PolicyDefinition::name,
-                                                                                                           &PolicyDefinition::ivalue);
+       std::string query = schema::PolicyDefinitionTable.insert(PolicyDefinition::Name,
+                                                                                                                        PolicyDefinition::Ivalue);
        database::Statement stmt(*database, query);
        stmt.bind(1, pd.name);
        stmt.bind(2, pd.ivalue);
@@ -147,7 +147,7 @@ void PolicyStorage::enroll(const std::string& name)
        /// Make admin deactivated as default.
        Admin admin = {name , 0};
 
-       std::string query = schema::admin.insert(&Admin::name, &Admin::activated);
+       std::string query = schema::AdminTable.insert(Admin::Name, Admin::Activated);
        database::Statement stmt(*database, query);
        stmt.bind(1, admin.name);
        stmt.bind(2, admin.activated);
index 6fd4a34..ae4b578 100644 (file)
@@ -153,7 +153,7 @@ T& Crud<T>::insert(ColumnTypes&&... cts)
 {
        static_cast<T*>(this)->cache.clear();
 
-       auto columnNames = static_cast<T*>(this)->getColumnNames(std::forward<ColumnTypes>(cts)...);
+       auto columnNames = static_cast<T*>(this)->_getColumnNames(std::forward<ColumnTypes>(cts)...);
 
        std::stringstream ss;
        ss << "INSERT INTO " << static_cast<T*>(this)->name << " (";
index 23a75b1..9fccaae 100644 (file)
@@ -147,8 +147,8 @@ TEST(QueryBuilderTsqbTests, DELETE)
 
 TEST(QueryBuilderTsqbTests, INSERT)
 {
-       std::string insert1 = admin.insert(&Admin::id, &Admin::pkg, &Admin::uid, &Admin::key);
-       std::string insert2 = admin.insert(&Admin::id, &Admin::pkg, &Admin::key);
+       std::string insert1 = AdminTable.insert(Admin::Id, Admin::Pkg, Admin::Uid, Admin::Key);
+       std::string insert2 = AdminTable.insert(Admin::Id, Admin::Pkg, Admin::Key);
 
        EXPECT_EQ(insert1, "INSERT INTO admin (id, pkg, uid, key) VALUES (?, ?, ?, ?)");
        EXPECT_EQ(insert2, "INSERT INTO admin (id, pkg, key) VALUES (?, ?, ?)");