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
  26 #include "log.h"
  27 #include "filesys.h"
  28 #include "exceptions.h"
  29 #include "util/string.h"
  30
  31 #include "leveldb/db.h"
  32
  33
  34 #define ENSURE_STATUS_OK(s) \
  35         if (!(s).ok()) { \
  36                 throw DatabaseException(std::string("LevelDB error: ") + \
  37                                 (s).ToString()); \
  38         }
  39
  40
  41 Database_LevelDB::Database_LevelDB(const std::string &savedir)
  42 {
  43         leveldb::Options options;
  44         options.create_if_missing = true;
  45         leveldb::Status status = leveldb::DB::Open(options,
  46                 savedir + DIR_DELIM + "map.db", &m_database);
  47         ENSURE_STATUS_OK(status);
  48 }
  49
  50 Database_LevelDB::~Database_LevelDB()
  51 {
  52         delete m_database;
  53 }
  54
  55 bool Database_LevelDB::saveBlock(const v3s16 &pos, const std::string &data)
  56 {
  57         leveldb::Status status = m_database->Put(leveldb::WriteOptions(),
  58                         i64tos(getBlockAsInteger(pos)), data);
  59         if (!status.ok()) {
  60                 warningstream << "saveBlock: LevelDB error saving block "
  61                         << PP(pos) << ": " << status.ToString() << std::endl;
  62                 return false;
  63         }
  64
  65         return true;
  66 }
  67
  68 void Database_LevelDB::loadBlock(const v3s16 &pos, std::string *block)
  69 {
  70         std::string datastr;
  71         leveldb::Status status = m_database->Get(leveldb::ReadOptions(),
  72                 i64tos(getBlockAsInteger(pos)), &datastr);
  73
  74         *block = (status.ok()) ? datastr : "";
  75 }
  76
  77 bool Database_LevelDB::deleteBlock(const v3s16 &pos)
  78 {
  79         leveldb::Status status = m_database->Delete(leveldb::WriteOptions(),
  80                         i64tos(getBlockAsInteger(pos)));
  81         if (!status.ok()) {
  82                 warningstream << "deleteBlock: LevelDB error deleting block "
  83                         << PP(pos) << ": " << status.ToString() << std::endl;
  84                 return false;
  85         }
  86
  87         return true;
  88 }
  89
  90 void Database_LevelDB::listAllLoadableBlocks(std::vector<v3s16> &dst)
  91 {
  92         leveldb::Iterator* it = m_database->NewIterator(leveldb::ReadOptions());
  93         for (it->SeekToFirst(); it->Valid(); it->Next()) {
  94                 dst.push_back(getIntegerAsBlock(stoi64(it->key().ToString())));
  95         }
  96         ENSURE_STATUS_OK(it->status());  // Check for any errors found during the scan
  97         delete it;
  98 }
  99
 100 #endif // USE_LEVELDB
 101