首頁 > 後端開發 > Golang > 在 Golang grpc 伺服器方法的攔截器中重複使用日誌客戶端

在 Golang grpc 伺服器方法的攔截器中重複使用日誌客戶端

PHPz
發布: 2024-02-06 09:03:12
轉載
444 人瀏覽過

在 Golang grpc 服务器方法的拦截器中重用日志客户端

問題內容

我正在用 go 建立 grpc 伺服器。目前伺服器提供了三種方法:

  • 提交作業
  • 取消作業
  • 取得作業狀態

我正在使用 datadog 記錄一些指標,例如請求計數、持續時間等。 submitjob 方法如下所示:

func (s *myserver) submitjob(ctx context.context, request *submitjobrequest) (*submitjobresponse, error) {
        s.dd_client.logrequestcount("submitjob")
        start_time := time.now()
        defer s.dd_client.logrequestduration("submitjob", time.since(start_time))

        resp := somefunc()

    return resp, nil
}
登入後複製

這三種不同的伺服器方法中的日誌記錄程式碼幾乎相同。因此,我想知道如何避免這種重複。

我注意到golang grpc有攔截器的概念。基本上我可以定義一個攔截器函數,並使用它來進行伺服器方法呼叫的前/後處理。

按照文檔,我寫了一個攔截器,如下所示:

func unaryInterceptor(ctx context.Context,
    req interface{},
    info *grpc.UnaryServerInfo,
    handler grpc.UnaryHandler,
) (interface{}, error) {
    dd_client := NewDatadogClient()
    defer dd_client.Close()

        dd_client.LogRequestCount(info.FullMethod)

        start_time := time.Now()

    resp, err := handler(ctx, req)

        dd_client.LogRequestDuration(info.FullMethod, time.Since(start_time))

    return resp, err
}
登入後複製

問題是,每次呼叫攔截器時,它都會建立並銷毀一個新的 datadog 用戶端。我覺得這是沒有必要的。但是由於 unaryinterceptor 只是一個方法而不是一個類,所以我沒有看到一種方法可以創建一次 datadog 客戶端並稍後重用它?有辦法滿足我的需求嗎?


正確答案


是的,不要當場創建它,而是將其移動到另一個檔案/包,我們稱之為 datadog.go

#您需要將其宣告為單例,因此它在所有用法中只有 1 個指針,例如......

(我使用類型“datadogtype”來引用函數“newdatadogclient”返回的資料類型,請指定套件和版本以給出更精確的範例程式碼)

//your package code....


var dataDog DataDogType

//access it through a wrapper, the wrapper will init the var the first time, 
//the remaining times it will be reused

func GetDataDog()(*DataDogType) {

if dataDog == nil {
    dataDog = NewDatadogClient()
}


return dataDog

}
登入後複製

現在,根據您的用例,如果您將其作為指標返回並關閉它,您將關閉它的所有用途,因此返回指標本身並省略關閉它的部分,或者返回值,因此它是一個副本原來的

順便說一句,您可以分享一下您正在使用的 datadog 版本和特定套件嗎?

以上是在 Golang grpc 伺服器方法的攔截器中重複使用日誌客戶端的詳細內容。更多資訊請關注PHP中文網其他相關文章!

來源:stackoverflow.com
本網站聲明
本文內容由網友自願投稿,版權歸原作者所有。本站不承擔相應的法律責任。如發現涉嫌抄襲或侵權的內容,請聯絡admin@php.cn
熱門教學
更多>
最新下載
更多>
網站特效
網站源碼
網站素材
前端模板