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

编程并不是一个机械性的工作，而是需要有思考，有创新的工作，语法是固定的，但解决问题的思路则是依靠人的思维，这就需要我们坚持学习和更新自己的知识。今天golang学习网就整理分享《重复使用日志客户端在 Golang grpc 服务器方法的拦截器中》，文章讲解的知识点主要包括，如果你对Golang方面的知识点感兴趣，就不要错过golang学习网，在这可以对大家的知识积累有所帮助，助力开发能力的提升。

我正在用 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 服务器方法的拦截器中》文章吧，也可关注golang学习网公众号了解相关技术文章。