rocksdb.c

Go to the documentation of this file.

 
#include "config.h"
#include <stddef.h>
#include <rocksdb/c.h>
#include "mutt/lib.h"
#include "lib.h"
 
struct RocksDbCtx
{
  rocksdb_t *db;
  rocksdb_options_t *options;
  rocksdb_readoptions_t *read_options;
  rocksdb_writeoptions_t *write_options;
  char *err;
};
 
static void *store_rocksdb_open(const char *path)
{
  if (!path)
    return NULL;
 
  struct RocksDbCtx *ctx = mutt_mem_malloc(sizeof(struct RocksDbCtx));
 
  /* RocksDB store errors in form of strings */
  ctx->err = NULL;
 
  /* setup generic options, create new db and limit log to one file */
  ctx->options = rocksdb_options_create();
  rocksdb_options_set_create_if_missing(ctx->options, 1);
  rocksdb_options_set_keep_log_file_num(ctx->options, 1);
 
  /* setup read options, we verify with checksums */
  ctx->read_options = rocksdb_readoptions_create();
  rocksdb_readoptions_set_verify_checksums(ctx->read_options, 1);
 
  /* setup write options, no sync needed, disable WAL */
  ctx->write_options = rocksdb_writeoptions_create();
  rocksdb_writeoptions_set_sync(ctx->write_options, 0);
  rocksdb_writeoptions_disable_WAL(ctx->write_options, 1);
 
  rocksdb_options_set_compression(ctx->options, rocksdb_no_compression);
 
  /* open database and check for error in ctx->error */
  ctx->db = rocksdb_open(ctx->options, path, &ctx->err);
  if (ctx->err)
  {
    rocksdb_free(ctx->err);
    FREE(&ctx);
    return NULL;
  }
 
  return ctx;
}
 
static void *store_rocksdb_fetch(void *store, const char *key, size_t klen, size_t *vlen)
{
  if (!store)
    return NULL;
 
  struct RocksDbCtx *ctx = store;
 
  void *rv = rocksdb_get(ctx->db, ctx->read_options, key, klen, vlen, &ctx->err);
  if (ctx->err)
  {
    rocksdb_free(ctx->err);
    ctx->err = NULL;
    return NULL;
  }
 
  return rv;
}
 
static void store_rocksdb_free(void *store, void **ptr)
{
  FREE(ptr);
}
 
static int store_rocksdb_store(void *store, const char *key, size_t klen,
                               void *value, size_t vlen)
{
  if (!store)
    return -1;
 
  struct RocksDbCtx *ctx = store;
 
  rocksdb_put(ctx->db, ctx->write_options, key, klen, value, vlen, &ctx->err);
  if (ctx->err)
  {
    rocksdb_free(ctx->err);
    ctx->err = NULL;
    return -1;
  }
 
  return 0;
}
 
static int store_rocksdb_delete_record(void *store, const char *key, size_t klen)
{
  if (!store)
    return -1;
 
  struct RocksDbCtx *ctx = store;
 
  rocksdb_delete(ctx->db, ctx->write_options, key, klen, &ctx->err);
  if (ctx->err)
  {
    rocksdb_free(ctx->err);
    ctx->err = NULL;
    return -1;
  }
 
  return 0;
}
 
static void store_rocksdb_close(void **ptr)
{
  if (!ptr || !*ptr)
    return;
 
  struct RocksDbCtx *ctx = *ptr;
 
  /* close database and free resources */
  rocksdb_close(ctx->db);
  rocksdb_options_destroy(ctx->options);
  rocksdb_readoptions_destroy(ctx->read_options);
  rocksdb_writeoptions_destroy(ctx->write_options);
 
  FREE(ptr);
  *ptr = NULL;
}
 
static const char *store_rocksdb_version(void)
{
/* return sth. like "RocksDB 6.7.3" */
#define RDBVER(major, minor, patch) #major "." #minor "." #patch
  return "RocksDB " RDBVER(ROCKSDB_MAJOR, ROCKSDB_MINOR, ROCKSDB_PATCH);
}
 
STORE_BACKEND_OPS(rocksdb)