mirror of
https://github.com/uroni/urbackup_backend.git
synced 2025-10-26 11:36:50 +00:00
126 lines
3.3 KiB
C++
126 lines
3.3 KiB
C++
/*************************************************************************
|
|
* UrBackup - Client/Server backup system
|
|
* Copyright (C) 2011-2016 Martin Raiber
|
|
*
|
|
* This program is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU Affero General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU Affero General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Affero General Public License
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
**************************************************************************/
|
|
|
|
#include "Interface/Types.h"
|
|
#include "Interface/Database.h"
|
|
|
|
#include "SettingsReader.h"
|
|
#include "stringtools.h"
|
|
#include "Server.h"
|
|
|
|
#include "DBSettingsReader.h"
|
|
|
|
#include <iostream>
|
|
|
|
CDBSettingsReader::CDBSettingsReader(THREAD_ID tid, DATABASE_ID did, const std::string &pTable, const std::string &pSQL)
|
|
{
|
|
table=pTable;
|
|
db=Server->getDatabase(tid, did);
|
|
if(pSQL.empty() )
|
|
query=db->Prepare("SELECT value FROM "+table+" WHERE key=?", false);
|
|
else
|
|
query=db->Prepare(pSQL, false);
|
|
}
|
|
|
|
CDBSettingsReader::CDBSettingsReader(IDatabase *pDB, const std::string &pTable, const std::string &pSQL)
|
|
: db(pDB)
|
|
{
|
|
table=pTable;
|
|
if(pSQL.empty() )
|
|
query=pDB->Prepare("SELECT value FROM "+table+" WHERE key=?", false);
|
|
else
|
|
query=pDB->Prepare(pSQL, false);
|
|
}
|
|
|
|
CDBSettingsReader::~CDBSettingsReader()
|
|
{
|
|
db->destroyQuery(query);
|
|
}
|
|
|
|
bool CDBSettingsReader::getValue(std::string key, std::string *value)
|
|
{
|
|
if(query==NULL)
|
|
{
|
|
return false;
|
|
}
|
|
|
|
query->Bind(key);
|
|
db_results res=query->Read();
|
|
query->Reset();
|
|
|
|
if( res.size()>0 )
|
|
{
|
|
*value=res[0]["value"];
|
|
return true;
|
|
}
|
|
else
|
|
return false;
|
|
}
|
|
|
|
std::vector<std::string> CDBSettingsReader::getKeys()
|
|
{
|
|
return std::vector<std::string>();
|
|
}
|
|
|
|
CDBMemSettingsReader::CDBMemSettingsReader(THREAD_ID tid, DATABASE_ID did, const std::string & pTable, const std::string & pSQL)
|
|
{
|
|
init(Server->getDatabase(tid, did), pTable, pSQL);
|
|
}
|
|
|
|
CDBMemSettingsReader::CDBMemSettingsReader(IDatabase * pDB, const std::string & pTable, const std::string & pSQL)
|
|
{
|
|
init(pDB, pTable, pSQL);
|
|
}
|
|
|
|
bool CDBMemSettingsReader::getValue(std::string key, std::string * value)
|
|
{
|
|
str_map::iterator it = table.find(key);
|
|
if (it != table.end())
|
|
{
|
|
*value = it->second;
|
|
return true;
|
|
}
|
|
else
|
|
{
|
|
return false;
|
|
}
|
|
}
|
|
|
|
std::vector<std::string> CDBMemSettingsReader::getKeys()
|
|
{
|
|
std::vector<std::string> ret;
|
|
for (str_map::iterator it = table.begin(); it != table.end(); ++it)
|
|
{
|
|
ret.push_back(it->first);
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
void CDBMemSettingsReader::init(IDatabase * pDB, const std::string & pTable, const std::string & pSQL)
|
|
{
|
|
db_results res;
|
|
if (pSQL.empty())
|
|
res = pDB->Read("SELECT key, value FROM " + pTable + " WHERE key=?");
|
|
else
|
|
res = pDB->Read(pSQL);
|
|
|
|
for (size_t i = 0; i < res.size(); ++i)
|
|
{
|
|
table.insert(std::make_pair(res[i]["key"], res[i]["value"]));
|
|
}
|
|
} |