futriix/src/IStorage.h

#pragma once
#include <functional>
#include "sds.h"

class IStorageFactory
{
public:
    typedef void (*key_load_iterator)(const char *rgchKey, size_t cchKey, void *privdata);

    virtual ~IStorageFactory() {}
    virtual class IStorage *create(int db, key_load_iterator itr, void *privdata) = 0;
    virtual const char *name() const = 0;
    virtual size_t totalDiskspaceUsed() const = 0;
    virtual bool FSlow() const = 0;
};

class IStorage
{
public:
    typedef std::function<bool(const char *, size_t, const void *, size_t)> callback;
    typedef std::function<void(const char *, size_t, const void *, size_t)> callbackSingle;

    virtual ~IStorage();

    virtual void insert(const char *key, size_t cchKey, void *data, size_t cb, bool fOverwire) = 0;
    virtual bool erase(const char *key, size_t cchKey) = 0;
    virtual void retrieve(const char *key, size_t cchKey, callbackSingle fn) const = 0;
    virtual size_t clear() = 0;
    virtual bool enumerate(callback fn) const = 0;
    virtual size_t count() const = 0;

    virtual void bulkInsert(sds *rgkeys, sds *rgvals, size_t celem) {
        beginWriteBatch();
        for (size_t ielem = 0; ielem < celem; ++ielem) {
            insert(rgkeys[ielem], sdslen(rgkeys[ielem]), rgvals[ielem], sdslen(rgvals[ielem]), false);
        }
        endWriteBatch();
    }

    virtual void beginWriteBatch() {} // NOP
    virtual void endWriteBatch() {} // NOP

    virtual void batch_lock() {} // NOP
    virtual void batch_unlock() {} // NOP

    virtual void flush() = 0;

    /* This is permitted to be a shallow clone */
    virtual const IStorage *clone() const = 0;
};