Skip to content

Commit

Permalink
DB::loadBlock copy removal & DB backend cleanup
Browse files Browse the repository at this point in the history
* Remove the copy from db::loadBlock by using a pointer to the destination
* cleanup db backend, the child backend doesn't have to set their functions as virtual
  • Loading branch information
nerzhul committed May 17, 2016
1 parent decbd39 commit 1434014
Show file tree
Hide file tree
Showing 11 changed files with 46 additions and 46 deletions.
11 changes: 7 additions & 4 deletions src/database-dummy.cpp
Expand Up @@ -30,13 +30,16 @@ bool Database_Dummy::saveBlock(const v3s16 &pos, const std::string &data)
return true;
}

std::string Database_Dummy::loadBlock(const v3s16 &pos)
void Database_Dummy::loadBlock(const v3s16 &pos, std::string *block)
{
s64 i = getBlockAsInteger(pos);
std::map<s64, std::string>::iterator it = m_database.find(i);
if (it == m_database.end())
return "";
return it->second;
if (it == m_database.end()) {
*block = "";
return;
}

*block = it->second;
}

bool Database_Dummy::deleteBlock(const v3s16 &pos)
Expand Down
8 changes: 4 additions & 4 deletions src/database-dummy.h
Expand Up @@ -28,10 +28,10 @@ with this program; if not, write to the Free Software Foundation, Inc.,
class Database_Dummy : public Database
{
public:
virtual bool saveBlock(const v3s16 &pos, const std::string &data);
virtual std::string loadBlock(const v3s16 &pos);
virtual bool deleteBlock(const v3s16 &pos);
virtual void listAllLoadableBlocks(std::vector<v3s16> &dst);
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);

private:
std::map<s64, std::string> m_database;
Expand Down
7 changes: 2 additions & 5 deletions src/database-leveldb.cpp
Expand Up @@ -65,16 +65,13 @@ bool Database_LevelDB::saveBlock(const v3s16 &pos, const std::string &data)
return true;
}

std::string Database_LevelDB::loadBlock(const v3s16 &pos)
void Database_LevelDB::loadBlock(const v3s16 &pos, std::string *block)
{
std::string datastr;
leveldb::Status status = m_database->Get(leveldb::ReadOptions(),
i64tos(getBlockAsInteger(pos)), &datastr);

if(status.ok())
return datastr;
else
return "";
*block = (status.ok()) ? datastr : "";
}

bool Database_LevelDB::deleteBlock(const v3s16 &pos)
Expand Down
8 changes: 4 additions & 4 deletions src/database-leveldb.h
Expand Up @@ -34,10 +34,10 @@ class Database_LevelDB : public Database
Database_LevelDB(const std::string &savedir);
~Database_LevelDB();

virtual bool saveBlock(const v3s16 &pos, const std::string &data);
virtual std::string loadBlock(const v3s16 &pos);
virtual bool deleteBlock(const v3s16 &pos);
virtual void listAllLoadableBlocks(std::vector<v3s16> &dst);
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);

private:
leveldb::DB *m_database;
Expand Down
11 changes: 7 additions & 4 deletions src/database-redis.cpp
Expand Up @@ -101,7 +101,7 @@ bool Database_Redis::saveBlock(const v3s16 &pos, const std::string &data)
return true;
}

std::string Database_Redis::loadBlock(const v3s16 &pos)
void Database_Redis::loadBlock(const v3s16 &pos, std::string *block)
{
std::string tmp = i64tos(getBlockAsInteger(pos));
redisReply *reply = static_cast<redisReply *>(redisCommand(ctx,
Expand All @@ -111,12 +111,13 @@ std::string Database_Redis::loadBlock(const v3s16 &pos)
throw FileNotGoodException(std::string(
"Redis command 'HGET %s %s' failed: ") + ctx->errstr);
}

switch (reply->type) {
case REDIS_REPLY_STRING: {
std::string str(reply->str, reply->len);
*block = std::string(reply->str, reply->len);
// std::string copies the memory so this won't cause any problems
freeReplyObject(reply);
return str;
return;
}
case REDIS_REPLY_ERROR: {
std::string errstr(reply->str, reply->len);
Expand All @@ -127,11 +128,13 @@ std::string Database_Redis::loadBlock(const v3s16 &pos)
"Redis command 'HGET %s %s' errored: ") + errstr);
}
case REDIS_REPLY_NIL: {
*block = "";
// block not found in database
freeReplyObject(reply);
return "";
return;
}
}

