我正在用 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中文網其他相關文章!