重构seeder
This commit is contained in:
18
internal/core/seeder/utils.go
Normal file
18
internal/core/seeder/utils.go
Normal file
@@ -0,0 +1,18 @@
|
||||
package seeder
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
|
||||
"gorm.io/gorm"
|
||||
)
|
||||
|
||||
// isTableEmpty 检查给定模型对应的数据库表是否为空。
|
||||
// 注意:此函数需要从 database 包中移动过来,或者在 seeder 包中重新定义,
|
||||
// 为了避免循环依赖,这里选择在 seeder 包中重新定义。
|
||||
func isTableEmpty(tx *gorm.DB, model interface{}) (bool, error) {
|
||||
var count int64
|
||||
if err := tx.Model(model).Count(&count).Error; err != nil {
|
||||
return false, fmt.Errorf("查询表记录数失败: %w", err)
|
||||
}
|
||||
return count == 0, nil
|
||||
}
|
||||
Reference in New Issue
Block a user