github.com/igggame/nebulas-go@v2.1.0+incompatible/nbre/fs/bc_storage_session.h (about)

     1  // Copyright (C) 2017 go-nebulas authors
     2  //
     3  // This file is part of the go-nebulas library.
     4  //
     5  // the go-nebulas library is free software: you can redistribute it and/or
     6  // modify
     7  // it under the terms of the GNU General Public License as published by
     8  // the Free Software Foundation, either version 3 of the License, or
     9  // (at your option) any later version.
    10  //
    11  // the go-nebulas library is distributed in the hope that it will be useful,
    12  // but WITHOUT ANY WARRANTY; without even the implied warranty of
    13  // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    14  // GNU General Public License for more details.
    15  //
    16  // You should have received a copy of the GNU General Public License
    17  // along with the go-nebulas library.  If not, see
    18  // <http://www.gnu.org/licenses/>.
    19  //
    20  
    21  #pragma once
    22  #include "common/common.h"
    23  #include "fs/rocksdb_storage.h"
    24  #include "fs/storage.h"
    25  #include "util/singleton.h"
    26  
    27  namespace neb {
    28  namespace fs {
    29  class bc_storage_session : public util::singleton<bc_storage_session> {
    30  public:
    31    bc_storage_session();
    32    ~bc_storage_session();
    33  
    34    void init(const std::string &path, enum storage_open_flag flag);
    35  
    36    bytes get_bytes(const bytes &key);
    37  
    38    template <typename FixBytes> bytes get_bytes(const FixBytes &key) {
    39      return get_bytes(from_fix_bytes(key));
    40    }
    41  
    42    inline bytes get_bytes(const std::string &key) {
    43      return get_bytes(string_to_byte(key));
    44    }
    45  
    46    inline std::string get_string(const bytes &key) {
    47      return byte_to_string(get_bytes(key));
    48    }
    49    inline std::string get_string(const std::string &key) {
    50      return byte_to_string(get_bytes(key));
    51    }
    52  
    53    void put_bytes(const bytes &key, const bytes &value);
    54    inline void put(const bytes &key, const bytes &value) {
    55      return put_bytes(key, value);
    56    }
    57    inline void put(const std::string &key, const bytes &value) {
    58      return put_bytes(string_to_byte(key), value);
    59    }
    60    inline void put(const bytes &key, const std::string &value) {
    61      return put_bytes(key, string_to_byte(value));
    62    }
    63    inline void put(const std::string &key, const std::string &value) {
    64      return put_bytes(string_to_byte(key), string_to_byte(value));
    65    }
    66  
    67  protected:
    68    std::unique_ptr<rocksdb_storage> m_storage;
    69    boost::shared_mutex m_mutex;
    70    std::string m_path;
    71    bool m_init_already;
    72    enum storage_open_flag m_open_flag;
    73  };
    74  } // namespace fs
    75  } // namespace neb