mSqlConnection = new DB::SqlConnection(path,
DB::SqlConnection::Flag::None,
DB::SqlConnection::Flag::RW);
+ initDataCommands();
} catch (DB::SqlConnection::Exception::Base &e) {
LogError("Database initialization error: " << e.DumpToString());
ThrowMsg(PrivilegeDb::Exception::IOError,
};
}
+void PrivilegeDb::initDataCommands()
+{
+ for (auto &it : Queries) {
+ m_commands.push_back(mSqlConnection->PrepareDataCommand(it.second));
+ }
+}
+
PrivilegeDb::~PrivilegeDb()
{
delete mSqlConnection;
bool PrivilegeDb::PkgIdExists(const std::string &pkgId)
{
return try_catch<bool>([&] {
- DB::SqlConnection::DataCommandAutoPtr command =
- mSqlConnection->PrepareDataCommand(
- Queries.at(QueryType::EPkgIdExists));
- command->BindString(1, pkgId.c_str());
- if (command->Step()) {
- // pkgId found in the database
- command->Reset();
- return true;
- };
+ DB::SqlConnection::DataCommandAutoPtr &command =
+ m_commands.at(static_cast<size_t>(QueryType::EPkgIdExists));
- // pkgId not found in the database
- return false;
+ command->Reset();
+ command->BindString(1, pkgId.c_str());
+ return command->Step();
});
}
bool PrivilegeDb::GetAppPkgId(const std::string &appId, std::string &pkgId)
{
return try_catch<bool>([&] {
- DB::SqlConnection::DataCommandAutoPtr command =
- mSqlConnection->PrepareDataCommand(Queries.at(QueryType::EGetPkgId));
+ DB::SqlConnection::DataCommandAutoPtr &command =
+ m_commands.at(static_cast<size_t>(QueryType::EGetPkgId));
+
+ command->Reset();
command->BindString(1, appId.c_str());
if (!command->Step()) {
pkgIdIsNew = !(this->PkgIdExists(pkgId));
try_catch<void>([&] {
- DB::SqlConnection::DataCommandAutoPtr command =
- mSqlConnection->PrepareDataCommand(
- Queries.at(QueryType::EAddApplication));
+ DB::SqlConnection::DataCommandAutoPtr &command =
+ m_commands.at(static_cast<size_t>(QueryType::EAddApplication));
+ command->Reset();
command->BindString(1, appId.c_str());
command->BindString(2, pkgId.c_str());
command->BindInteger(3, static_cast<unsigned int>(uid));
Queries.at(QueryType::EAddApplication));
};
- command->Reset();
LogDebug("Added appId: " << appId << ", pkgId: " << pkgId);
});
}
return;
}
- DB::SqlConnection::DataCommandAutoPtr command =
- mSqlConnection->PrepareDataCommand(
- Queries.at(QueryType::ERemoveApplication));
+ DB::SqlConnection::DataCommandAutoPtr &command =
+ m_commands.at(static_cast<size_t>(QueryType::ERemoveApplication));
+ command->Reset();
command->BindString(1, appId.c_str());
command->BindInteger(2, static_cast<unsigned int>(uid));
Queries.at(QueryType::ERemoveApplication));
};
- command->Reset();
LogDebug("Removed appId: " << appId);
pkgIdIsNoMore = !(this->PkgIdExists(pkgId));
std::vector<std::string> ¤tPrivileges)
{
try_catch<void>([&] {
- DB::SqlConnection::DataCommandAutoPtr command =
- mSqlConnection->PrepareDataCommand(
- Queries.at(QueryType::EGetPkgPrivileges));
+ DB::SqlConnection::DataCommandAutoPtr &command =
+ m_commands.at(static_cast<size_t>(QueryType::EGetPkgPrivileges));
+
+ command->Reset();
command->BindString(1, pkgId.c_str());
command->BindInteger(2, static_cast<unsigned int>(uid));
void PrivilegeDb::RemoveAppPrivileges(const std::string &appId, uid_t uid)
{
try_catch<void>([&] {
- DB::SqlConnection::DataCommandAutoPtr command =
- mSqlConnection->PrepareDataCommand(Queries.at(QueryType::ERemoveAppPrivileges));
+ DB::SqlConnection::DataCommandAutoPtr &command =
+ m_commands.at(static_cast<size_t>(QueryType::ERemoveAppPrivileges));
+ command->Reset();
command->BindString(1, appId.c_str());
command->BindInteger(2, static_cast<unsigned int>(uid));
if (command->Step()) {
const std::vector<std::string> &privileges)
{
try_catch<void>([&] {
- DB::SqlConnection::DataCommandAutoPtr command =
- mSqlConnection->PrepareDataCommand(Queries.at(QueryType::EAddAppPrivileges));
+ DB::SqlConnection::DataCommandAutoPtr &command =
+ m_commands.at(static_cast<size_t>(QueryType::EAddAppPrivileges));
+
+ command->Reset();
command->BindString(1, appId.c_str());
command->BindInteger(2, static_cast<unsigned int>(uid));
std::vector<std::string> &groups)
{
try_catch<void>([&] {
- DB::SqlConnection::DataCommandAutoPtr command =
- mSqlConnection->PrepareDataCommand(
- Queries.at(QueryType::EGetPrivilegeGroups));
+ DB::SqlConnection::DataCommandAutoPtr &command =
+ m_commands.at(static_cast<size_t>(QueryType::EGetPrivilegeGroups));
+
+ command->Reset();
command->BindString(1, privilege.c_str());
while (command->Step()) {
void PrivilegeDb::GetUserApps(uid_t uid, std::vector<std::string> &apps)
{
try_catch<void>([&] {
- DB::SqlConnection::DataCommandAutoPtr command =
- mSqlConnection->PrepareDataCommand(
- Queries.at(QueryType::EGetUserApps));
+ DB::SqlConnection::DataCommandAutoPtr &command =
+ m_commands.at(static_cast<size_t>(QueryType::EGetUserApps));
+
+ command->Reset();
command->BindInteger(1, static_cast<unsigned int>(uid));
apps.clear();
while (command->Step()) {