3 Copyright (C) 2014 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.
28 #include "database-redis.h"
32 #include "mapsector.h"
34 #include "serialization.h"
39 Database_Redis::Database_Redis(ServerMap *map, std::string savedir)
42 conf.readConfigFile((std::string(savedir) + DIR_DELIM + "world.mt").c_str());
45 tmp = conf.get("redis_address");
46 hash = conf.get("redis_hash");
47 } catch(SettingNotFoundException e) {
48 throw SettingNotFoundException("Set redis_address and redis_hash in world.mt to use the redis backend");
50 const char *addr = tmp.c_str();
51 int port = conf.exists("redis_port") ? conf.getU16("redis_port") : 6379;
52 ctx = redisConnect(addr, port);
54 throw FileNotGoodException("Cannot allocate redis context");
56 std::string err = std::string("Connection error: ") + ctx->errstr;
58 throw FileNotGoodException(err);
63 int Database_Redis::Initialized(void)
68 void Database_Redis::beginSave() {
70 reply = (redisReply*) redisCommand(ctx, "MULTI");
72 throw FileNotGoodException(std::string("redis command 'MULTI' failed: ") + ctx->errstr);
73 freeReplyObject(reply);
76 void Database_Redis::endSave() {
78 reply = (redisReply*) redisCommand(ctx, "EXEC");
80 throw FileNotGoodException(std::string("redis command 'EXEC' failed: ") + ctx->errstr);
81 freeReplyObject(reply);
84 bool Database_Redis::saveBlock(MapBlock *block)
86 DSTACK(__FUNCTION_NAME);
88 v3s16 p3d = block->getPos();
91 Dummy blocks are not written
95 errorstream << "WARNING: saveBlock: Not writing dummy block "
96 << PP(p3d) << std::endl;
100 // Format used for writing
101 u8 version = SER_FMT_VER_HIGHEST_WRITE;
104 [0] u8 serialization version
108 std::ostringstream o(std::ios_base::binary);
109 o.write((char*)&version, 1);
111 block->serialize(o, version, true);
112 // Write block to database
113 std::string tmp1 = o.str();
114 std::string tmp2 = i64tos(getBlockAsInteger(p3d));
116 redisReply *reply = (redisReply *)redisCommand(ctx, "HSET %s %s %b",
117 hash.c_str(), tmp2.c_str(), tmp1.c_str(), tmp1.size());
119 errorstream << "WARNING: saveBlock: redis command 'HSET' failed on "
120 "block " << PP(p3d) << ": " << ctx->errstr << std::endl;
121 freeReplyObject(reply);
125 if (reply->type == REDIS_REPLY_ERROR) {
126 errorstream << "WARNING: saveBlock: save block " << PP(p3d)
127 << "failed" << std::endl;
128 freeReplyObject(reply);
132 // We just wrote it to the disk so clear modified flag
133 block->resetModified();
134 freeReplyObject(reply);
138 MapBlock* Database_Redis::loadBlock(v3s16 blockpos)
140 v2s16 p2d(blockpos.X, blockpos.Z);
142 std::string tmp = i64tos(getBlockAsInteger(blockpos));
144 reply = (redisReply*) redisCommand(ctx, "HGET %s %s", hash.c_str(), tmp.c_str());
146 throw FileNotGoodException(std::string("redis command 'HGET %s %s' failed: ") + ctx->errstr);
148 if (reply->type == REDIS_REPLY_STRING && reply->len == 0) {
149 freeReplyObject(reply);
150 errorstream << "Blank block data in database (reply->len == 0) ("
151 << blockpos.X << "," << blockpos.Y << "," << blockpos.Z << ")" << std::endl;
153 if (g_settings->getBool("ignore_world_load_errors")) {
154 errorstream << "Ignoring block load error. Duck and cover! "
155 << "(ignore_world_load_errors)" << std::endl;
157 throw SerializationError("Blank block data in database");
162 if (reply->type == REDIS_REPLY_STRING) {
164 Make sure sector is loaded
166 MapSector *sector = srvmap->createSector(p2d);
169 std::istringstream is(std::string(reply->str, reply->len), std::ios_base::binary);
170 freeReplyObject(reply); // std::string copies the memory so we can already do this here
171 u8 version = SER_FMT_VER_INVALID;
172 is.read((char *)&version, 1);
175 throw SerializationError("ServerMap::loadBlock(): Failed"
176 " to read MapBlock version");
178 MapBlock *block = NULL;
179 bool created_new = false;
180 block = sector->getBlockNoCreateNoEx(blockpos.Y);
183 block = sector->createBlankBlockNoInsert(blockpos.Y);
188 block->deSerialize(is, version, true);
190 // If it's a new block, insert it to the map
192 sector->insertBlock(block);
194 // We just loaded it from, so it's up-to-date.
195 block->resetModified();
197 catch (SerializationError &e)
199 errorstream << "Invalid block data in database"
200 << " (" << blockpos.X << "," << blockpos.Y << "," << blockpos.Z
201 << ") (SerializationError): " << e.what() << std::endl;
202 // TODO: Block should be marked as invalid in memory so that it is
203 // not touched but the game can run
205 if (g_settings->getBool("ignore_world_load_errors")) {
206 errorstream << "Ignoring block load error. Duck and cover! "
207 << "(ignore_world_load_errors)" << std::endl;
209 throw SerializationError("Invalid block data in database");
213 return srvmap->getBlockNoCreateNoEx(blockpos); // should not be using this here
218 void Database_Redis::listAllLoadableBlocks(std::list<v3s16> &dst)
221 reply = (redisReply*) redisCommand(ctx, "HKEYS %s", hash.c_str());
223 throw FileNotGoodException(std::string("redis command 'HKEYS %s' failed: ") + ctx->errstr);
224 if(reply->type != REDIS_REPLY_ARRAY)
225 throw FileNotGoodException("Failed to get keys from database");
226 for(size_t i = 0; i < reply->elements; i++)
228 assert(reply->element[i]->type == REDIS_REPLY_STRING);
229 dst.push_back(getIntegerAsBlock(stoi64(reply->element[i]->str)));
231 freeReplyObject(reply);
234 Database_Redis::~Database_Redis()