golang 集成sentry:PostgreSQL

本文介绍通过 pgx库集成sentry, pgx 特点是速度快,可以完全控制,并且支持所有 postgres 数据类型。

pgx地址: https://github.com/jackc/pgx

最简单的接入方式是自定义一个logger, 然后注入到pgx, 完整示例如下

go 复制代码
package main

import (
	"context"
	"log"
	"os"
	"time"

	"github.com/getsentry/sentry-go"
	"github.com/jackc/pgx/v4"
	"github.com/jackc/pgx/v4/log/logrusadapter"
	"github.com/jackc/pgx/v4/pgxpool"
	"github.com/sirupsen/logrus"
)

// TracingLogger does two things at once, logging and tracing
type TracingLogger struct {
	logger *logrusadapter.Logger
}

func NewTracingLogger(logger logrus.FieldLogger) *TracingLogger {
	return &TracingLogger{logger: logrusadapter.NewLogger(logger)}
}

func (l *TracingLogger) Log(ctx context.Context, level pgx.LogLevel, msg string, data map[string]interface{}) {
	l.logger.Log(ctx, level, msg, data)

	span := sentry.StartSpan(ctx, "postgres "+msg)
	defer span.Finish()

	if data != nil {
		if span.Data == nil {
			span.Data = make(map[string]interface{})
		}
		for key, value := range data {
			value, ok := value.(string)
			if ok {
				span.Data[key] = value
			}
		}
		// since this method is called after a query is done,
		// we need to correct span start time
		execTime, ok := data["time"].(time.Duration)
		if ok {
			span.StartTime = time.Now().Add(-execTime)
		}
	}
}

func main() {
	err := sentry.Init(sentry.ClientOptions{
		Debug:              true,
		Dsn:                "https://a5eac4fa3396cbfac8fb4baa6a9c03a3@o4504291071688704.ingest.sentry.io/4506715873804288",
		AttachStacktrace:   true,
		EnableTracing:      true,
		SampleRate:         1.0,
		TracesSampleRate:   1.0,
		ProfilesSampleRate: 1.0,
	})
	if err != nil {
		log.Fatalf("sentry.Init: %s", err)
	}
	defer sentry.Flush(2 * time.Second)

	config, err := pgxpool.ParseConfig(os.Getenv("DATABASE_URL"))
	if err != nil {
		log.Fatal(err)
	}

	logger := logrus.New()

	config.ConnConfig.Logger = NewTracingLogger(logger)

	conn, err := pgxpool.ConnectConfig(ctx, config)

	if err != nil {
		log.Fatal(err)
	}

	defer conn.Close()

	ctx := context.Background()
	tx := sentry.StartTransaction(ctx, "test_pgx")
	defer tx.Finish()

	ctx = tx.Context()

	err = conn.QueryRow(ctx, "SELECT 1").Scan()
	if err != nil {
		log.Fatal(err)
	}
}

参考:

https://anymindgroup.com/news/tech-blog/15724/

相关推荐
程序员爱钓鱼2 小时前
Go语言实战案例-滑动窗口最大值
后端·google·go
zhuyasen10 小时前
Go 分布式任务和定时任务太难?sasynq 让异步任务从未如此简单
后端·go
同志们17 小时前
LiteLLM Go: 多平台LLM客户端统一接口实现
llm·go
用户895356032822017 小时前
LaPluma : 一个轻盈的 Go 数据流处理库
go
vv安的浅唱17 小时前
Golang基础笔记七之指针,值类型和引用类型
后端·go
猫头虎20 小时前
2025年02月11日 Go生态洞察:Go 1.24 发布亮点全面剖析
开发语言·后端·python·golang·go·beego·go1.19
DemonAvenger1 天前
HTTP/2在Go中的实现与优化
网络协议·架构·go
程序员爱钓鱼1 天前
Go语言实战案例-括号匹配算法
后端·google·go
程序员爱钓鱼1 天前
Go语言实战案例-判断字符串是否由另一个字符串的字母组成
后端·google·go
ん贤2 天前
Zap日志库指南
后端·go