forked from cerc-io/plugeth
cbf2579691
This PR adds metrics for p2p dialing, which gives us visibility into the quality of the dial candidates returned by our discovery methods.
74 lines
2.4 KiB
Go
74 lines
2.4 KiB
Go
// Copyright 2023 The go-ethereum Authors
|
|
// This file is part of the go-ethereum library.
|
|
//
|
|
// The go-ethereum library is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU Lesser General Public License as published by
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version.
|
|
//
|
|
// The go-ethereum library is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU Lesser General Public License for more details.
|
|
//
|
|
// You should have received a copy of the GNU Lesser General Public License
|
|
// along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
package discover
|
|
|
|
import (
|
|
"fmt"
|
|
"net"
|
|
|
|
"github.com/ethereum/go-ethereum/metrics"
|
|
)
|
|
|
|
const (
|
|
moduleName = "discover"
|
|
// ingressMeterName is the prefix of the per-packet inbound metrics.
|
|
ingressMeterName = moduleName + "/ingress"
|
|
|
|
// egressMeterName is the prefix of the per-packet outbound metrics.
|
|
egressMeterName = moduleName + "/egress"
|
|
)
|
|
|
|
var (
|
|
bucketsCounter []metrics.Counter
|
|
ingressTrafficMeter = metrics.NewRegisteredMeter(ingressMeterName, nil)
|
|
egressTrafficMeter = metrics.NewRegisteredMeter(egressMeterName, nil)
|
|
)
|
|
|
|
func init() {
|
|
for i := 0; i < nBuckets; i++ {
|
|
bucketsCounter = append(bucketsCounter, metrics.NewRegisteredCounter(fmt.Sprintf("%s/bucket/%d/count", moduleName, i), nil))
|
|
}
|
|
}
|
|
|
|
// meteredConn is a wrapper around a net.UDPConn that meters both the
|
|
// inbound and outbound network traffic.
|
|
type meteredUdpConn struct {
|
|
UDPConn
|
|
}
|
|
|
|
func newMeteredConn(conn UDPConn) UDPConn {
|
|
// Short circuit if metrics are disabled
|
|
if !metrics.Enabled {
|
|
return conn
|
|
}
|
|
return &meteredUdpConn{UDPConn: conn}
|
|
}
|
|
|
|
// Read delegates a network read to the underlying connection, bumping the udp ingress traffic meter along the way.
|
|
func (c *meteredUdpConn) ReadFromUDP(b []byte) (n int, addr *net.UDPAddr, err error) {
|
|
n, addr, err = c.UDPConn.ReadFromUDP(b)
|
|
ingressTrafficMeter.Mark(int64(n))
|
|
return n, addr, err
|
|
}
|
|
|
|
// Write delegates a network write to the underlying connection, bumping the udp egress traffic meter along the way.
|
|
func (c *meteredUdpConn) WriteToUDP(b []byte, addr *net.UDPAddr) (n int, err error) {
|
|
n, err = c.UDPConn.WriteToUDP(b, addr)
|
|
egressTrafficMeter.Mark(int64(n))
|
|
return n, err
|
|
}
|