1 files changed, 52 insertions, 0 deletions
diff --git a/swarm/storage/dbapi.go b/swarm/storage/dbapi.go
new file mode 100644
index 000000000..24234b031
--- /dev/null
+++ b/swarm/storage/dbapi.go
@@ -0,0 +1,52 @@
+// Copyright 2018 The go-ethereum Authors
+// This file is part of the go-ethereum library.
+//
+// The go-ethereum library is free software: you can redistribute it and/or modify
+// it under the terms of the GNU Lesser General Public License as published by
+// the Free Software Foundation, either version 3 of the License, or
+// (at your option) any later version.
+//
+// The go-ethereum library is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+// GNU Lesser General Public License for more details.
+//
+// You should have received a copy of the GNU Lesser General Public License
+// along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
+
+package storage
+
+// wrapper of db-s to provide mockable custom local chunk store access to syncer
+type DBAPI struct {
+	db  *LDBStore
+	loc *LocalStore
+}
+
+func NewDBAPI(loc *LocalStore) *DBAPI {
+	return &DBAPI{loc.DbStore, loc}
+}
+
+// to obtain the chunks from address or request db entry only
+func (d *DBAPI) Get(addr Address) (*Chunk, error) {
+	return d.loc.Get(addr)
+}
+
+// current storage counter of chunk db
+func (d *DBAPI) CurrentBucketStorageIndex(po uint8) uint64 {
+	return d.db.CurrentBucketStorageIndex(po)
+}
+
+// iteration storage counter and proximity order
+func (d *DBAPI) Iterator(from uint64, to uint64, po uint8, f func(Address, uint64) bool) error {
+	return d.db.SyncIterator(from, to, po, f)
+}
+
+// to obtain the chunks from address or request db entry only
+func (d *DBAPI) GetOrCreateRequest(addr Address) (*Chunk, bool) {
+	return d.loc.GetOrCreateRequest(addr)
+}
+
+// to obtain the chunks from key or request db entry only
+func (d *DBAPI) Put(chunk *Chunk) {
+	d.loc.Put(chunk)
+}