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結構體接收結果，有效避免了敏感字段的返回。
兩種方法都能有效地過濾敏感信息，選擇哪種方法取決于具體需求和代碼風格。 鉤子函數方法更簡潔，適用于所有查詢；自定義結構體方法則更靈活，可以精確控制返回的字段。
? 版權聲明
文章版權歸作者所有，未經允許請勿轉載。
THE END
