mirror of
https://github.com/status-im/status-go.git
synced 2025-01-10 22:56:40 +00:00
eeca435064
Update vendor Integrate rendezvous into status node Add a test with failover using rendezvous Use multiple servers in client Use discovery V5 by default and test that node can be started with rendezvous discovet Fix linter Update rendezvous client to one with instrumented stream Address feedback Fix test with updated topic limits Apply several suggestions Change log to debug for request errors because we continue execution Remove web3js after rebase Update rendezvous package
84 lines
2.0 KiB
Go
84 lines
2.0 KiB
Go
package metrics
|
|
|
|
import (
|
|
flow "github.com/libp2p/go-flow-metrics"
|
|
peer "github.com/libp2p/go-libp2p-peer"
|
|
protocol "github.com/libp2p/go-libp2p-protocol"
|
|
)
|
|
|
|
type Stats struct {
|
|
TotalIn int64
|
|
TotalOut int64
|
|
RateIn float64
|
|
RateOut float64
|
|
}
|
|
|
|
type BandwidthCounter struct {
|
|
totalIn flow.Meter
|
|
totalOut flow.Meter
|
|
|
|
protocolIn flow.MeterRegistry
|
|
protocolOut flow.MeterRegistry
|
|
|
|
peerIn flow.MeterRegistry
|
|
peerOut flow.MeterRegistry
|
|
}
|
|
|
|
func NewBandwidthCounter() *BandwidthCounter {
|
|
return new(BandwidthCounter)
|
|
}
|
|
|
|
func (bwc *BandwidthCounter) LogSentMessage(size int64) {
|
|
bwc.totalOut.Mark(uint64(size))
|
|
}
|
|
|
|
func (bwc *BandwidthCounter) LogRecvMessage(size int64) {
|
|
bwc.totalIn.Mark(uint64(size))
|
|
}
|
|
|
|
func (bwc *BandwidthCounter) LogSentMessageStream(size int64, proto protocol.ID, p peer.ID) {
|
|
bwc.protocolOut.Get(string(proto)).Mark(uint64(size))
|
|
bwc.peerOut.Get(string(p)).Mark(uint64(size))
|
|
}
|
|
|
|
func (bwc *BandwidthCounter) LogRecvMessageStream(size int64, proto protocol.ID, p peer.ID) {
|
|
bwc.protocolIn.Get(string(proto)).Mark(uint64(size))
|
|
bwc.peerIn.Get(string(p)).Mark(uint64(size))
|
|
}
|
|
|
|
func (bwc *BandwidthCounter) GetBandwidthForPeer(p peer.ID) (out Stats) {
|
|
inSnap := bwc.peerIn.Get(string(p)).Snapshot()
|
|
outSnap := bwc.peerOut.Get(string(p)).Snapshot()
|
|
|
|
return Stats{
|
|
TotalIn: int64(inSnap.Total),
|
|
TotalOut: int64(outSnap.Total),
|
|
RateIn: inSnap.Rate,
|
|
RateOut: outSnap.Rate,
|
|
}
|
|
}
|
|
|
|
func (bwc *BandwidthCounter) GetBandwidthForProtocol(proto protocol.ID) (out Stats) {
|
|
inSnap := bwc.protocolIn.Get(string(proto)).Snapshot()
|
|
outSnap := bwc.protocolOut.Get(string(proto)).Snapshot()
|
|
|
|
return Stats{
|
|
TotalIn: int64(inSnap.Total),
|
|
TotalOut: int64(outSnap.Total),
|
|
RateIn: inSnap.Rate,
|
|
RateOut: outSnap.Rate,
|
|
}
|
|
}
|
|
|
|
func (bwc *BandwidthCounter) GetBandwidthTotals() (out Stats) {
|
|
inSnap := bwc.totalIn.Snapshot()
|
|
outSnap := bwc.totalOut.Snapshot()
|
|
|
|
return Stats{
|
|
TotalIn: int64(inSnap.Total),
|
|
TotalOut: int64(outSnap.Total),
|
|
RateIn: inSnap.Rate,
|
|
RateOut: outSnap.Rate,
|
|
}
|
|
}
|