Replace contextx.ShrinkDeadline with context.WithTimeout (#598)

This commit is contained in:
zjbztianya
2021-04-05 21:20:35 +08:00
committed by GitHub
parent 425430f67c
commit d88da4cc88
3 changed files with 3 additions and 7 deletions

View File

@@ -3,8 +3,6 @@ package fx
import (
"context"
"time"
"github.com/tal-tech/go-zero/core/contextx"
)
var (
@@ -23,7 +21,7 @@ func DoWithTimeout(fn func() error, timeout time.Duration, opts ...DoOption) err
for _, opt := range opts {
parentCtx = opt()
}
ctx, cancel := contextx.ShrinkDeadline(parentCtx, timeout)
ctx, cancel := context.WithTimeout(parentCtx, timeout)
defer cancel()
// create channel with buffer size 1 to avoid goroutine leak