errorstream << "loadBlock: loading block " << PP(pos)
<< " returned invalid reply type " << reply->type
<< ": " << std::string(reply->str, reply->len) << std::endl;
Expand Down
12 changes: 6 additions & 6 deletions src/database-redis.h
Expand Up @@ -36,13 +36,13 @@ class Database_Redis : public Database
Database_Redis(Settings &conf);
~Database_Redis();

virtual void beginSave();
virtual void endSave();
void beginSave();
void endSave();

virtual bool saveBlock(const v3s16 &pos, const std::string &data);
virtual std::string loadBlock(const v3s16 &pos);
virtual bool deleteBlock(const v3s16 &pos);
virtual void listAllLoadableBlocks(std::vector<v3s16> &dst);
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);

private:
redisContext *ctx;
Expand Down
11 changes: 4 additions & 7 deletions src/database-sqlite3.cpp
Expand Up @@ -237,28 +237,25 @@ bool Database_SQLite3::saveBlock(const v3s16 &pos, const std::string &data)
return true;
}

std::string Database_SQLite3::loadBlock(const v3s16 &pos)
void Database_SQLite3::loadBlock(const v3s16 &pos, std::string *block)
{
verifyDatabase();

bindPos(m_stmt_read, pos);

if (sqlite3_step(m_stmt_read) != SQLITE_ROW) {
sqlite3_reset(m_stmt_read);
return "";
return;
}

const char *data = (const char *) sqlite3_column_blob(m_stmt_read, 0);
size_t len = sqlite3_column_bytes(m_stmt_read, 0);

std::string s;
if (data)
s = std::string(data, len);
*block = (data) ? std::string(data, len) : "";

sqlite3_step(m_stmt_read);
// We should never get more than 1 row, so ok to reset
sqlite3_reset(m_stmt_read);

return s;
}

void Database_SQLite3::createDatabase()
Expand Down
16 changes: 8 additions & 8 deletions src/database-sqlite3.h
Expand Up @@ -31,16 +31,16 @@ class Database_SQLite3 : public Database
{
public:
Database_SQLite3(const std::string &savedir);
~Database_SQLite3();

virtual void beginSave();
virtual void endSave();
void beginSave();
void endSave();

virtual bool saveBlock(const v3s16 &pos, const std::string &data);
virtual std::string loadBlock(const v3s16 &pos);
virtual bool deleteBlock(const v3s16 &pos);
virtual void listAllLoadableBlocks(std::vector<v3s16> &dst);
virtual bool initialized() const { return m_initialized; }
~Database_SQLite3();
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);
bool initialized() const { return m_initialized; }

private:
// Open the database
Expand Down
2 changes: 1 addition & 1 deletion src/database.h
Expand Up @@ -38,7 +38,7 @@ class Database
virtual void endSave() {}

virtual bool saveBlock(const v3s16 &pos, const std::string &data) = 0;
virtual std::string loadBlock(const v3s16 &pos) = 0;
virtual void loadBlock(const v3s16 &pos, std::string *block) = 0;
virtual bool deleteBlock(const v3s16 &pos) = 0;

static s64 getBlockAsInteger(const v3s16 &pos);
Expand Down
3 changes: 2 additions & 1 deletion src/main.cpp
Expand Up @@ -948,7 +948,8 @@ static bool migrate_database(const GameParams &game_params, const Settings &cmd_
for (std::vector<v3s16>::const_iterator it = blocks.begin(); it != blocks.end(); ++it) {
if (kill) return false;

const std::string &data = old_db->loadBlock(*it);
std::string data;
old_db->loadBlock(*it, &data);
if (!data.empty()) {
new_db->saveBlock(*it, data);
} else {
Expand Down
3 changes: 1 addition & 2 deletions src/map.cpp
Expand Up @@ -3442,8 +3442,7 @@ MapBlock* ServerMap::loadBlock(v3s16 blockpos)
v2s16 p2d(blockpos.X, blockpos.Z);

std::string ret;

ret = dbase->loadBlock(blockpos);
dbase->loadBlock(blockpos, &ret);
if (ret != "") {
loadBlock(&ret, blockpos, createSector(p2d), false);
return getBlockNoCreateNoEx(blockpos);
Expand Down

0 comments on commit 1434014

Please sign in to comment.