在 Golang 中如何创建全局通用上下文来集成新遗迹？

在Golang实战开发的过程中，我们经常会遇到一些这样那样的问题，然后要卡好半天，等问题解决了才发现原来一些细节知识点还是没有掌握好。今天golang学习网就整理分享《在 Golang 中如何创建全局通用上下文来集成新遗迹？》，聊聊，希望可以帮助到正在努力赚钱的你。

我正在 main.go 中创建新的遗物事务，并且必须将其传递给处理程序，然后传递给控制器​​等。有没有一种方法可以全局定义它，然后可以在任何处理程序、控制器或数据库事务中访问？

import(
    "github.com/gin-gonic/gin"
    newrelic "github.com/newrelic/go-agent"
)

// main.go
func newrelicHandler() (gin.HandlerFunc, newrelic.Application) {
    newrelicConfig := newrelic.NewConfig("NEW_RELIC_APP_NAME", "NEW_RELIC_APP_KEY")
    app, err := newrelic.NewApplication(newrelicConfig)
    if err != nil {
        return func(c *gin.Context) {
            c.Next()
        }, app
    }
    return func(c *gin.Context) {
        txn := app.StartTransaction(c.Request.URL.Path, c.Writer, c.Request)
        c.Set("newRelicTransaction", txn)
        apm, ok := c.Get("newRelicTransaction")
        defer txn.End()
        c.Next()
    }, app
}

func main() {
    r := gin.New()
    x, app := newrelicHandler()
    r.Use(x)
}


// test/handler.go

func (t *TestHandler) Display(c *gin.Context) {
    apmTxn, ok := c.Get("newRelicTransaction")

    test, err := t.testCtrl.Display(apmTxn)
    if err != nil {
        return err
    }

    c.JSON(http.StatusOK, gin.H{"message": "success"})
}


// test/controller.go

func (t *TestCtrl) Display(txn interface{}) {
    apmTxn, ok := txn.(newrelic.Transaction)
    if !ok {
        fmt.Println("ERR")
    }
    segment := newrelic.StartSegment(apmTxn, "SomeSegmentName")
    // get data
    segment.End()
    return 
}

正确答案

避免使用全局上下文，而是在入口点创建一个全局上下文，然后将其作为参数传递给任何需要它的函数。

您可以使用gin框架提供的nrgin包。

并且在main()函数中

• 创建 newrelic 的实例 - newrelic.newapplication(cfg)
• 调用 - nrgin.middleware(app) 函数并传入 newrelic 实例。这会将 gin 交易上下文密钥 - newrelictransaction 添加到上下文中。
• 将第 2 步中的函数注册为所有路由的中间件 - router.use(nrgin.middleware(app))

然后，您可以将这个相同的上下文对象传递给可以接受 context.context 类型参数的其他函数，因为 gin.context 只是实现 go 的 context 接口。

示例代码

import "github.com/newrelic/go-agent/_integrations/nrgin/v1"

func main() {
    cfg := newrelic.NewConfig("Gin App", mustGetEnv("NEW_RELIC_LICENSE_KEY"))
    
    app, err := newrelic.NewApplication(cfg)
    if nil != err {
        fmt.Println(err)
    }

    router := gin.Default()
    router.Use(nrgin.Middleware(app))

    router.GET("/example-get", GetController)

    router.Run(":80")
}

func GetController(c *gin.Context) {
    if txn := nrgin.Transaction(c); nil != txn {
        txn.SetName("custom-name")
    }

    databaseGet(c)

    c.Writer.WriteString("example response...")
}

func databaseGet(c context.Context) {
    if txn := nrgin.Transaction(c); nil != txn {
        txn.SetName("custom-name")
    }

    c.Writer.WriteString("example response...")
}

到这里，我们也就讲完了《在 Golang 中如何创建全局通用上下文来集成新遗迹？》的内容了。个人认为，基础知识的学习和巩固，是为了更好的将其运用到项目中，欢迎关注golang学习网公众号，带你了解更多关于的知识点！