prysm-pulse/beacon-chain/p2p/sender.go
Preston Van Loon 8d234014a4
Fix broadcast ssz (#3423)
* add two types of encoding/decoding ssz

* fix tests

* lint

* lint
2019-09-08 19:34:52 -07:00

39 lines
956 B
Go

package p2p
import (
"context"
"reflect"
"time"
"github.com/gogo/protobuf/proto"
"github.com/libp2p/go-libp2p-core/network"
"github.com/libp2p/go-libp2p-core/peer"
"github.com/libp2p/go-libp2p-core/protocol"
)
// Send a message to a specific peer. The returned stream may be used for reading, but has been
// closed for writing.
func (s *Service) Send(ctx context.Context, message proto.Message, pid peer.ID) (network.Stream, error) {
topic := RPCTypeMapping[reflect.TypeOf(message)] + s.Encoding().ProtocolSuffix()
// TTFB_TIME (5s) + RESP_TIMEOUT (10s).
ctx, cancel := context.WithTimeout(ctx, 15*time.Second)
defer cancel()
stream, err := s.host.NewStream(ctx, pid, protocol.ID(topic))
if err != nil {
return nil, err
}
if _, err := s.Encoding().EncodeWithLength(stream, message); err != nil {
return nil, err
}
// Close stream for writing.
if err := stream.Close(); err != nil {
return nil, err
}
return stream, nil
}