67 lines
2.4 KiB
Go
67 lines
2.4 KiB
Go
|
package torrent
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"encoding/base64"
|
||
|
|
|
||
|
|
"github.com/anacrolix/dht/v2"
|
||
|
|
"github.com/anacrolix/torrent"
|
||
|
|
"go.opentelemetry.io/otel/attribute"
|
||
|
|
"go.opentelemetry.io/otel/metric"
|
||
|
|
)
|
||
|
|
|
||
|
|
func registerTorrentMetrics(client *torrent.Client) error {
|
||
|
|
meterTotalPeers, _ := meter.Int64ObservableGauge("torrent.peers.total")
|
||
|
|
meterActivePeers, _ := meter.Int64ObservableGauge("torrent.peers.active")
|
||
|
|
meterSeeders, _ := meter.Int64ObservableGauge("torrent.seeders")
|
||
|
|
meterDownloaded, _ := meter.Int64ObservableGauge("torrent.downloaded", metric.WithUnit("By"))
|
||
|
|
meterIO, _ := meter.Int64ObservableGauge("torrent.io", metric.WithUnit("By"))
|
||
|
|
|
||
|
|
_, err := meter.RegisterCallback(func(ctx context.Context, o metric.Observer) error {
|
||
|
|
for _, v := range client.Torrents() {
|
||
|
|
as := attribute.NewSet(
|
||
|
|
attribute.String("infohash", v.InfoHash().HexString()),
|
||
|
|
attribute.String("name", v.Name()),
|
||
|
|
attribute.Int64("size", v.Length()),
|
||
|
|
)
|
||
|
|
stats := v.Stats()
|
||
|
|
o.ObserveInt64(meterTotalPeers, int64(stats.TotalPeers), metric.WithAttributeSet(as))
|
||
|
|
o.ObserveInt64(meterActivePeers, int64(stats.ActivePeers), metric.WithAttributeSet(as))
|
||
|
|
o.ObserveInt64(meterSeeders, int64(stats.ConnectedSeeders), metric.WithAttributeSet(as))
|
||
|
|
o.ObserveInt64(meterIO, stats.BytesRead.Int64(), metric.WithAttributeSet(as), metric.WithAttributes(attribute.String("direction", "download")))
|
||
|
|
o.ObserveInt64(meterIO, stats.BytesWritten.Int64(), metric.WithAttributeSet(as), metric.WithAttributes(attribute.String("direction", "upload")))
|
||
|
|
o.ObserveInt64(meterDownloaded, v.BytesCompleted(), metric.WithAttributeSet(as))
|
||
|
|
}
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}, meterTotalPeers, meterActivePeers, meterSeeders, meterIO, meterDownloaded)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func registerDhtMetrics(client *torrent.Client) error {
|
||
|
|
meterDhtNodes, _ := meter.Int64ObservableGauge("torrent.dht.nodes")
|
||
|
|
|
||
|
|
_, err := meter.RegisterCallback(func(ctx context.Context, o metric.Observer) error {
|
||
|
|
servers := client.DhtServers()
|
||
|
|
for _, dhtSrv := range servers {
|
||
|
|
stats, ok := dhtSrv.Stats().(dht.ServerStats)
|
||
|
|
if !ok {
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
id := dhtSrv.ID()
|
||
|
|
as := attribute.NewSet(
|
||
|
|
attribute.String("id", base64.StdEncoding.EncodeToString(id[:])),
|
||
|
|
attribute.String("address", dhtSrv.Addr().String()),
|
||
|
|
)
|
||
|
|
o.ObserveInt64(meterDhtNodes, int64(stats.Nodes), metric.WithAttributeSet(as))
|
||
|
|
}
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}, meterDhtNodes)
|
||
|
|
|
||
|
|
return err
|
||
|
|
}
|