在 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中文網其他相關文章！