libp2p/go-libp2p-resource-manager
1
0
Fork 0
mirror of https://github.com/libp2p/go-libp2p-resource-manager.git synced 2025-03-31 22:50:19 +08:00
Code Issues Projects Releases Wiki Activity

Revert "Add multimetrics reporter"

Browse Source 
This reverts commit 81fb49d3d68153b610b245d6f730f4d6b34b3706.
This commit is contained in:
Marco Munizaga 2022-06-20 18:05:29 -07:00
parent 3a06e88227
commit 1c172ace77
1 changed files with 2 additions and 115 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

117
metrics.go
View File

@ -47,20 +47,10 @@ type metrics struct {
reporter MetricsReporter reporter MetricsReporter
} }
// WithMetrics is a resource manager option to enable metrics collection. Can be // WithMetrics is a resource manager option to enable metrics collection
// called multiple times to add multiple reporters.
func WithMetrics(reporter MetricsReporter) Option { func WithMetrics(reporter MetricsReporter) Option {
return func(r *resourceManager) error { return func(r *resourceManager) error {
if r.metrics == nil { r.metrics = &metrics{reporter: reporter}
r.metrics = &metrics{reporter: reporter}
} else if multimetrics, ok := r.metrics.reporter.(*MultiMetricsReporter); ok {
multimetrics.reporters = append(multimetrics.reporters, reporter)
} else {
// This was a single reporter. Lets convert it to a multimetrics reporter
r.metrics = &metrics{
reporter: &MultiMetricsReporter{reporters: []MetricsReporter{r.metrics.reporter, reporter}},
}
}
return nil return nil
} }
} }
@ -176,106 +166,3 @@ func (m *metrics) BlockMemory(size int) {
m.reporter.BlockMemory(size) m.reporter.BlockMemory(size)
} }
// MultiMetricsReporter is a helper that allows you to report to multiple metrics reporters.
type MultiMetricsReporter struct {
reporters []MetricsReporter
}
// AllowConn is invoked when opening a connection is allowed
func (mmr *MultiMetricsReporter) AllowConn(dir network.Direction, usefd bool) {
for _, r := range mmr.reporters {
r.AllowConn(dir, usefd)
}
}
// BlockConn is invoked when opening a connection is blocked
func (mmr *MultiMetricsReporter) BlockConn(dir network.Direction, usefd bool) {
for _, r := range mmr.reporters {
r.BlockConn(dir, usefd)
}
}
// AllowStream is invoked when opening a stream is allowed
func (mmr *MultiMetricsReporter) AllowStream(p peer.ID, dir network.Direction) {
for _, r := range mmr.reporters {
r.AllowStream(p, dir)
}
}
// BlockStream is invoked when opening a stream is blocked
func (mmr *MultiMetricsReporter) BlockStream(p peer.ID, dir network.Direction) {
for _, r := range mmr.reporters {
r.BlockStream(p, dir)
}
}
// AllowPeer is invoked when attaching ac onnection to a peer is allowed
func (mmr *MultiMetricsReporter) AllowPeer(p peer.ID) {
for _, r := range mmr.reporters {
r.AllowPeer(p)
}
}
// BlockPeer is invoked when attaching ac onnection to a peer is blocked
func (mmr *MultiMetricsReporter) BlockPeer(p peer.ID) {
for _, r := range mmr.reporters {
r.BlockPeer(p)
}
}
// AllowProtocol is invoked when setting the protocol for a stream is allowed
func (mmr *MultiMetricsReporter) AllowProtocol(proto protocol.ID) {
for _, r := range mmr.reporters {
r.AllowProtocol(proto)
}
}
// BlockProtocol is invoked when setting the protocol for a stream is blocked
func (mmr *MultiMetricsReporter) BlockProtocol(proto protocol.ID) {
for _, r := range mmr.reporters {
r.BlockProtocol(proto)
}
}
// BlockedProtocolPeer is invoekd when setting the protocol for a stream is blocked at the per protocol peer scope
func (mmr *MultiMetricsReporter) BlockProtocolPeer(proto protocol.ID, p peer.ID) {
for _, r := range mmr.reporters {
r.BlockProtocolPeer(proto, p)
}
}
// AllowPService is invoked when setting the protocol for a stream is allowed
func (mmr *MultiMetricsReporter) AllowService(svc string) {
for _, r := range mmr.reporters {
r.AllowService(svc)
}
}
// BlockPService is invoked when setting the protocol for a stream is blocked
func (mmr *MultiMetricsReporter) BlockService(svc string) {
for _, r := range mmr.reporters {
r.BlockService(svc)
}
}
// BlockedServicePeer is invoked when setting the service for a stream is blocked at the per service peer scope
func (mmr *MultiMetricsReporter) BlockServicePeer(svc string, p peer.ID) {
for _, r := range mmr.reporters {
r.BlockServicePeer(svc, p)
}
}
// AllowMemory is invoked when a memory reservation is allowed
func (mmr *MultiMetricsReporter) AllowMemory(size int) {
for _, r := range mmr.reporters {
r.AllowMemory(size)
}
}
// BlockMemory is invoked when a memory reservation is blocked
func (mmr *MultiMetricsReporter) BlockMemory(size int) {
for _, r := range mmr.reporters {
r.BlockMemory(size)
}
}