Go语言GORM框架：高效过滤查询结果中的敏感数据
在使用GORM进行数据库查询时，经常会遇到需要处理包含敏感信息的结构体模型，例如merchant结构体，其中包含密码等敏感字段。本文介绍两种高效方法，在GORM查询结果中快速过滤掉这些敏感信息，只返回必要字段，例如用户名。
场景描述
假设我们有一个merchant结构体：
type merchant struct {
id int `json:"id" gorm:"comment:''"`
username string `json:"username" gorm:"comment:'用户名'"`
password string `json:"password" gorm:"comment:'密码'"`
// ...其他字段
}
我们需要在查询结果中排除password字段，只保留username等必要字段。传统的逐行处理方法过于繁琐：
m := map[string]interface{}{}
for _, item := range rows {
m["username"] = item.username
// ...其他字段处理
}
高效解决方案
方法一：利用GORM钩子函数
GORM的afterfind钩子函数允许我们在查询结果返回前修改数据。我们可以利用此功能清空敏感字段：
func (u *merchant) AfterFind(tx *gorm.DB) (err error) {
u.password = ""
return
}
此钩子函数会在每次查询后自动清空password字段，确保敏感信息不会出现在查询结果中。
方法二：使用自定义结构体接收部分字段
定义一个只包含必要字段的结构体来接收查询结果：
// 只包含部分字段的结构体
type MerchantBase struct {
Id int `json:"id"`
Username string `json:"username"`
}
// 查询并使用自定义结构体接收结果
db.Model(&merchant{}).Select("id, username").Find(&[]MerchantBase{})
通过Select方法指定需要查询的字段，并使用MerchantBase结构体接收结果，有效避免了敏感字段的返回。
两种方法都能有效地过滤敏感信息，选择哪种方法取决于具体需求和代码风格。 钩子函数方法更简洁，适用于所有查询；自定义结构体方法则更灵活，可以精确控制返回的字段。
