OutputDB Class Reference

Public Member Functions
	OutputDB (std::string sample, const Config &config)
	~OutputDB ()
void	Put (std::string &serializeKey, std::string &serializeString)

Private Attributes
leveldb::DB *	fLevelDB
leveldb::WriteOptions	fWriteOptions
MDB_env *	mdb_env
MDB_dbi	mdb_dbi
MDB_val	mdb_key
MDB_val	mdb_data
MDB_txn *	mdb_txn

Detailed Description

Definition at line 134 of file cvnCreateDB.cc.

Constructor & Destructor Documentation

OutputDB::OutputDB	(	std::string	sample,
		const Config &	config
	)

Definition at line 152 of file cvnCreateDB.cc.

                                                         :
  fLevelDB(0),  mdb_env(0), mdb_txn(0) {

  std::string outputDir;
  if (sample=="test") 
    outputDir=config.fTestOutputDir;
  else
    outputDir=config.fTrainOutputDir;

  if (config.fOutputFormat=="LevelDB"){
    leveldb::Options fileOptions;
    fileOptions.error_if_exists   = config.fErrorIfExists;
    fileOptions.create_if_missing = config.fCreateIfMissing;
    fileOptions.write_buffer_size = config.fWriteBufferSize;

    fWriteOptions.sync = config.fWriteSync;

    if(!leveldb::DB::Open(fileOptions, outputDir, &fLevelDB).ok()) {
      std::cout << "Problem opening the database: "
                << outputDir << std::endl;
      exit(1);
    }
  }

  else if (config.fOutputFormat=="LMDB") {
    mkdir(outputDir.c_str(),0777);
    mdb_env_create(&mdb_env);
    mdb_env_set_mapsize(mdb_env, 10737418240);
    mdb_env_open(mdb_env, outputDir.c_str(), 0, 0777);
    mdb_txn_begin(mdb_env, NULL, 0, &mdb_txn);
    mdb_dbi_open(mdb_txn,NULL, 0, &mdb_dbi);
  }

  else {
    std::cout << "Unrecognized output format " << config.fOutputFormat << std::endl;
    exit(1);
  }

}

OutputDB::~OutputDB

(

)

Definition at line 192 of file cvnCreateDB.cc.

                    {
  if (mdb_txn)  mdb_txn_commit(mdb_txn);
}

Member Function Documentation

void OutputDB::Put	(	std::string &	serializeKey,
		std::string &	serializeString
	)

Definition at line 196 of file cvnCreateDB.cc.

                                                                       {
  if (fLevelDB) {
    fLevelDB->Put(fWriteOptions, serializeKey, serializeString);
  } //end if LevelDB
  else {//it must be LMDB
    mdb_data.mv_size=serializeString.size();
    mdb_data.mv_data=reinterpret_cast<void*>(&serializeString[0]);
    mdb_key.mv_size=serializeKey.size();
    mdb_key.mv_data=reinterpret_cast<void*>(&serializeKey[0]);
    if ( mdb_put(mdb_txn,mdb_dbi,&mdb_key,&mdb_data,0)!= MDB_SUCCESS){
      std::cout<< "ERROR: Events not loaded correctly" <<std::endl;
    }//end if put fails
  }//end if LMDB
}//end OutputDB::Put

Member Data Documentation

leveldb::DB* OutputDB::fLevelDB

private

Definition at line 142 of file cvnCreateDB.cc.

leveldb::WriteOptions OutputDB::fWriteOptions

private

Definition at line 143 of file cvnCreateDB.cc.

MDB_val OutputDB::mdb_data

private

Definition at line 147 of file cvnCreateDB.cc.

MDB_dbi OutputDB::mdb_dbi

private

Definition at line 146 of file cvnCreateDB.cc.

MDB_env* OutputDB::mdb_env

private

Definition at line 145 of file cvnCreateDB.cc.

MDB_val OutputDB::mdb_key

private

Definition at line 147 of file cvnCreateDB.cc.

MDB_txn* OutputDB::mdb_txn

private

Definition at line 148 of file cvnCreateDB.cc.

---

The documentation for this class was generated from the following file:

• dunereco/dunereco/CVN/levelDB/cvnCreateDB.cc