Go gorm框架获取sql，tableName和主键值

1、背景

最近一直在忙着go的项目，眼看项目马上要结束了，来补一补博客。我们在使用go的gorm框架的时候，需要将运行的sql打印到我们指定的log文件中亦或是需要对我们的sql进行业务处理，这时候我们设置了LogMode为true时，就可以在控制台打印运行时sql，但这样并不能满足业务需求。

2、方法

首先我们需要在连接完数据库以后，开启我们的日志打印

// 连接数据库
mysqlDB, err := gorm.Open("mysql", DbUrl)
// 开启日志
mysqlDB.LogMode(true)

然后我们调用gorm的Callback方法， 定义自定义的回调方法

// 监听delete方法
mysqlDB.Callback().Delete().Register("gorm:deleteSql", callback)
// 监听update方法
mysqlDB.Callback().Update().Register("gorm:updateSql", callback)
// 监听create方法
mysqlDB.Callback().Create().Register("gorm:insertSql", callback)

定义回调方法

func callback(scope *gorm.Scope) {
// scope.SQL   我们需要拿到的sql，无参数
// scope.SQLVars 	sql的参数值
// scope.TableName()		sql对应的表名
// scope.PrimaryKeyValue()	sql对应的主键值
// 处理sql 拼装sql和value
	s := reflect.ValueOf(value)
	for i := 0; i < s.Len(); i++ {
		// 每次取代一个？
		dbSql = strings.Replace(dbSql, "?", "'%v'", 1)
		// 赋值
		dbSql = fmt.Sprintf(dbSql, s.Index(i))
	}
}
// 输出完整的sql

我们看下gorm.Scope的结构体

type Scope struct {
	Search          *search
	Value           interface{}
	SQL             string
	SQLVars         []interface{}
	db              *DB
	instanceID      string
	primaryKeyField *Field
	skipLeft        bool
	fields          *[]*Field
	selectAttrs     *[]string
}

可以看到里面包含了我们一般业务所需要的东西，我们需要做的就是拿出数值，然后去处理就可以了，拿到了我们所需要的值以后，后面我们怎么处理都可以啦。