代码推送
This commit is contained in:
@@ -9,7 +9,6 @@ import (
|
||||
"gotest/internal/model"
|
||||
|
||||
"github.com/glebarez/sqlite"
|
||||
"golang.org/x/crypto/bcrypt"
|
||||
"gorm.io/driver/mysql"
|
||||
"gorm.io/gorm"
|
||||
)
|
||||
@@ -61,7 +60,7 @@ func Init(cfg *config.DatabaseConfig) (*DB, error) {
|
||||
// auto_migrate 自动建表/更新表结构
|
||||
if cfg.AutoMigrate {
|
||||
if err := db.AutoMigrate(
|
||||
&model.User{}, &model.Config{},
|
||||
&model.User{}, &model.Config{}, &model.AccessLog{},
|
||||
&model.Company{}, &model.EnterpriseNews{}, &model.EnterpriseService{}, &model.TeamMember{},
|
||||
&model.ProductCategory{}, &model.Product{},
|
||||
&model.RestaurantCategory{}, &model.Restaurant{}, &model.MenuCategory{}, &model.MenuItem{},
|
||||
@@ -102,9 +101,9 @@ func Init(cfg *config.DatabaseConfig) (*DB, error) {
|
||||
return &DB{db}, nil
|
||||
}
|
||||
|
||||
// seedData 初始化基础数据,仅在用户表为空时执行
|
||||
// seedData 初始化基础数据
|
||||
func seedData(db *gorm.DB) error {
|
||||
// 检查是否已有用户,避免重复初始化
|
||||
// 检查是否已有用户
|
||||
var count int64
|
||||
if err := db.Model(&model.User{}).Count(&count).Error; err != nil {
|
||||
return err
|
||||
@@ -114,24 +113,7 @@ func seedData(db *gorm.DB) error {
|
||||
return nil
|
||||
}
|
||||
|
||||
// 创建默认管理员账号 admin / admin123
|
||||
hashedPassword, err := bcrypt.GenerateFromPassword([]byte("admin123"), bcrypt.DefaultCost)
|
||||
if err != nil {
|
||||
return fmt.Errorf("生成管理员密码哈希失败: %w", err)
|
||||
}
|
||||
|
||||
admin := model.User{
|
||||
Username: "admin",
|
||||
Password: string(hashedPassword),
|
||||
Role: "admin",
|
||||
Status: 1,
|
||||
}
|
||||
|
||||
if err := db.Create(&admin).Error; err != nil {
|
||||
return fmt.Errorf("创建默认管理员失败: %w", err)
|
||||
}
|
||||
log.Println("默认管理员账号已创建: admin / admin123(请及时修改密码)")
|
||||
|
||||
log.Println("未检测到管理员账号,请通过后台初始化页面创建管理员")
|
||||
return nil
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user