43 lines
1.4 KiB
Go
43 lines
1.4 KiB
Go
package publicrpc
|
|
|
|
import (
|
|
gossipv1 "github.com/certusone/wormhole/bridge/pkg/proto/gossip/v1"
|
|
publicrpcv1 "github.com/certusone/wormhole/bridge/pkg/proto/publicrpc/v1"
|
|
"go.uber.org/zap"
|
|
)
|
|
|
|
// PublicrpcServer implements the publicrpc gRPC service.
|
|
type PublicrpcServer struct {
|
|
publicrpcv1.UnimplementedPublicrpcServer
|
|
rawHeartbeatListeners *RawHeartbeatConns
|
|
logger *zap.Logger
|
|
}
|
|
|
|
func NewPublicrpcServer(logger *zap.Logger, rawHeartbeatListeners *RawHeartbeatConns) *PublicrpcServer {
|
|
return &PublicrpcServer{
|
|
rawHeartbeatListeners: rawHeartbeatListeners,
|
|
logger: logger.Named("publicrpcserver"),
|
|
}
|
|
}
|
|
|
|
func (s *PublicrpcServer) GetRawHeartbeats(req *publicrpcv1.GetRawHeartbeatsRequest, stream publicrpcv1.Publicrpc_GetRawHeartbeatsServer) error {
|
|
s.logger.Info("gRPC heartbeat stream opened by client")
|
|
|
|
// create a channel and register it for heartbeats
|
|
receiveChan := make(chan *gossipv1.Heartbeat, 50)
|
|
// clientId is the reference to the subscription that we will use for unsubscribing when the client disconnects.
|
|
clientId := s.rawHeartbeatListeners.subscribeHeartbeats(receiveChan)
|
|
|
|
for {
|
|
select {
|
|
// Exit on stream context done
|
|
case <-stream.Context().Done():
|
|
s.logger.Info("raw heartbeat stream closed by client", zap.Int("clientId", clientId))
|
|
s.rawHeartbeatListeners.unsubscribeHeartbeats(clientId)
|
|
return stream.Context().Err()
|
|
case msg := <-receiveChan:
|
|
stream.Send(msg)
|
|
}
|
|
}
|
|
}
|