56 lines
1.3 KiB
Go
56 lines
1.3 KiB
Go
|
package p2p
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"fmt"
|
||
|
|
"log"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/hyprspace/hyprspace/config"
|
||
|
|
"github.com/libp2p/go-libp2p/core/event"
|
||
|
|
"github.com/libp2p/go-libp2p/core/host"
|
||
|
|
"github.com/libp2p/go-libp2p/core/network"
|
||
|
|
"github.com/libp2p/go-libp2p/p2p/protocol/ping"
|
||
|
|
)
|
||
|
|
|
||
|
|
func RouteMetricsService(ctx context.Context, host host.Host, cfg *config.Config) {
|
||
|
|
subCon, err := host.EventBus().Subscribe(new(event.EvtPeerConnectednessChanged))
|
||
|
|
if err != nil {
|
||
|
|
log.Fatal(err)
|
||
|
|
}
|
||
|
|
fmt.Println("[-] Route metrics service ready")
|
||
|
|
for {
|
||
|
|
select {
|
||
|
|
case <-ctx.Done():
|
||
|
|
return
|
||
|
|
case ev := <-subCon.Out():
|
||
|
|
evt := ev.(event.EvtPeerConnectednessChanged)
|
||
|
|
_, found := config.FindPeer(cfg.Peers, evt.Peer)
|
||
|
|
if found {
|
||
|
|
if evt.Connectedness == network.Connected {
|
||
|
|
ctx2, cancel := context.WithDeadline(ctx, time.Now().Add(30*time.Second))
|
||
|
|
ch := ping.Ping(ctx2, host, evt.Peer)
|
||
|
|
go func() {
|
||
|
|
t := time.NewTimer(15 * time.Second)
|
||
|
|
for {
|
||
|
|
select {
|
||
|
|
case <-t.C:
|
||
|
|
cancel()
|
||
|
|
case <-ctx2.Done():
|
||
|
|
return
|
||
|
|
case res := <-ch:
|
||
|
|
if res.Error == nil {
|
||
|
|
host.Peerstore().RecordLatency(evt.Peer, res.RTT)
|
||
|
|
}
|
||
|
|
time.Sleep(5 * time.Second)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}()
|
||
|
|
// wait a little before spawning another ping goroutine
|
||
|
|
time.Sleep(1 * time.Second)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|