日本免费高清视频-国产福利视频导航-黄色在线播放国产-天天操天天操天天操天天操|www.shdianci.com

學(xué)無先后,達(dá)者為師

網(wǎng)站首頁(yè) 編程語言 正文

Go?gRPC超時(shí)控制Deadlines用法詳解_Golang

作者:煎魚eddycjy ? 更新時(shí)間: 2022-08-07 編程語言

Deadlines

今天將介紹 gRPC Deadlines 的用法,這一個(gè)必備技巧。內(nèi)容也比較簡(jiǎn)單。

Deadlines 意指截止時(shí)間,在 gRPC 中強(qiáng)調(diào) TL;DR(Too long, Don't read)并建議始終設(shè)定截止日期,為什么呢?

為什么要設(shè)置

當(dāng)未設(shè)置 Deadlines 時(shí),將采用默認(rèn)的 DEADLINE_EXCEEDED(這個(gè)時(shí)間非常大)

如果產(chǎn)生了阻塞等待,就會(huì)造成大量正在進(jìn)行的請(qǐng)求都會(huì)被保留,并且所有請(qǐng)求都有可能達(dá)到最大超時(shí)

這會(huì)使服務(wù)面臨資源耗盡的風(fēng)險(xiǎn),例如內(nèi)存,這會(huì)增加服務(wù)的延遲,或者在最壞的情況下可能導(dǎo)致整個(gè)進(jìn)程崩潰

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:會(huì)返回最終上下文截止時(shí)間。第一個(gè)形參為父上下文,第二個(gè)形參為調(diào)整的截止時(shí)間。若父級(jí)時(shí)間早于子級(jí)時(shí)間,則以父級(jí)時(shí)間為準(zhǔn),否則以子級(jí)時(shí)間為最終截止時(shí)間
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 passed
		return 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:很常見的另外一個(gè)方法,是便捷操作。實(shí)際上是對(duì)于 WithDeadline 的封裝

func WithTimeout(parent Context, timeout time.Duration) (Context, CancelFunc) {
	return WithDeadline(parent, time.Now().Add(timeout))
}

status.FromError:返回 GRPCStatus 的具體錯(cuò)誤碼,若為非法,則直接返回 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 已經(jīng)設(shè)置了截止時(shí)間。Server 勢(shì)必要去檢測(cè)它

否則如果 Client 已經(jīng)結(jié)束掉了,Server 還傻傻的在那執(zhí)行,這對(duì)資源是一種極大的浪費(fèi)

因此在這里需要用 ctx.Err() == context.Canceled 進(jìn)行判斷,為了模擬場(chǎng)景我們加了循環(huán)和睡眠 ?

驗(yàn)證

重新啟動(dòng) server.go 和 client.go,得到結(jié)果:

$ go run client.go
2018/10/06 17:45:55 client.Search err: deadline
exit status 1

總結(jié)

本章節(jié)比較簡(jiǎn)單,你需要知道以下知識(shí)點(diǎn):

  • 怎么設(shè)置 Deadlines
  • 為什么要設(shè)置 Deadlines

你要清楚地明白到,gRPC Deadlines 是很重要的,否則這小小的功能點(diǎn)就會(huì)要了你生產(chǎn)的命。

示例源碼

https://github.com/eddycjy/go-grpc-example

原文鏈接:https://juejin.cn/post/6844904017017962504

欄目分類
最近更新