blob: 58889dd94b46907ded7cf9b4c799273d6d62344b (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
|
#include "db.h"
#include <iomanip>
#include <iostream>
#include <mutex>
#include <unordered_map>
#include <sstream>
static std::unordered_map<std::string,leveldb::DB*> dbs;
static std::mutex getDBmutex;
static bool sensor_name_is_sane(std::string& name) {
for (auto it = name.begin(); it != name.end(); ++it) {
if (not ((*it >= '0' and *it <= '9') or
(*it >= 'A' and *it <= 'Z') or
(*it >= 'a' and *it <= 'z') or
(*it == '.'))) {
return false;
}
}
return true;
}
std::string make_key(const uint64_t timestamp) {
std::stringstream key;
key << "ts-";
key << std::setfill('0') << std::setw(20) << timestamp;
return key.str();
}
leveldb::DB *getDB(std::string& name) {
getDBmutex.lock();
if (dbs.find(name) == dbs.end()) {
if (not sensor_name_is_sane(name)) {
getDBmutex.unlock();
return nullptr;
}
leveldb::DB *db;
leveldb::Options options;
options.create_if_missing = true;
leveldb::Status status = leveldb::DB::Open(options, "data/"+name, &db);
if (not status.ok()) {
std::cout << status.ToString() << std::endl;
getDBmutex.unlock();
return nullptr;
}
dbs[name] = db;
getDBmutex.unlock();
return db;
} else {
getDBmutex.unlock();
return dbs.at(name);
}
}
bool db_insert(std::string& name, const uint64_t timestamp, std::string& value) {
leveldb::DB *db = getDB(name);
if (db == nullptr) return false;
auto status = db->Put(leveldb::WriteOptions(), make_key(timestamp), value);
return status.ok();
}
void closeDB() {
std::cout << "Close Databases: ";
auto it = dbs.begin();
while (it != dbs.end()) {
std::cout << (*it).first << ". ";
delete (*it).second;
dbs.erase(it++); //post increment!
}
std::cout << std::endl;
}
|
