56 lines
1.1 KiB
Go
56 lines
1.1 KiB
Go
|
package rpcbroadcaster
|
||
|
|
|
||
|
|
import (
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/nspcc-dev/neo-go/pkg/rpc/request"
|
||
|
|
"go.uber.org/zap"
|
||
|
|
)
|
||
|
|
|
||
|
|
// RPCBroadcaster represent generic RPC broadcaster.
|
||
|
|
type RPCBroadcaster struct {
|
||
|
|
Clients map[string]*RPCClient
|
||
|
|
Log *zap.Logger
|
||
|
|
Responses chan request.RawParams
|
||
|
|
|
||
|
|
close chan struct{}
|
||
|
|
sendTimeout time.Duration
|
||
|
|
}
|
||
|
|
|
||
|
|
// NewRPCBroadcaster returns new RPC broadcaster instance.
|
||
|
|
func NewRPCBroadcaster(log *zap.Logger, sendTimeout time.Duration) *RPCBroadcaster {
|
||
|
|
return &RPCBroadcaster{
|
||
|
|
Clients: make(map[string]*RPCClient),
|
||
|
|
Log: log,
|
||
|
|
close: make(chan struct{}),
|
||
|
|
Responses: make(chan request.RawParams),
|
||
|
|
sendTimeout: sendTimeout,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// Run implements oracle.Broadcaster.
|
||
|
|
func (r *RPCBroadcaster) Run() {
|
||
|
|
for _, c := range r.Clients {
|
||
|
|
go c.run()
|
||
|
|
}
|
||
|
|
for {
|
||
|
|
select {
|
||
|
|
case <-r.close:
|
||
|
|
return
|
||
|
|
case ps := <-r.Responses:
|
||
|
|
for _, c := range r.Clients {
|
||
|
|
select {
|
||
|
|
case c.responses <- ps:
|
||
|
|
default:
|
||
|
|
c.log.Error("can't send response, channel is full")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// Shutdown implements oracle.Broadcaster.
|
||
|
|
func (r *RPCBroadcaster) Shutdown() {
|
||
|
|
close(r.close)
|
||
|
|
}
|