最新公告
  • 欢迎您光临码农资源网,本站秉承服务宗旨 履行“站长”责任,销售只是起点 服务永无止境!加入我们
  • 在 Golang grpc 服务器方法的拦截器中重用日志客户端

    在 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 版本和特定包吗?

    想要了解更多内容,请持续关注码农资源网,一起探索发现编程世界的无限可能!
    本站部分资源来源于网络,仅限用于学习和研究目的,请勿用于其他用途。
    如有侵权请发送邮件至1943759704@qq.com删除

    码农资源网 » 在 Golang grpc 服务器方法的拦截器中重用日志客户端
    • 7会员总数(位)
    • 25846资源总数(个)
    • 0本周发布(个)
    • 0 今日发布(个)
    • 293稳定运行(天)

    提供最优质的资源集合

    立即查看 了解详情