mirror of
https://github.com/kaspanet/kaspad.git
synced 2025-06-07 06:36:46 +00:00
Add P2P/RPC name to the gRPC logs (#1448)
* Add P2P/RPC name to the gRPC logs * Add p2p/rpc name in more logs
This commit is contained in:
parent
819ec9f2a7
commit
ed85f09742
@ -15,14 +15,16 @@ type gRPCServer struct {
|
|||||||
onConnectedHandler server.OnConnectedHandler
|
onConnectedHandler server.OnConnectedHandler
|
||||||
listeningAddresses []string
|
listeningAddresses []string
|
||||||
server *grpc.Server
|
server *grpc.Server
|
||||||
|
name string
|
||||||
}
|
}
|
||||||
|
|
||||||
// newGRPCServer creates a gRPC server
|
// newGRPCServer creates a gRPC server
|
||||||
func newGRPCServer(listeningAddresses []string, maxMessageSize int) *gRPCServer {
|
func newGRPCServer(listeningAddresses []string, maxMessageSize int, name string) *gRPCServer {
|
||||||
log.Debugf("Created new GRPC server with maxMessageSize %d", maxMessageSize)
|
log.Debugf("Created new %s GRPC server with maxMessageSize %d", name, maxMessageSize)
|
||||||
return &gRPCServer{
|
return &gRPCServer{
|
||||||
server: grpc.NewServer(grpc.MaxRecvMsgSize(maxMessageSize), grpc.MaxSendMsgSize(maxMessageSize)),
|
server: grpc.NewServer(grpc.MaxRecvMsgSize(maxMessageSize), grpc.MaxSendMsgSize(maxMessageSize)),
|
||||||
listeningAddresses: listeningAddresses,
|
listeningAddresses: listeningAddresses,
|
||||||
|
name: name,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -44,17 +46,17 @@ func (s *gRPCServer) Start() error {
|
|||||||
func (s *gRPCServer) listenOn(listenAddr string) error {
|
func (s *gRPCServer) listenOn(listenAddr string) error {
|
||||||
listener, err := net.Listen("tcp", listenAddr)
|
listener, err := net.Listen("tcp", listenAddr)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return errors.Wrapf(err, "error listening on %s", listenAddr)
|
return errors.Wrapf(err, "%s error listening on %s", s.name, listenAddr)
|
||||||
}
|
}
|
||||||
|
|
||||||
spawn("gRPCServer.listenOn-Serve", func() {
|
spawn(fmt.Sprintf("%s.gRPCServer.listenOn-Serve", s.name), func() {
|
||||||
err := s.server.Serve(listener)
|
err := s.server.Serve(listener)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
panics.Exit(log, fmt.Sprintf("error serving on %s: %+v", listenAddr, err))
|
panics.Exit(log, fmt.Sprintf("error serving %s on %s: %+v", s.name, listenAddr, err))
|
||||||
}
|
}
|
||||||
})
|
})
|
||||||
|
|
||||||
log.Infof("Server listening on %s", listenAddr)
|
log.Infof("%s Server listening on %s", s.name, listenAddr)
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -86,7 +88,7 @@ func (s *gRPCServer) handleInboundConnection(ctx context.Context, stream grpcStr
|
|||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
log.Infof("Incoming connection from %s", peerInfo.Addr)
|
log.Infof("%s Incoming connection from %s", s.name, peerInfo.Addr)
|
||||||
|
|
||||||
<-connection.stopChan
|
<-connection.stopChan
|
||||||
|
|
||||||
|
@ -22,7 +22,7 @@ const p2pMaxMessageSize = 10 * 1024 * 1024 // 10MB
|
|||||||
|
|
||||||
// NewP2PServer creates a new P2PServer
|
// NewP2PServer creates a new P2PServer
|
||||||
func NewP2PServer(listeningAddresses []string) (server.P2PServer, error) {
|
func NewP2PServer(listeningAddresses []string) (server.P2PServer, error) {
|
||||||
gRPCServer := newGRPCServer(listeningAddresses, p2pMaxMessageSize)
|
gRPCServer := newGRPCServer(listeningAddresses, p2pMaxMessageSize, "P2P")
|
||||||
p2pServer := &p2pServer{gRPCServer: *gRPCServer}
|
p2pServer := &p2pServer{gRPCServer: *gRPCServer}
|
||||||
protowire.RegisterP2PServer(gRPCServer.server, p2pServer)
|
protowire.RegisterP2PServer(gRPCServer.server, p2pServer)
|
||||||
return p2pServer, nil
|
return p2pServer, nil
|
||||||
@ -37,7 +37,7 @@ func (p *p2pServer) MessageStream(stream protowire.P2P_MessageStreamServer) erro
|
|||||||
// Connect connects to the given address
|
// Connect connects to the given address
|
||||||
// This is part of the P2PServer interface
|
// This is part of the P2PServer interface
|
||||||
func (p *p2pServer) Connect(address string) (server.Connection, error) {
|
func (p *p2pServer) Connect(address string) (server.Connection, error) {
|
||||||
log.Infof("Dialing to %s", address)
|
log.Infof("%s Dialing to %s", p.name, address)
|
||||||
|
|
||||||
const dialTimeout = 30 * time.Second
|
const dialTimeout = 30 * time.Second
|
||||||
ctx, cancel := context.WithTimeout(context.Background(), dialTimeout)
|
ctx, cancel := context.WithTimeout(context.Background(), dialTimeout)
|
||||||
@ -45,19 +45,19 @@ func (p *p2pServer) Connect(address string) (server.Connection, error) {
|
|||||||
|
|
||||||
gRPCClientConnection, err := grpc.DialContext(ctx, address, grpc.WithInsecure(), grpc.WithBlock())
|
gRPCClientConnection, err := grpc.DialContext(ctx, address, grpc.WithInsecure(), grpc.WithBlock())
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, errors.Wrapf(err, "error connecting to %s", address)
|
return nil, errors.Wrapf(err, "%s error connecting to %s", p.name, address)
|
||||||
}
|
}
|
||||||
|
|
||||||
client := protowire.NewP2PClient(gRPCClientConnection)
|
client := protowire.NewP2PClient(gRPCClientConnection)
|
||||||
stream, err := client.MessageStream(context.Background(), grpc.UseCompressor(gzip.Name),
|
stream, err := client.MessageStream(context.Background(), grpc.UseCompressor(gzip.Name),
|
||||||
grpc.MaxCallRecvMsgSize(p2pMaxMessageSize), grpc.MaxCallSendMsgSize(p2pMaxMessageSize))
|
grpc.MaxCallRecvMsgSize(p2pMaxMessageSize), grpc.MaxCallSendMsgSize(p2pMaxMessageSize))
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, errors.Wrapf(err, "error getting client stream for %s", address)
|
return nil, errors.Wrapf(err, "%s error getting client stream for %s", p.name, address)
|
||||||
}
|
}
|
||||||
|
|
||||||
peerInfo, ok := peer.FromContext(stream.Context())
|
peerInfo, ok := peer.FromContext(stream.Context())
|
||||||
if !ok {
|
if !ok {
|
||||||
return nil, errors.Errorf("error getting stream peer info from context for %s", address)
|
return nil, errors.Errorf("%s error getting stream peer info from context for %s", p.name, address)
|
||||||
}
|
}
|
||||||
tcpAddress, ok := peerInfo.Addr.(*net.TCPAddr)
|
tcpAddress, ok := peerInfo.Addr.(*net.TCPAddr)
|
||||||
if !ok {
|
if !ok {
|
||||||
@ -71,7 +71,7 @@ func (p *p2pServer) Connect(address string) (server.Connection, error) {
|
|||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
|
||||||
log.Infof("Connected to %s", address)
|
log.Infof("%s Connected to %s", p.name, address)
|
||||||
|
|
||||||
return connection, nil
|
return connection, nil
|
||||||
}
|
}
|
||||||
|
@ -16,7 +16,7 @@ const RPCMaxMessageSize = 1024 * 1024 * 1024 // 1 GB
|
|||||||
|
|
||||||
// NewRPCServer creates a new RPCServer
|
// NewRPCServer creates a new RPCServer
|
||||||
func NewRPCServer(listeningAddresses []string) (server.Server, error) {
|
func NewRPCServer(listeningAddresses []string) (server.Server, error) {
|
||||||
gRPCServer := newGRPCServer(listeningAddresses, RPCMaxMessageSize)
|
gRPCServer := newGRPCServer(listeningAddresses, RPCMaxMessageSize, "RPC")
|
||||||
rpcServer := &rpcServer{gRPCServer: *gRPCServer}
|
rpcServer := &rpcServer{gRPCServer: *gRPCServer}
|
||||||
protowire.RegisterRPCServer(gRPCServer.server, rpcServer)
|
protowire.RegisterRPCServer(gRPCServer.server, rpcServer)
|
||||||
return rpcServer, nil
|
return rpcServer, nil
|
||||||
|
Loading…
x
Reference in New Issue
Block a user