waku-org/go-waku

package common

import (
    "context"
    "time"

    "github.com/libp2p/go-libp2p/core/host"
    "github.com/libp2p/go-libp2p/core/peer"
    "github.com/libp2p/go-libp2p/core/peerstore"
    "github.com/libp2p/go-libp2p/p2p/protocol/ping"
)

type Pinger interface {
    PingPeer(ctx context.Context, peerInfo peer.AddrInfo) (time.Duration, error)
}

type defaultPingImpl struct {
    host host.Host
}

func NewDefaultPinger(host host.Host) Pinger {
    return &defaultPingImpl{
        host: host,
    }
}

func (d *defaultPingImpl) PingPeer(ctx context.Context, peerInfo peer.AddrInfo) (time.Duration, error) {
    d.host.Peerstore().AddAddrs(peerInfo.ID, peerInfo.Addrs, peerstore.AddressTTL)
    pingResultCh := ping.Ping(ctx, d.host, peerInfo.ID)
    select {
    case <-ctx.Done():
        return 0, ctx.Err()
    case r := <-pingResultCh:
        if r.Error != nil {
            return 0, r.Error
        }
        return r.RTT, nil
    }
}