You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
77 lines
2.5 KiB
77 lines
2.5 KiB
package utils
|
|
|
|
import (
|
|
"github.com/ethereum/go-ethereum/log"
|
|
net "github.com/libp2p/go-libp2p-net"
|
|
ma "github.com/multiformats/go-multiaddr"
|
|
)
|
|
|
|
// ConnLogger is a LibP2P connection logger that logs to an Ethereum logger.
|
|
// It logs all listener/connection/stream open/close activities at debug level.
|
|
// To use one, add it on a LibP2P host swarm as a notifier, ex:
|
|
//
|
|
// connLogger := utils.NewConnLogger(
|
|
// host.Network().Notify(connLogger)
|
|
type ConnLogger struct {
|
|
l log.Logger
|
|
}
|
|
|
|
func netLogger(n net.Network, l log.Logger) log.Logger {
|
|
return l.New(
|
|
"netLocalPeer", n.LocalPeer(),
|
|
"netListenAddresses", n.ListenAddresses())
|
|
}
|
|
|
|
func connLogger(c net.Conn, l log.Logger) log.Logger {
|
|
return l.New(
|
|
"connLocalPeer", c.LocalPeer(),
|
|
"connLocalAddr", c.LocalMultiaddr(),
|
|
"connRemotePeer", c.RemotePeer(),
|
|
"connRemoteAddr", c.RemoteMultiaddr())
|
|
}
|
|
|
|
func streamLogger(s net.Stream, l log.Logger) log.Logger {
|
|
return connLogger(s.Conn(), l).New("streamProtocolID", s.Protocol())
|
|
}
|
|
|
|
// Listen logs a listener starting listening on an address.
|
|
func (cl ConnLogger) Listen(n net.Network, ma ma.Multiaddr) {
|
|
WithCaller(netLogger(n, cl.l)).
|
|
Debug("listener starting", "listenAddress", ma)
|
|
}
|
|
|
|
// ListenClose logs a listener stopping listening on an address.
|
|
func (cl ConnLogger) ListenClose(n net.Network, ma ma.Multiaddr) {
|
|
WithCaller(netLogger(n, cl.l)).
|
|
Debug("listener closing", "listenAddress", ma)
|
|
}
|
|
|
|
// Connected logs a connection getting opened.
|
|
func (cl ConnLogger) Connected(n net.Network, c net.Conn) {
|
|
WithCaller(connLogger(c, netLogger(n, cl.l))).Debug("connected")
|
|
}
|
|
|
|
// Disconnected logs a connection getting closed.
|
|
func (cl ConnLogger) Disconnected(n net.Network, c net.Conn) {
|
|
WithCaller(connLogger(c, netLogger(n, cl.l))).Debug("disconnected")
|
|
}
|
|
|
|
// OpenedStream logs a new stream getting opened.
|
|
func (cl ConnLogger) OpenedStream(n net.Network, s net.Stream) {
|
|
WithCaller(streamLogger(s, netLogger(n, cl.l))).Debug("stream opened")
|
|
}
|
|
|
|
// ClosedStream logs a stream getting closed.
|
|
func (cl ConnLogger) ClosedStream(n net.Network, s net.Stream) {
|
|
WithCaller(streamLogger(s, netLogger(n, cl.l))).Debug("stream closed")
|
|
}
|
|
|
|
// NewConnLogger returns a new connection logger that uses the given
|
|
// Ethereum logger. See ConnLogger for usage.
|
|
func NewConnLogger(l log.Logger) *ConnLogger {
|
|
return &ConnLogger{l: l}
|
|
}
|
|
|
|
// RootConnLogger is a LibP2P connection logger that logs to Ethereum root
|
|
// logger. See ConnLogger for usage.
|
|
var RootConnLogger = NewConnLogger(log.Root())
|
|
|