ledisdb/store/leveldb/leveldb_ext.cc

// +build leveldb

#include "leveldb_ext.h"

#include <stdlib.h>
//#include <string>

//#include "leveldb/db.h"

//using namespace leveldb;

extern "C" {

// static bool SaveError(char** errptr, const Status& s) {
//   assert(errptr != NULL);
//   if (s.ok()) {
//     return false;
//   } else if (*errptr == NULL) {
//     *errptr = strdup(s.ToString().c_str());
//   } else {
//     free(*errptr);
//     *errptr = strdup(s.ToString().c_str());
//   }
//   return true;
// }

// void* leveldb_get_ext(
//     leveldb_t* db,
//     const leveldb_readoptions_t* options,
//     const char* key, size_t keylen,
//     char** valptr,
//     size_t* vallen,
//     char** errptr) {

//     std::string *tmp = new(std::string);

//     //very tricky, maybe changed with c++ leveldb upgrade
//     Status s = (*(DB**)db)->Get(*(ReadOptions*)options, Slice(key, keylen), tmp);

//     if (s.ok()) {
//         *valptr = (char*)tmp->data();
//         *vallen = tmp->size();
//     } else {
//         delete(tmp);
//         tmp = NULL;
//         *valptr = NULL;
//         *vallen = 0;
//         if (!s.IsNotFound()) {
//             SaveError(errptr, s);
//         }
//     }
//     return tmp;
// }

// void leveldb_get_free_ext(void* context) {
//     std::string* s = (std::string*)context;

//     delete(s);
// }


unsigned char leveldb_iter_seek_to_first_ext(leveldb_iterator_t* iter) {
    leveldb_iter_seek_to_first(iter);
    return leveldb_iter_valid(iter);
}

unsigned char leveldb_iter_seek_to_last_ext(leveldb_iterator_t* iter) {
    leveldb_iter_seek_to_last(iter);
    return leveldb_iter_valid(iter);
}

unsigned char leveldb_iter_seek_ext(leveldb_iterator_t* iter, const char* k, size_t klen) {
    leveldb_iter_seek(iter, k, klen);
    return leveldb_iter_valid(iter);
}

unsigned char leveldb_iter_next_ext(leveldb_iterator_t* iter) {
    leveldb_iter_next(iter);
    return leveldb_iter_valid(iter);
}

unsigned char leveldb_iter_prev_ext(leveldb_iterator_t* iter) {
    leveldb_iter_prev(iter);
    return leveldb_iter_valid(iter);
}

extern void leveldb_writebatch_iterate_put(void*, const char* k, size_t klen, const char* v, size_t vlen);
extern void leveldb_writebatch_iterate_delete(void*, const char* k, size_t klen);

void leveldb_writebatch_iterate_ext(leveldb_writebatch_t* w, void *p) {
    leveldb_writebatch_iterate(w, p, 
        leveldb_writebatch_iterate_put, leveldb_writebatch_iterate_delete);
}

}