gRPC超時攔截器實(shí)現(xiàn)示例

更新時間：2022年06月15日 16:49:03 作者：尹東勛

這篇文章主要為大家介紹了gRPC超時攔截器實(shí)現(xiàn)示例，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進(jìn)步，早日升職加薪

介紹

本文介紹如何通過 rk-boot 快速搭建 gRPC 超時攔截器。

什么是 gRPC 超時攔截器？

攔截器會攔截 gRPC 請求，并根據(jù)策略返回超時錯誤。

安裝

go get github.com/rookie-ninja/rk-boot
go get github.com/rookie-ninja/rk-grpc

快速開始

使用 rk-boot 啟動的 gRPC 服務(wù)。

支持全局超時和 API 超時設(shè)定。

1.創(chuàng)建 boot.yaml

boot.yaml 文件告訴 rk-boot 如何啟動 gRPC 服務(wù)。

為了驗(yàn)證，我們啟動了 commonService，commonService 里包含了一系列常用 API，例如 /rk/v1/gc。

設(shè)定全局超時為 5秒，讓 GC 的超時時間定位 1 毫秒，GC 一般會超過 1 毫秒。

---
grpc:
  - name: greeter                                   # Required
    port: 8080                                      # Required
    enabled: true                                   # Required
    commonService:
      enabled: true                                 # Optional, Enable common service for testing
    interceptors:
      timeout:
        enabled: true                               # Optional, default: false
        timeoutMs: 5000                             # Optional, default: 5000
        paths: 
          - path: "/rk.api.v1.RkCommonService/Gc"   # Optional, default: ""
            timeoutMs: 1                            # Optional, default: 5000

2.創(chuàng)建 main.go

// Copyright (c) 2021 rookie-ninja
//
// Use of this source code is governed by an Apache-style
// license that can be found in the LICENSE file.
package main
import (
	"context"
	"github.com/rookie-ninja/rk-boot"
        _ "github.com/rookie-ninja/rk-grpc/boot"
)
// Application entrance.
func main() {
	// Create a new boot instance.
	boot := rkboot.NewBoot()
	// Bootstrap
	boot.Bootstrap(context.Background())
	// Wait for shutdown sig
	boot.WaitForShutdownSig(context.Background())
}

3.啟動 main.go

$ go run main.go

4.驗(yàn)證

發(fā)送 GC 請求。

$ grpcurl -plaintext localhost:8080 rk.api.v1.RkCommonService.Gc
ERROR:
  Code: Canceled
  Message: Request timed out!
  Details:
  1)	{"@type":"type.googleapis.com/rk.api.v1.ErrorDetail","code":1,"message":"[from-grpc] Request timed out!","status":"Canceled"}

$ curl -X GET localhost:8080/rk/v1/gc
{
    "error":{
        "code":408,
        "status":"Request Timeout",
        "message":"Request timed out!",
        "details":[
            {
                "code":1,
                "status":"Canceled",
                "message":"[from-grpc] Request timed out!"
            }
        ]
    }
}

以上就是gRPC超時攔截器實(shí)現(xiàn)示例的詳細(xì)內(nèi)容，更多關(guān)于gRPC超時攔截器的資料請關(guān)注腳本之家其它相關(guān)文章！

您可能感興趣的文章: