simplify timeoutinterceptor (#840)

Co-authored-by: chenmusheng <chenmusheng@laoyuegou.com>
This commit is contained in:
masonchen2014
2021-07-24 21:51:46 +08:00
committed by GitHub
parent ff7443c6a7
commit cb8d9d413a
4 changed files with 12 additions and 42 deletions

View File

@@ -17,27 +17,6 @@ func TimeoutInterceptor(timeout time.Duration) grpc.UnaryClientInterceptor {
ctx, cancel := context.WithTimeout(ctx, timeout)
defer cancel()
// create channel with buffer size 1 to avoid goroutine leak
done := make(chan error, 1)
panicChan := make(chan interface{}, 1)
go func() {
defer func() {
if p := recover(); p != nil {
panicChan <- p
}
}()
done <- invoker(ctx, method, req, reply, cc, opts...)
}()
select {
case p := <-panicChan:
panic(p)
case err := <-done:
return err
case <-ctx.Done():
return ctx.Err()
}
return invoker(ctx, method, req, reply, cc, opts...)
}
}