151 lines
4.0 KiB
C++
151 lines
4.0 KiB
C++
/*
|
|
Minetest
|
|
Copyright (C) 2013 celeron55, Perttu Ahola <celeron55@gmail.com>
|
|
|
|
This program is free software; you can redistribute it and/or modify
|
|
it under the terms of the GNU Lesser General Public License as published by
|
|
the Free Software Foundation; either version 3.0 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 Lesser General Public License for more details.
|
|
|
|
You should have received a copy of the GNU Lesser General Public License along
|
|
with this program; if not, write to the Free Software Foundation, Inc.,
|
|
51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
|
|
*/
|
|
|
|
#ifndef DATABASE_POSTGRESQL_HEADER
|
|
#define DATABASE_POSTGRESQL_HEADER
|
|
|
|
#include <string>
|
|
#include <libpq-fe.h>
|
|
#include "database.h"
|
|
#include "util/basic_macros.h"
|
|
|
|
class Settings;
|
|
|
|
class Database_PostgreSQL: public Database
|
|
{
|
|
public:
|
|
Database_PostgreSQL(const std::string &connect_string);
|
|
~Database_PostgreSQL();
|
|
|
|
void beginSave();
|
|
void endSave();
|
|
|
|
bool initialized() const;
|
|
|
|
|
|
protected:
|
|
// Conversion helpers
|
|
inline int pg_to_int(PGresult *res, int row, int col)
|
|
{
|
|
return atoi(PQgetvalue(res, row, col));
|
|
}
|
|
|
|
inline u32 pg_to_uint(PGresult *res, int row, int col)
|
|
{
|
|
return (u32) atoi(PQgetvalue(res, row, col));
|
|
}
|
|
|
|
inline float pg_to_float(PGresult *res, int row, int col)
|
|
{
|
|
return (float) atof(PQgetvalue(res, row, col));
|
|
}
|
|
|
|
inline v3s16 pg_to_v3s16(PGresult *res, int row, int col)
|
|
{
|
|
return v3s16(
|
|
pg_to_int(res, row, col),
|
|
pg_to_int(res, row, col + 1),
|
|
pg_to_int(res, row, col + 2)
|
|
);
|
|
}
|
|
|
|
inline PGresult *execPrepared(const char *stmtName, const int paramsNumber,
|
|
const void **params,
|
|
const int *paramsLengths = NULL, const int *paramsFormats = NULL,
|
|
bool clear = true, bool nobinary = true)
|
|
{
|
|
return checkResults(PQexecPrepared(m_conn, stmtName, paramsNumber,
|
|
(const char* const*) params, paramsLengths, paramsFormats,
|
|
nobinary ? 1 : 0), clear);
|
|
}
|
|
|
|
inline PGresult *execPrepared(const char *stmtName, const int paramsNumber,
|
|
const char **params, bool clear = true, bool nobinary = true)
|
|
{
|
|
return execPrepared(stmtName, paramsNumber,
|
|
(const void **)params, NULL, NULL, clear, nobinary);
|
|
}
|
|
|
|
void createTableIfNotExists(const std::string &table_name, const std::string &definition);
|
|
void verifyDatabase();
|
|
|
|
// Database initialization
|
|
void connectToDatabase();
|
|
virtual void createDatabase() = 0;
|
|
virtual void initStatements() = 0;
|
|
inline void prepareStatement(const std::string &name, const std::string &sql)
|
|
{
|
|
checkResults(PQprepare(m_conn, name.c_str(), sql.c_str(), 0, NULL));
|
|
}
|
|
|
|
const int getPGVersion() const { return m_pgversion; }
|
|
private:
|
|
// Database connectivity checks
|
|
void ping();
|
|
|
|
// Database usage
|
|
PGresult *checkResults(PGresult *res, bool clear = true);
|
|
|
|
// Attributes
|
|
std::string m_connect_string;
|
|
PGconn *m_conn;
|
|
int m_pgversion;
|
|
};
|
|
|
|
class MapDatabasePostgreSQL : private Database_PostgreSQL, public MapDatabase
|
|
{
|
|
public:
|
|
MapDatabasePostgreSQL(const std::string &connect_string);
|
|
virtual ~MapDatabasePostgreSQL() {}
|
|
|
|
bool saveBlock(const v3s16 &pos, const std::string &data);
|
|
void loadBlock(const v3s16 &pos, std::string *block);
|
|
bool deleteBlock(const v3s16 &pos);
|
|
void listAllLoadableBlocks(std::vector<v3s16> &dst);
|
|
|
|
void beginSave() { Database_PostgreSQL::beginSave(); }
|
|
void endSave() { Database_PostgreSQL::endSave(); }
|
|
|
|
protected:
|
|
virtual void createDatabase();
|
|
virtual void initStatements();
|
|
};
|
|
|
|
class PlayerDatabasePostgreSQL : private Database_PostgreSQL, public PlayerDatabase
|
|
{
|
|
public:
|
|
PlayerDatabasePostgreSQL(const std::string &connect_string);
|
|
virtual ~PlayerDatabasePostgreSQL() {}
|
|
|
|
void savePlayer(RemotePlayer *player);
|
|
bool loadPlayer(RemotePlayer *player, PlayerSAO *sao);
|
|
bool removePlayer(const std::string &name);
|
|
void listPlayers(std::vector<std::string> &res);
|
|
|
|
protected:
|
|
virtual void createDatabase();
|
|
virtual void initStatements();
|
|
|
|
private:
|
|
bool playerDataExists(const std::string &playername);
|
|
};
|
|
|
|
#endif
|
|
|