Checking in some results with a smaller, slower disk.
[bluesky.git] / kvstore / backend.h
1 #ifndef _BACKEND_H_
2 #define _BACKEND_H_ 1
3
4 extern "C"
5 {
6 #include <db.h>
7 }
8
9 #include <string>
10 #include <map>
11
12 #include <boost/shared_ptr.hpp>
13
14 #include "util.h"
15
16 using namespace std;
17 using boost::shared_ptr;
18
19 namespace kvstore
20 {
21
22 class Backend
23 {
24 public:
25     virtual ~Backend() {};
26
27     virtual bool Put(const string &key,
28                      const string &value) = 0;
29
30     virtual bool Get(const string &key,
31                      string *value) = 0;
32
33 };
34
35 class MemoryBackend : public Backend
36 {
37 public:
38     virtual ~MemoryBackend();
39
40     virtual bool Put(const string &key,
41                      const string &value);
42
43     virtual bool Get(const string &key,
44                      string *value);
45
46 private:
47     boost::mutex _lock;
48     typedef map<string, string> map_t;
49     map_t _map;
50 };
51
52 class BDBBackend : public Backend
53 {
54 public:
55     BDBBackend(const vector<string> &paths,
56                bool flush=true,
57                bool log_in_memory = false);
58
59     virtual ~BDBBackend();
60     virtual bool Put(const string &key,
61                      const string &value);
62
63     virtual bool Get(const string &key,
64                      string *value);
65
66 private:
67     class BDBDatabase;
68
69     inline size_t LookupKeyDB(const string &key);
70
71     vector< shared_ptr<BDBDatabase> > _dbs;
72 };
73
74 } // namespace kvstore
75
76 #endif