diff --git a/swarm/network/stream/common_test.go b/swarm/network/stream/common_test.go index 7997e977e9..1deb6ffba9 100644 --- a/swarm/network/stream/common_test.go +++ b/swarm/network/stream/common_test.go @@ -63,11 +63,12 @@ func NewStreamerService(ctx *adapters.ServiceContext) (node.Service, error) { id := ctx.Config.ID addr := toAddr(id) kad := network.NewKademlia(addr.Over(), network.NewKadParams()) - store := stores[id] - db := storage.NewDBAPI(store.(*storage.LocalStore)) + store := stores[id].(*storage.LocalStore) + db := storage.NewDBAPI(store) delivery := NewDelivery(kad, db) deliveries[id] = delivery - r := NewRegistry(addr, delivery, store, defaultSkipCheck) + netStore := storage.NewNetStore(store, nil) + r := NewRegistry(addr, delivery, netStore, defaultSkipCheck) RegisterSwarmSyncerServer(r, db) RegisterSwarmSyncerClient(r, db) go func() { diff --git a/swarm/storage/localstore.go b/swarm/storage/localstore.go index 066a27028a..898e7f18ea 100644 --- a/swarm/storage/localstore.go +++ b/swarm/storage/localstore.go @@ -94,6 +94,7 @@ func NewTestLocalStoreForAddr(path string, basekey []byte) (*LocalStore, error) // LocalStore is itself a chunk store // unsafe, in that the data is not integrity checked func (self *LocalStore) Put(chunk *Chunk) { + chunk.Size = int64(binary.LittleEndian.Uint64(chunk.SData[0:8])) self.memStore.Put(chunk) go func() { self.DbStore.Put(chunk) diff --git a/swarm/storage/netstore.go b/swarm/storage/netstore.go index 265baa5337..bbf702fdc8 100644 --- a/swarm/storage/netstore.go +++ b/swarm/storage/netstore.go @@ -36,17 +36,28 @@ func NewNetStore(localStore *LocalStore, retrieve func(chunk *Chunk) error) *Net // Get is the entrypoint for local retrieve requests // waits for response or times out func (self *NetStore) Get(key Key) (chunk *Chunk, err error) { - var created bool - chunk, created = self.localStore.GetOrCreateRequest(key) - if chunk.ReqC == nil { - return chunk, nil - } - - if created { - if err := self.retrieve(chunk); err != nil { + if self.retrieve == nil { + chunk, err = self.localStore.Get(key) + if err == nil { + return chunk, nil + } + if err != ErrFetching { return nil, err } + } else { + var created bool + chunk, created = self.localStore.GetOrCreateRequest(key) + if chunk.ReqC == nil { + return chunk, nil + } + + if created { + if err := self.retrieve(chunk); err != nil { + return nil, err + } + } } + t := time.NewTicker(searchTimeout) defer t.Stop()