src/database-leveldb.cpp

   1 /*
   2 Minetest
   3 Copyright (C) 2013 celeron55, Perttu Ahola <celeron55@gmail.com>
   4
   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.
   9
  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.
  14
  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.
  18 */
  19
  20 #include "config.h"
  21
  22 #if USE_LEVELDB
  23
  24 #include "database-leveldb.h"
  25 #include "leveldb/db.h"
  26
  27 #include "map.h"
  28 #include "mapsector.h"
  29 #include "mapblock.h"
  30 #include "serialization.h"
  31 #include "main.h"
  32 #include "settings.h"
  33 #include "log.h"
  34 #include "filesys.h"
  35
  36 #define ENSURE_STATUS_OK(s) \
  37         if (!(s).ok()) { \
  38                 throw FileNotGoodException(std::string("LevelDB error: ") + (s).ToString()); \
  39         }
  40
  41 Database_LevelDB::Database_LevelDB(ServerMap *map, std::string savedir)
  42 {
  43         leveldb::Options options;
  44         options.create_if_missing = true;
  45         leveldb::Status status = leveldb::DB::Open(options, savedir + DIR_DELIM + "map.db", &m_database);
  46         ENSURE_STATUS_OK(status);
  47         srvmap = map;
  48 }
  49
  50 int Database_LevelDB::Initialized(void)
  51 {
  52         return 1;
  53 }
  54
  55 void Database_LevelDB::beginSave() {}
  56 void Database_LevelDB::endSave() {}
  57
  58 bool Database_LevelDB::saveBlock(v3s16 blockpos, std::string &data)
  59 {
  60         leveldb::Status status = m_database->Put(leveldb::WriteOptions(),
  61                         i64tos(getBlockAsInteger(blockpos)), data);
  62         if (!status.ok()) {
  63                 errorstream << "WARNING: saveBlock: LevelDB error saving block "
  64                         << PP(blockpos) << ": " << status.ToString() << std::endl;
  65                 return false;
  66         }
  67
  68         return true;
  69 }
  70
  71 std::string Database_LevelDB::loadBlock(v3s16 blockpos)
  72 {
  73         std::string datastr;
  74         leveldb::Status status = m_database->Get(leveldb::ReadOptions(),
  75                 i64tos(getBlockAsInteger(blockpos)), &datastr);
  76
  77         if(status.ok())
  78                 return datastr;
  79         else
  80                 return "";
  81 }
  82
  83 bool Database_LevelDB::deleteBlock(v3s16 blockpos)
  84 {
  85         leveldb::Status status = m_database->Delete(leveldb::WriteOptions(),
  86                         i64tos(getBlockAsInteger(blockpos)));
  87         if (!status.ok()) {
  88                 errorstream << "WARNING: deleteBlock: LevelDB error deleting block "
  89                         << PP(blockpos) << ": " << status.ToString() << std::endl;
  90                 return false;
  91         }
  92
  93         return true;
  94 }
  95
  96 void Database_LevelDB::listAllLoadableBlocks(std::vector<v3s16> &dst)
  97 {
  98         leveldb::Iterator* it = m_database->NewIterator(leveldb::ReadOptions());
  99         for (it->SeekToFirst(); it->Valid(); it->Next()) {
 100                 dst.push_back(getIntegerAsBlock(stoi64(it->key().ToString())));
 101         }
 102         ENSURE_STATUS_OK(it->status());  // Check for any errors found during the scan
 103         delete it;
 104 }
 105
 106 Database_LevelDB::~Database_LevelDB()
 107 {
 108         delete m_database;
 109 }
 110 #endif