Line data    Source code

       1             : 
       2             : /* Copyright (c) 2014, Stefan.Eilemann@epfl.ch
       3             :  *
       4             :  * This library is free software; you can redistribute it and/or modify it under
       5             :  * the terms of the GNU Lesser General Public License version 2.1 as published
       6             :  * by the Free Software Foundation.
       7             :  *
       8             :  * This library is distributed in the hope that it will be useful, but WITHOUT
       9             :  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
      10             :  * FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public License for more
      11             :  * details.
      12             :  *
      13             :  * You should have received a copy of the GNU Lesser General Public License
      14             :  * along with this library; if not, write to the Free Software Foundation, Inc.,
      15             :  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
      16             :  */
      17             : 
      18             : #ifdef LUNCHBOX_USE_LEVELDB
      19             : #include <lunchbox/log.h>
      20             : #include <lunchbox/compiler.h>
      21             : 
      22             : #include <leveldb/db.h>
      23             : 
      24             : namespace lunchbox
      25             : {
      26             : namespace db = ::leveldb;
      27             : 
      28             : namespace leveldb
      29             : {
      30             : namespace
      31             : {
      32           7 : db::DB* _open( const URI& uri )
      33             : {
      34           7 :     db::DB* db = 0;
      35           7 :     db::Options options;
      36           7 :     options.create_if_missing = true;
      37           7 :     const std::string& path = uri.getPath().empty() ?
      38           7 :                               "persistentMap.leveldb" : uri.getPath();
      39          14 :     const db::Status status = db::DB::Open( options, path, &db );
      40           7 :     if( !status.ok( ))
      41           1 :         LBTHROW( std::runtime_error( status.ToString( )));
      42          13 :     return db;
      43             : }
      44             : }
      45             : 
      46             : class PersistentMap : public detail::PersistentMap
      47             : {
      48             : public:
      49           7 :     PersistentMap( const URI& uri ) : _db( _open( uri )) {}
      50             : 
      51          12 :     virtual ~PersistentMap() { delete _db; }
      52             : 
      53           8 :     static bool handles( const URI& uri )
      54           8 :         { return uri.getScheme() == "leveldb"; }
      55             : 
      56          18 :     bool insert( const std::string& key, const void* data, const size_t size )
      57             :         final
      58             :     {
      59          18 :         const db::Slice value( (const char*)data, size );
      60          18 :         return _db->Put( db::WriteOptions(), key, value ).ok();
      61             :     }
      62             : 
      63          33 :     std::string operator [] ( const std::string& key ) const final
      64             :     {
      65          33 :         std::string value;
      66          33 :         if( _db->Get( db::ReadOptions(), key, &value ).ok( ))
      67          27 :             return value;
      68           6 :         return std::string();
      69             :     }
      70             : 
      71           0 :     bool contains( const std::string& key ) const final
      72             :     {
      73           0 :         std::string value;
      74           0 :         return _db->Get( db::ReadOptions(), key, &value ).ok();
      75             :     }
      76             : 
      77             : private:
      78             :     db::DB* const _db;
      79             : };
      80             : }
      81             : }
      82             : 
      83             : #endif