lmdb-safe/scale-example.cc

#include "lmdb-safe.hh"


struct MDBVal
{
  MDBVal(unsigned int v) : d_v(v)
  {
    d_mdbval.mv_size = sizeof(d_v);
    d_mdbval.mv_data = &d_v;
  }
  operator const MDB_val&()
  {
    return d_mdbval;
  }
  unsigned int d_v;
  MDB_val d_mdbval;
};


int main(int argc, char** argv)
{
  auto env = getMDBEnv("./database", 0, 0600);
  auto dbi = env->openDB(0, MDB_CREATE | MDB_INTEGERKEY);
  auto txn = env->getRWTransaction();

  unsigned int limit=20000000;
  if(argc > 1)
    limit = atoi(argv[1]);
  
  cout<<"Counting records.. "; cout.flush();
  auto cursor=txn.getCursor(dbi);
  MDBOutVal key, data;
  int count=0;
  while(!cursor.get(key, data, count ? MDB_NEXT : MDB_FIRST)) {
    auto d = data.get<unsigned long>();
    if(d==17)
      cout <<"Got 17!"<<endl;
    count++;
  }
  cout<<"Have "<<count<<"!"<<endl;
  
  cout<<"Clearing records.. "; cout.flush();
  mdb_drop(txn, dbi, 0); // clear records
  cout<<"Done!"<<endl;

  cout << "Adding "<<limit<<" values  .. "; cout.flush();
  for(unsigned long n = 0 ; n < limit; ++n) {
    txn.put(dbi, n, n, MDB_APPEND);
  }
  cout <<"Done!"<<endl;
  cout <<"Calling commit.. "; cout.flush();
  txn.commit();
  cout<<"Done!"<<endl;
}