forks/go-ethereum
1
0
Fork 1
mirror of https://github.com/ethereum/go-ethereum.git synced 2026-08-17 17:33:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

swarm/network/stream: Subscribe to stream to retrieve chunks

Browse source
This commit is contained in:
Balint Gabor 2018-03-06 11:55:29 +01:00
parent b777b0dff8
commit 746d1d6a54
4 changed files with 13 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
swarm/network/stream/common_test.go
View file

@ -78,7 +78,7 @@ func NewStreamerService(ctx *adapters.ServiceContext) (node.Service, error) {
db := storage.NewDBAPI(store)
delivery := NewDelivery(kad, db)
deliveries[id] = delivery
r := NewRegistry(addr, delivery, db, state.NewMemStore(), defaultSkipCheck, false)
r := NewRegistry(addr, delivery, db, state.NewMemStore(), defaultSkipCheck, false, false)
RegisterSwarmSyncerServer(r, db)
RegisterSwarmSyncerClient(r, db)
go func() {
@ -109,7 +109,7 @@ func newStreamerTester(t *testing.T) (*p2ptest.ProtocolTester, *Registry, *stora
db := storage.NewDBAPI(localStore)
delivery := NewDelivery(to, db)
streamer := NewRegistry(addr, delivery, db, state.NewMemStore(), defaultSkipCheck, false)
streamer := NewRegistry(addr, delivery, db, state.NewMemStore(), defaultSkipCheck, false, false)
teardown := func() {
streamer.Close()
removeDataDir()

2
swarm/network/stream/intervals_test.go
View file

@ -50,7 +50,7 @@ func newIntervalsStreamerService(ctx *adapters.ServiceContext) (node.Service, er
db := storage.NewDBAPI(store)
delivery := NewDelivery(kad, db)
deliveries[id] = delivery
r := NewRegistry(addr, delivery, db, state.NewMemStore(), defaultSkipCheck, false)
r := NewRegistry(addr, delivery, db, state.NewMemStore(), defaultSkipCheck, false, false)
r.RegisterClientFunc(externalStreamName, func(p *Peer, t []byte, live bool) (Client, error) {
return newTestExternalClient(t, db), nil

10
swarm/network/stream/stream.go
View file

@ -58,10 +58,11 @@ type Registry struct {
delivery *Delivery
intervalsStore state.Store
doSync bool
doRetrieve bool
}
// NewRegistry is Streamer constructor
func NewRegistry(addr *network.BzzAddr, delivery *Delivery, db *storage.DBAPI, intervalsStore state.Store, skipCheck, doSync bool) *Registry {
func NewRegistry(addr *network.BzzAddr, delivery *Delivery, db *storage.DBAPI, intervalsStore state.Store, skipCheck, doSync, doRetrieve bool) *Registry {
streamer := &Registry{
addr: addr,
skipCheck: skipCheck,
@ -71,6 +72,7 @@ func NewRegistry(addr *network.BzzAddr, delivery *Delivery, db *storage.DBAPI, i
delivery: delivery,
intervalsStore: intervalsStore,
doSync: doSync,
doRetrieve: doRetrieve,
}
streamer.api = NewAPI(streamer)
delivery.getPeer = streamer.getPeer
@ -312,6 +314,12 @@ func (r *Registry) Run(p *network.BzzPeer) error {
return true
})
}
if r.doRetrieve {
err := r.Subscribe(p.ID(), NewStream(swarmChunkServerStreamName, nil, false), nil, Top)
if err != nil {
return err
}
}
return sp.Run(sp.HandleMsg)
}

2
swarm/swarm.go
View file

@ -153,7 +153,7 @@ func NewSwarm(ctx *node.ServiceContext, backend chequebook.Backend, config *api.
if err != nil {
return
}
self.streamer = stream.NewRegistry(addr, delivery, db, stateStore, false, true)
self.streamer = stream.NewRegistry(addr, delivery, db, stateStore, false, true, true)
self.bzz = network.NewBzz(bzzconfig, to, stateStore, stream.Spec, self.streamer.Run)