This repository has been archived by the owner on Oct 28, 2021. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 2.2k
/
OverlayDB.cpp
147 lines (131 loc) · 3.66 KB
/
OverlayDB.cpp
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
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
// Aleth: Ethereum C++ client, tools and libraries.
// Copyright 2014-2019 Aleth Authors.
// Licensed under the GNU General Public License, Version 3.
#include <thread>
#include <libdevcore/db.h>
#include <libdevcore/Common.h>
#include "SHA3.h"
#include "OverlayDB.h"
#include "TrieDB.h"
namespace dev
{
namespace
{
inline db::Slice toSlice(h256 const& _h)
{
return db::Slice(reinterpret_cast<char const*>(_h.data()), _h.size);
}
inline db::Slice toSlice(std::string const& _str)
{
return db::Slice(_str.data(), _str.size());
}
inline db::Slice toSlice(bytes const& _b)
{
return db::Slice(reinterpret_cast<char const*>(&_b[0]), _b.size());
}
} // namespace
OverlayDB::~OverlayDB() = default;
void OverlayDB::commit()
{
if (m_db)
{
auto writeBatch = m_db->createWriteBatch();
// cnote << "Committing nodes to disk DB:";
#if DEV_GUARDED_DB
DEV_READ_GUARDED(x_this)
#endif
{
for (auto const& i: m_main)
{
if (i.second.second)
writeBatch->insert(toSlice(i.first), toSlice(i.second.first));
// cnote << i.first << "#" << m_main[i.first].second;
}
for (auto const& i: m_aux)
if (i.second.second)
{
bytes b = i.first.asBytes();
b.push_back(255); // for aux
writeBatch->insert(toSlice(b), toSlice(i.second.first));
}
}
for (unsigned i = 0; i < 10; ++i)
{
try
{
m_db->commit(std::move(writeBatch));
break;
}
catch (boost::exception const& ex)
{
if (i == 9)
{
cwarn << "Fail writing to state database. Bombing out.";
exit(-1);
}
cwarn << "Error writing to state database: " << boost::diagnostic_information(ex);
cwarn << "Sleeping for" << (i + 1) << "seconds, then retrying.";
std::this_thread::sleep_for(std::chrono::seconds(i + 1));
}
}
#if DEV_GUARDED_DB
DEV_WRITE_GUARDED(x_this)
#endif
{
m_aux.clear();
m_main.clear();
}
}
}
bytes OverlayDB::lookupAux(h256 const& _h) const
{
bytes ret = StateCacheDB::lookupAux(_h);
if (!ret.empty() || !m_db)
return ret;
bytes b = _h.asBytes();
b.push_back(255); // for aux
std::string const v = m_db->lookup(toSlice(b));
if (v.empty())
cwarn << "Aux not found: " << _h;
return asBytes(v);
}
void OverlayDB::rollback()
{
#if DEV_GUARDED_DB
WriteGuard l(x_this);
#endif
m_main.clear();
}
std::string OverlayDB::lookup(h256 const& _h) const
{
std::string ret = StateCacheDB::lookup(_h);
if (!ret.empty() || !m_db)
return ret;
return m_db->lookup(toSlice(_h));
}
bool OverlayDB::exists(h256 const& _h) const
{
if (StateCacheDB::exists(_h))
return true;
return m_db && m_db->exists(toSlice(_h));
}
void OverlayDB::kill(h256 const& _h)
{
if (!StateCacheDB::kill(_h))
{
if (m_db)
{
if (!m_db->exists(toSlice(_h)))
{
// No point node ref decreasing for EmptyTrie since we never bother incrementing it
// in the first place for empty storage tries.
if (_h != EmptyTrie)
cnote << "Decreasing DB node ref count below zero with no DB node. Probably "
"have a corrupt Trie."
<< _h;
// TODO: for 1.1: ref-counted triedb.
}
}
}
}
}