swarm: plan bee for content storage and distribution on web3
This change imports the Swarm protocol codebase. Compared to the 'swarm' branch, a few mostly cosmetic changes had to be made: * The various redundant log message prefixes are gone. * All files now have LGPLv3 license headers. * Minor code changes were needed to please go vet and make the tests pass on Windows. * Further changes were required to adapt to the go-ethereum develop branch and its new Go APIs. Some code has not (yet) been brought over: * swarm/cmd/bzzhash: will reappear as cmd/bzzhash later * swarm/cmd/bzzup.sh: will be reimplemented in cmd/bzzup * swarm/cmd/makegenesis: will reappear somehow * swarm/examples/album: will move to a separate repository * swarm/examples/filemanager: ditto * swarm/examples/files: will not be merged * swarm/test/*: will not be merged * swarm/services/swear: will reappear as contracts/swear when needed
This commit is contained in:
ΞTHΞЯSPHΞЯΞ
Felix Lange
221
swarm/network/syncdb_test.go
Normal file
221
swarm/network/syncdb_test.go
Normal file
@ -0,0 +1,221 @@
|
||||
// Copyright 2016 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 network
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"io/ioutil"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"github.com/ethereum/go-ethereum/crypto"
|
||||
"github.com/ethereum/go-ethereum/logger"
|
||||
"github.com/ethereum/go-ethereum/logger/glog"
|
||||
"github.com/ethereum/go-ethereum/swarm/storage"
|
||||
)
|
||||
|
||||
func init() {
|
||||
glog.SetV(0)
|
||||
glog.SetToStderr(true)
|
||||
}
|
||||
|
||||
type testSyncDb struct {
|
||||
*syncDb
|
||||
c int
|
||||
t *testing.T
|
||||
fromDb chan bool
|
||||
delivered [][]byte
|
||||
sent []int
|
||||
dbdir string
|
||||
at int
|
||||
}
|
||||
|
||||
func newTestSyncDb(priority, bufferSize, batchSize int, dbdir string, t *testing.T) *testSyncDb {
|
||||
if len(dbdir) == 0 {
|
||||
tmp, err := ioutil.TempDir(os.TempDir(), "syncdb-test")
|
||||
if err != nil {
|
||||
t.Fatalf("unable to create temporary direcory %v: %v", tmp, err)
|
||||
}
|
||||
dbdir = tmp
|
||||
}
|
||||
db, err := storage.NewLDBDatabase(filepath.Join(dbdir, "requestdb"))
|
||||
if err != nil {
|
||||
t.Fatalf("unable to create db: %v", err)
|
||||
}
|
||||
self := &testSyncDb{
|
||||
fromDb: make(chan bool),
|
||||
dbdir: dbdir,
|
||||
t: t,
|
||||
}
|
||||
h := crypto.Sha3Hash([]byte{0})
|
||||
key := storage.Key(h[:])
|
||||
self.syncDb = newSyncDb(db, key, uint(priority), uint(bufferSize), uint(batchSize), self.deliver)
|
||||
// kick off db iterator right away, if no items on db this will allow
|
||||
// reading from the buffer
|
||||
return self
|
||||
|
||||
}
|
||||
|
||||
func (self *testSyncDb) close() {
|
||||
self.db.Close()
|
||||
os.RemoveAll(self.dbdir)
|
||||
}
|
||||
|
||||
func (self *testSyncDb) push(n int) {
|
||||
for i := 0; i < n; i++ {
|
||||
self.buffer <- storage.Key(crypto.Sha3([]byte{byte(self.c)}))
|
||||
self.sent = append(self.sent, self.c)
|
||||
self.c++
|
||||
}
|
||||
glog.V(logger.Debug).Infof("pushed %v requests", n)
|
||||
}
|
||||
|
||||
func (self *testSyncDb) draindb() {
|
||||
it := self.db.NewIterator()
|
||||
defer it.Release()
|
||||
for {
|
||||
it.Seek(self.start)
|
||||
if !it.Valid() {
|
||||
return
|
||||
}
|
||||
k := it.Key()
|
||||
if len(k) == 0 || k[0] == 1 {
|
||||
return
|
||||
}
|
||||
it.Release()
|
||||
it = self.db.NewIterator()
|
||||
}
|
||||
}
|
||||
|
||||
func (self *testSyncDb) deliver(req interface{}, quit chan bool) bool {
|
||||
_, db := req.(*syncDbEntry)
|
||||
key, _, _, _, err := parseRequest(req)
|
||||
if err != nil {
|
||||
self.t.Fatalf("unexpected error of key %v: %v", key, err)
|
||||
}
|
||||
self.delivered = append(self.delivered, key)
|
||||
select {
|
||||
case self.fromDb <- db:
|
||||
return true
|
||||
case <-quit:
|
||||
return false
|
||||
}
|
||||
}
|
||||
|
||||
func (self *testSyncDb) expect(n int, db bool) {
|
||||
var ok bool
|
||||
// for n items
|
||||
for i := 0; i < n; i++ {
|
||||
ok = <-self.fromDb
|
||||
if self.at+1 > len(self.delivered) {
|
||||
self.t.Fatalf("expected %v, got %v", self.at+1, len(self.delivered))
|
||||
}
|
||||
if len(self.sent) > self.at && !bytes.Equal(crypto.Sha3([]byte{byte(self.sent[self.at])}), self.delivered[self.at]) {
|
||||
self.t.Fatalf("expected delivery %v/%v/%v to be hash of %v, from db: %v = %v", i, n, self.at, self.sent[self.at], ok, db)
|
||||
glog.V(logger.Debug).Infof("%v/%v/%v to be hash of %v, from db: %v = %v", i, n, self.at, self.sent[self.at], ok, db)
|
||||
}
|
||||
if !ok && db {
|
||||
self.t.Fatalf("expected delivery %v/%v/%v from db", i, n, self.at)
|
||||
}
|
||||
if ok && !db {
|
||||
self.t.Fatalf("expected delivery %v/%v/%v from cache", i, n, self.at)
|
||||
}
|
||||
self.at++
|
||||
}
|
||||
}
|
||||
|
||||
func TestSyncDb(t *testing.T) {
|
||||
priority := High
|
||||
bufferSize := 5
|
||||
batchSize := 2 * bufferSize
|
||||
s := newTestSyncDb(priority, bufferSize, batchSize, "", t)
|
||||
defer s.close()
|
||||
defer s.stop()
|
||||
s.dbRead(false, 0, s.deliver)
|
||||
s.draindb()
|
||||
|
||||
s.push(4)
|
||||
s.expect(1, false)
|
||||
// 3 in buffer
|
||||
time.Sleep(100 * time.Millisecond)
|
||||
s.push(3)
|
||||
// push over limit
|
||||
s.expect(1, false)
|
||||
// one popped from the buffer, then contention detected
|
||||
s.expect(4, true)
|
||||
s.push(4)
|
||||
s.expect(5, true)
|
||||
// depleted db, switch back to buffer
|
||||
s.draindb()
|
||||
s.push(5)
|
||||
s.expect(4, false)
|
||||
s.push(3)
|
||||
s.expect(4, false)
|
||||
// buffer depleted
|
||||
time.Sleep(100 * time.Millisecond)
|
||||
s.push(6)
|
||||
s.expect(1, false)
|
||||
// push into buffer full, switch to db
|
||||
s.expect(5, true)
|
||||
s.draindb()
|
||||
s.push(1)
|
||||
s.expect(1, false)
|
||||
}
|
||||
|
||||
func TestSaveSyncDb(t *testing.T) {
|
||||
amount := 30
|
||||
priority := High
|
||||
bufferSize := amount
|
||||
batchSize := 10
|
||||
s := newTestSyncDb(priority, bufferSize, batchSize, "", t)
|
||||
go s.dbRead(false, 0, s.deliver)
|
||||
s.push(amount)
|
||||
s.stop()
|
||||
s.db.Close()
|
||||
|
||||
s = newTestSyncDb(priority, bufferSize, batchSize, s.dbdir, t)
|
||||
go s.dbRead(false, 0, s.deliver)
|
||||
s.expect(amount, true)
|
||||
for i, key := range s.delivered {
|
||||
expKey := crypto.Sha3([]byte{byte(i)})
|
||||
if !bytes.Equal(key, expKey) {
|
||||
t.Fatalf("delivery %v expected to be key %x, got %x", i, expKey, key)
|
||||
}
|
||||
}
|
||||
s.push(amount)
|
||||
s.expect(amount, false)
|
||||
for i := amount; i < 2*amount; i++ {
|
||||
key := s.delivered[i]
|
||||
expKey := crypto.Sha3([]byte{byte(i - amount)})
|
||||
if !bytes.Equal(key, expKey) {
|
||||
t.Fatalf("delivery %v expected to be key %x, got %x", i, expKey, key)
|
||||
}
|
||||
}
|
||||
s.stop()
|
||||
s.db.Close()
|
||||
|
||||
s = newTestSyncDb(priority, bufferSize, batchSize, s.dbdir, t)
|
||||
defer s.close()
|
||||
defer s.stop()
|
||||
|
||||
go s.dbRead(false, 0, s.deliver)
|
||||
s.push(1)
|
||||
s.expect(1, false)
|
||||
|
||||
}
|
||||
Reference in New Issue
Block a user