这篇文章主要介绍了Go gRPC超时控制Deadlines怎么使用的相关知识,内容详细易懂,操作简单快捷,具有一定借鉴价值,相信大家阅读完这篇Go gRPC超时控制Deadlines怎么使用文章都会有所收获,下面我们一起来看看吧。
为什么要设置
当未设置 Deadlines 时,将采用默认的 DEADLINE_EXCEEDED(这个时间非常大)
如果产生了阻塞等待,就会造成大量正在进行的请求都会被保留,并且所有请求都有可能达到最大超时
这会使服务面临资源耗尽的风险,例如内存,这会增加服务的延迟,或者在最坏的情况下可能导致整个进程崩溃
gRPC
Client
func main() { ...ctx, cancel := context.WithDeadline(context.Background(), time.Now().Add(time.Duration(5 * time.Second)))defer cancel()client := pb.NewSearchServiceClient(conn)resp, err := client.Search(ctx, &pb.SearchRequest{Request: "gRPC",})if err != nil {statusErr, ok := status.FromError(err)if ok {if statusErr.Code() == codes.DeadlineExceeded {log.Fatalln("client.Search err: deadline")}}log.Fatalf("client.Search err: %v", err)}log.Printf("resp: %s", resp.GetResponse())}
context.WithDeadline:会返回最终上下文截止时间。第一个形参为父上下文,第二个形参为调整的截止时间。若父级时间早于子级时间,则以父级时间为准,否则以子级时间为最终截止时间
func WithDeadline(parent Context, d time.Time) (Context, CancelFunc) {if cur, ok := parent.Deadline(); ok && cur.Before(d) {// The current deadline is already sooner than the new one.return WithCancel(parent)}c := &timerCtx{cancelCtx: newCancelCtx(parent),deadline: d,}propagateCancel(parent, c)dur := time.Until(d)if dur <= 0 {c.cancel(true, DeadlineExceeded) // deadline has already passedreturn c, func() { c.cancel(true, Canceled) }}c.mu.Lock()defer c.mu.Unlock()if c.err == nil {c.timer = time.AfterFunc(dur, func() {c.cancel(true, DeadlineExceeded)})}return c, func() { c.cancel(true, Canceled) }}
context.WithTimeout:很常见的另外一个方法,是便捷操作。实际上是对于 WithDeadline 的封装
func WithTimeout(parent Context, timeout time.Duration) (Context, CancelFunc) {return WithDeadline(parent, time.Now().Add(timeout))}
status.FromError:返回 GRPCStatus 的具体错误码,若为非法,则直接返回 codes.Unknown
Server
type SearchService struct{}func (s *SearchService) Search(ctx context.Context, r *pb.SearchRequest) (*pb.SearchResponse, error) {for i := 0; i < 5; i++ {if ctx.Err() == context.Canceled {return nil, status.Errorf(codes.Canceled, "SearchService.Search canceled")}time.Sleep(1 * time.Second)}return &pb.SearchResponse{Response: r.GetRequest() + " Server"}, nil}func main() {...}
而在 Server 端,由于 Client 已经设置了截止时间。Server 势必要去检测它
否则如果 Client 已经结束掉了,Server 还傻傻的在那执行,这对资源是一种极大的浪费
因此在这里需要用 ctx.Err() == context.Canceled 进行判断,为了模拟场景我们加了循环和睡眠 ????
验证
重新启动 server.go 和 client.go,得到结果:
$ go run client.go2018/10/06 17:45:55 client.Search err: deadlineexit status 1
关于“Go gRPC超时控制Deadlines怎么使用”这篇文章的内容就介绍到这里,感谢各位的阅读!相信大家对“Go gRPC超时控制Deadlines怎么使用”知识都有一定的了解,大家如果还想学习更多知识,欢迎关注编程网行业资讯频道。