Jelajahi Sumber

Simplify contextx.ShrinkDeadline (#596)

Oraoto 4 tahun lalu
induk
melakukan
425430f67c
1 mengubah file dengan 1 tambahan dan 8 penghapusan
  1. 1 8
      core/contextx/deadline.go

+ 1 - 8
core/contextx/deadline.go

@@ -8,12 +8,5 @@ import (
 // ShrinkDeadline returns a new Context with proper deadline base on the given ctx and timeout.
 // ShrinkDeadline returns a new Context with proper deadline base on the given ctx and timeout.
 // And returns a cancel function as well.
 // And returns a cancel function as well.
 func ShrinkDeadline(ctx context.Context, timeout time.Duration) (context.Context, func()) {
 func ShrinkDeadline(ctx context.Context, timeout time.Duration) (context.Context, func()) {
-	if deadline, ok := ctx.Deadline(); ok {
-		leftTime := time.Until(deadline)
-		if leftTime < timeout {
-			timeout = leftTime
-		}
-	}
-
-	return context.WithDeadline(ctx, time.Now().Add(timeout))
+	return context.WithTimeout(ctx, timeout)
 }
 }