Golang日志中如何记录数据库操作

在Golang日志中如何记录数据库操作

在Golang项目里，给数据库操作加上日志记录，几乎是每个后端开发者的标配动作。虽然标准库提供了基础工具，但真要说到功能丰富和灵活性，大家往往会转向logrus、zap这类第三方日志库。今天，我们就以logrus为例，看看怎么把数据库操作的“来龙去脉”清晰地记录在案。

话不多说，咱们先从环境准备开始。第一步，自然是把logrus库安装到项目中：

go get github.com/sirupsen/logrus

环境就绪后，我们来写一段具体的代码。下面这个示例，完整演示了如何连接数据库，并在执行插入操作的关键节点记录日志：

package main

import (
	"database/sql"
	"fmt"
	_ "github.com/go-sql-driver/mysql"
	"github.com/sirupsen/logrus"
)

func main() {
	// 初始化logrus
	logrus.SetFormatter(&logrus.TextFormatter{
		FullTimestamp: true,
	})

	// 连接数据库
	db, err := sql.Open("mysql", "username:password@tcp(localhost:3306)/dbname")
	if err != nil {
		logrus.Fatal(err)
	}
	defer db.Close()

	// 插入数据
	insertData(db, "John Doe", 30)
}

func insertData(db *sql.DB, name string, age int) {
	// 记录开始插入数据
	logrus.Info("开始插入数据")

	// 插入数据的SQL语句
	query := "INSERT INTO users (name, age) VALUES (?, ?)"
	result, err := db.Exec(query, name, age)
	if err != nil {
		logrus.WithFields(logrus.Fields{
			"error": err,
		}).Error("插入数据失败")
		return
	}

	// 获取插入数据的ID
	lastInsertID, err := result.LastInsertId()
	if err != nil {
		logrus.WithFields(logrus.Fields{
			"error": err,
		}).Error("获取插入数据ID失败")
		return
	}

	// 记录插入数据成功
	logrus.WithFields(logrus.Fields{
		"last_insert_id": lastInsertID,
	}).Info("插入数据成功")
}

梳理一下这段代码的逻辑：我们首先初始化了logrus，并启用了完整的时间戳格式。紧接着建立数据库连接，然后调用insertData函数。这个函数的设计很有代表性——它在执行操作前记录一条开始信息，在操作成功或失败时，分别记录不同级别的日志（Info或Error），并且通过WithFields方法附加上下文信息，比如错误详情或新生成的主键ID。

当然，这只是一个起点。实际项目中，你可以根据需求灵活调整日志的详细程度、级别以及附加的字段内容，让日志真正成为排查问题的得力助手。