Skip to content

Commit f4c389d

Browse files
committed
fix: print a result if tcpulse client receives SIGINT/SIGTERM
1 parent 6c27709 commit f4c389d

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

client.go

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -47,13 +47,13 @@ func waitLim(ctx context.Context, rl ratelimit.Limiter) error {
4747
return ctx.Err()
4848
default:
4949
}
50-
50+
5151
done := make(chan struct{})
5252
go func() {
5353
defer close(done)
5454
rl.Take()
5555
}()
56-
56+
5757
select {
5858
case <-done:
5959
return nil
@@ -100,6 +100,10 @@ func (c *Client) ConnectToAddresses(ctx context.Context, addrs []string) error {
100100
}
101101

102102
if err := eg.Wait(); err != nil {
103+
if errors.Is(err, context.Canceled) || errors.Is(err, context.DeadlineExceeded) {
104+
slog.Warn("context canceled", "error", err)
105+
return nil
106+
}
103107
return fmt.Errorf("connection error: %w", err)
104108
}
105109
return nil

0 commit comments

Comments
 (0)