handle deadline exceeded in drain

johnstcn · johnstcn · commit 4f165bec834b · 2023-05-03T13:01:53.000+01:00
diff --git a/scaletest/trafficgen/run.go b/scaletest/trafficgen/run.go
@@ -170,6 +170,10 @@ func drainContext(ctx context.Context, src io.Reader, bufSize int64) error {
 				if xerrors.Is(err, io.EOF) {
 					return nil
 				}
+				// It's OK if the context is canceled.
+				if xerrors.Is(err, context.DeadlineExceeded) {
+					return nil
+				}
 				return err
 			}
 		}

Original file line number	Diff line number	Diff line change
`@@ -170,6 +170,10 @@ func drainContext(ctx context.Context, src io.Reader, bufSize int64) error {`
`170`	`170`	`if xerrors.Is(err, io.EOF) {`
`171`	`171`	`return nil`
`172`	`172`	`}`
	`173`	`+ // It's OK if the context is canceled.`
	`174`	`+ if xerrors.Is(err, context.DeadlineExceeded) {`
	`175`	`+ return nil`
	`176`	`+ }`
`173`	`177`	`return err`
`174`	`178`	`}`
`175`	`179`	`}`