unit-testing - 使用 glang 开发 Web 应用程序。由于单元测试，如何清除数据库中的脏数据

Question

我使用 gin 框架和 ORM 框架 gorm 开发 web 应用程序。数据库是 Mysql。

每个单元测试都会产生脏数据。所以我在创建对象时使用AfterCreate钩子记录表名和id。然后通过遍历记录列表删除脏数据。

我需要为每个模型添加钩子函数，例如：

func (f User) AfterCreate(tx *gorm.DB) (err error) {
    DBTestAddRecord([2]interface{}{f.ID, &User{}})
    return
}

它可以工作，但很烦人。还有其他解决方案吗？