123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146 |
- /*
- 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 2.1 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.
- */
- #pragma once
- #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 = nullptr;
- int m_pgversion = 0;
- };
- class MapDatabasePostgreSQL : private Database_PostgreSQL, public MapDatabase
- {
- public:
- MapDatabasePostgreSQL(const std::string &connect_string);
- virtual ~MapDatabasePostgreSQL() = default;
- 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() = default;
- 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);
- };
|