3 Copyright (C) 2013 celeron55, Perttu Ahola <celeron55@gmail.com>
5 This program is free software; you can redistribute it and/or modify
6 it under the terms of the GNU Lesser General Public License as published by
7 the Free Software Foundation; either version 2.1 of the License, or
8 (at your option) any later version.
10 This program is distributed in the hope that it will be useful,
11 but WITHOUT ANY WARRANTY; without even the implied warranty of
12 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13 GNU Lesser General Public License for more details.
15 You should have received a copy of the GNU Lesser General Public License along
16 with this program; if not, write to the Free Software Foundation, Inc.,
17 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
21 SQLite format specification:
28 #include "database-sqlite3.h"
32 #include "exceptions.h"
34 #include "util/string.h"
39 #define SQLRES(s, r) \
41 throw FileNotGoodException(std::string(\
42 "SQLite3 database error (" \
43 __FILE__ ":" TOSTRING(__LINE__) \
45 sqlite3_errmsg(m_database)); \
47 #define SQLOK(s) SQLRES(s, SQLITE_OK)
49 #define PREPARE_STATEMENT(name, query) \
50 SQLOK(sqlite3_prepare_v2(m_database, query, -1, &m_stmt_##name, NULL))
52 #define FINALIZE_STATEMENT(statement) \
53 if (sqlite3_finalize(statement) != SQLITE_OK) { \
54 throw FileNotGoodException(std::string( \
55 "SQLite3: Failed to finalize " #statement ": ") + \
56 sqlite3_errmsg(m_database)); \
60 Database_SQLite3::Database_SQLite3(const std::string &savedir) :
71 void Database_SQLite3::beginSave() {
73 SQLRES(sqlite3_step(m_stmt_begin), SQLITE_DONE);
74 sqlite3_reset(m_stmt_begin);
77 void Database_SQLite3::endSave() {
79 SQLRES(sqlite3_step(m_stmt_end), SQLITE_DONE);
80 sqlite3_reset(m_stmt_end);
83 void Database_SQLite3::openDatabase()
85 if (m_database) return;
87 std::string dbp = m_savedir + DIR_DELIM + "map.sqlite";
89 // Open the database connection
91 if (!fs::CreateAllDirs(m_savedir)) {
92 infostream << "Database_SQLite3: Failed to create directory \""
93 << m_savedir << "\"" << std::endl;
94 throw FileNotGoodException("Failed to create database "
98 bool needs_create = !fs::PathExists(dbp);
100 if (sqlite3_open_v2(dbp.c_str(), &m_database,
101 SQLITE_OPEN_READWRITE | SQLITE_OPEN_CREATE,
102 NULL) != SQLITE_OK) {
103 errorstream << "SQLite3 database failed to open: "
104 << sqlite3_errmsg(m_database) << std::endl;
105 throw FileNotGoodException("Cannot open database file");
112 std::string query_str = std::string("PRAGMA synchronous = ")
113 + itos(g_settings->getU16("sqlite_synchronous"));
114 SQLOK(sqlite3_exec(m_database, query_str.c_str(), NULL, NULL, NULL));
117 void Database_SQLite3::verifyDatabase()
119 if (m_initialized) return;
123 PREPARE_STATEMENT(begin, "BEGIN");
124 PREPARE_STATEMENT(end, "COMMIT");
125 PREPARE_STATEMENT(read, "SELECT `data` FROM `blocks` WHERE `pos` = ? LIMIT 1");
127 PREPARE_STATEMENT(write, "INSERT INTO `blocks` (`pos`, `data`) VALUES (?, ?)");
129 PREPARE_STATEMENT(write, "REPLACE INTO `blocks` (`pos`, `data`) VALUES (?, ?)");
131 PREPARE_STATEMENT(delete, "DELETE FROM `blocks` WHERE `pos` = ?");
132 PREPARE_STATEMENT(list, "SELECT `pos` FROM `blocks`");
134 m_initialized = true;
136 verbosestream << "ServerMap: SQLite3 database opened." << std::endl;
139 inline void Database_SQLite3::bindPos(sqlite3_stmt *stmt, const v3s16 &pos, int index)
141 SQLOK(sqlite3_bind_int64(stmt, index, getBlockAsInteger(pos)));
144 bool Database_SQLite3::deleteBlock(const v3s16 &pos)
148 bindPos(m_stmt_delete, pos);
150 bool good = sqlite3_step(m_stmt_delete) == SQLITE_DONE;
151 sqlite3_reset(m_stmt_delete);
154 errorstream << "WARNING: deleteBlock: Block failed to delete "
155 << PP(pos) << ": " << sqlite3_errmsg(m_database) << std::endl;
160 bool Database_SQLite3::saveBlock(const v3s16 &pos, const std::string &data)
166 * Note: For some unknown reason SQLite3 fails to REPLACE blocks on Android,
167 * deleting them and then inserting works.
169 bindPos(m_stmt_read, pos);
171 if (sqlite3_step(m_stmt_read) == SQLITE_ROW) {
174 sqlite3_reset(m_stmt_read);
177 bindPos(m_stmt_write, pos);
178 SQLOK(sqlite3_bind_blob(m_stmt_write, 2, data.data(), data.size(), NULL));
180 SQLRES(sqlite3_step(m_stmt_write), SQLITE_DONE)
181 sqlite3_reset(m_stmt_write);
186 std::string Database_SQLite3::loadBlock(const v3s16 &pos)
190 bindPos(m_stmt_read, pos);
192 if (sqlite3_step(m_stmt_read) != SQLITE_ROW) {
193 sqlite3_reset(m_stmt_read);
196 const char *data = (const char *) sqlite3_column_blob(m_stmt_read, 0);
197 size_t len = sqlite3_column_bytes(m_stmt_read, 0);
201 s = std::string(data, len);
203 sqlite3_step(m_stmt_read);
204 // We should never get more than 1 row, so ok to reset
205 sqlite3_reset(m_stmt_read);
210 void Database_SQLite3::createDatabase()
212 assert(m_database); // Pre-condition
213 SQLOK(sqlite3_exec(m_database,
214 "CREATE TABLE IF NOT EXISTS `blocks` (\n"
215 " `pos` INT PRIMARY KEY,\n"
221 void Database_SQLite3::listAllLoadableBlocks(std::vector<v3s16> &dst)
225 while (sqlite3_step(m_stmt_list) == SQLITE_ROW) {
226 dst.push_back(getIntegerAsBlock(sqlite3_column_int64(m_stmt_list, 0)));
228 sqlite3_reset(m_stmt_list);
231 Database_SQLite3::~Database_SQLite3()
233 FINALIZE_STATEMENT(m_stmt_read)
234 FINALIZE_STATEMENT(m_stmt_write)
235 FINALIZE_STATEMENT(m_stmt_list)
236 FINALIZE_STATEMENT(m_stmt_begin)
237 FINALIZE_STATEMENT(m_stmt_end)
238 FINALIZE_STATEMENT(m_stmt_delete)
240 if (sqlite3_close(m_database) != SQLITE_OK) {
241 errorstream << "Database_SQLite3::~Database_SQLite3(): "
242 << "Failed to close database: "
243 << sqlite3_errmsg(m_database) << std::endl;