3336c4f43e
* add hash prefixing to rocksdb * don't need to offset internal keys * remove hashslot prefix when loading from rocksdb * don't prefix internal keys * implement slot to keys with rocksdb * fix compile * add hashslot enum to test storage provider * add --flash option for tests * forgot return statement in getKeysInSlot * check for count mismatch * forgot ; * add assert to ensure correctness * fix warnings * add option to show logs of 1 server at a time * add fixed length prefix extractor * add machamp * switch machamp to main branch * add machamp script
35 lines
1.2 KiB
C++
35 lines
1.2 KiB
C++
#include "../IStorage.h"
|
|
#include <string>
|
|
#include <unordered_map>
|
|
|
|
class TestStorageFactory : public IStorageFactory
|
|
{
|
|
virtual class IStorage *create(int db, key_load_iterator itr, void *privdata) override;
|
|
virtual class IStorage *createMetadataDb() override;
|
|
virtual const char *name() const override;
|
|
virtual size_t totalDiskspaceUsed() const override { return 0; }
|
|
virtual bool FSlow() const override { return false; }
|
|
};
|
|
|
|
class TestStorageProvider final : public IStorage
|
|
{
|
|
std::unordered_map<std::string, std::string> m_map;
|
|
|
|
public:
|
|
TestStorageProvider();
|
|
virtual ~TestStorageProvider();
|
|
|
|
virtual void insert(const char *key, size_t cchKey, void *data, size_t cb, bool fHintOverwrite) override;
|
|
virtual bool erase(const char *key, size_t cchKey) override;
|
|
virtual void retrieve(const char *key, size_t cchKey, callbackSingle fn) const override;
|
|
virtual size_t clear() override;
|
|
virtual bool enumerate(callback fn) const override;
|
|
virtual bool enumerate_hashslot(callback fn, unsigned int hashslot) const override;
|
|
virtual size_t count() const override;
|
|
|
|
virtual void flush() override;
|
|
|
|
/* This is permitted to be a shallow clone */
|
|
virtual const IStorage *clone() const override;
|
|
};
|