fyrchik/frostfs-node
1
0
Fork 0
forked from TrueCloudLab/frostfs-node
Code Issues Pull requests Projects Releases Packages Wiki Activity

[#324] morph/subscriber: Subscribe on new chain blocks

Browse source 
Call SubscribeForNewBlocks in Subscriber's constructor. Provide
BlockNotifications interface method that returns block channel. Write new
blocks to the channel on notification events with BlockEventID type.

Signed-off-by: Leonard Lyubich <leonard@nspcc.ru>
This commit is contained in:
Leonard Lyubich 2021-01-20 20:05:52 +03:00 committed by Alex Vanin
parent 65eca6077e
commit c65d1d9db9
1 changed files with 22 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

22
pkg/morph/subscriber/subscriber.go
View file

@ -6,6 +6,7 @@ import (
"sync"
"time"
"github.com/nspcc-dev/neo-go/pkg/core/block"
"github.com/nspcc-dev/neo-go/pkg/core/state"
"github.com/nspcc-dev/neo-go/pkg/rpc/client"
"github.com/nspcc-dev/neo-go/pkg/rpc/response"
@ -19,6 +20,7 @@ type (
SubscribeForNotification(...util.Uint160) (<-chan *state.NotificationEvent, error)
UnsubscribeForNotification()
Close()
BlockNotifications() <-chan *block.Block
}
subscriber struct {
@ -28,6 +30,8 @@ type (
notify chan *state.NotificationEvent
notifyIDs map[util.Uint160]string
blockChan chan *block.Block
}
// Params is a group of Subscriber constructor parameters.
@ -99,6 +103,10 @@ func (s *subscriber) Close() {
s.client.Close()
}
func (s *subscriber) BlockNotifications() <-chan *block.Block {
return s.blockChan
}
func (s *subscriber) routeNotifications(ctx context.Context) {
for {
select {
@ -108,6 +116,7 @@ func (s *subscriber) routeNotifications(ctx context.Context) {
if !ok {
s.log.Warn("remote channel has been closed")
close(s.notify)
close(s.blockChan)
return
}
@ -121,6 +130,14 @@ func (s *subscriber) routeNotifications(ctx context.Context) {
}
s.notify <- notification
case response.BlockEventID:
b, ok := notification.Value.(*block.Block)
if !ok {
s.log.Error("can't cast block event value to block")
continue
}
s.blockChan <- b
default:
s.log.Debug("unsupported notification from the chain",
zap.Uint8("type", uint8(notification.Type)),
@ -146,12 +163,17 @@ func New(ctx context.Context, p *Params) (Subscriber, error) {
return nil, err
}
if _, err := wsClient.SubscribeForNewBlocks(nil); err != nil {
return nil, err
}
sub := &subscriber{
RWMutex: new(sync.RWMutex),
log: p.Log,
client: wsClient,
notify: make(chan *state.NotificationEvent),
notifyIDs: make(map[util.Uint160]string),
blockChan: make(chan *block.Block),
}
// Worker listens all events from neo-go websocket and puts them