synced/coder
1
0
Fork 0
mirror of https://github.com/coder/coder.git synced 2025-07-13 21:36:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: add coder ping (#6161)

Browse Source
This commit is contained in:
Colin Adler
2023-02-13 10:38:00 -06:00
committed by GitHub
parent 2157bff13f
commit a54de6093b
14 changed files with 276 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
tailnet/conn.go
View File

@ -426,7 +426,7 @@ func (c *Conn) Status() *ipnstate.Status {
// Ping sends a Disco ping to the Wireguard engine.
// The bool returned is true if the ping was performed P2P.
func (c *Conn) Ping(ctx context.Context, ip netip.Addr) (time.Duration, bool, error) {
func (c *Conn) Ping(ctx context.Context, ip netip.Addr) (time.Duration, bool, *ipnstate.PingResult, error) {
errCh := make(chan error, 1)
prChan := make(chan *ipnstate.PingResult, 1)
go c.wireguardEngine.Ping(ip, tailcfg.PingDisco, func(pr *ipnstate.PingResult) {
@ -438,11 +438,11 @@ func (c *Conn) Ping(ctx context.Context, ip netip.Addr) (time.Duration, bool, er
})
select {
case err := <-errCh:
return 0, false, err
return 0, false, nil, err
case <-ctx.Done():
return 0, false, ctx.Err()
return 0, false, nil, ctx.Err()
case pr := <-prChan:
return time.Duration(pr.LatencySeconds * float64(time.Second)), pr.Endpoint != "", nil
return time.Duration(pr.LatencySeconds * float64(time.Second)), pr.Endpoint != "", pr, nil
}
}
@ -471,7 +471,7 @@ func (c *Conn) AwaitReachable(ctx context.Context, ip netip.Addr) bool {
ctx, cancel := context.WithTimeout(ctx, 5*time.Minute)
defer cancel()
_, _, err := c.Ping(ctx, ip)
_, _, _, err := c.Ping(ctx, ip)
if err == nil {
completed()
}