代码推送

This commit is contained in:
2026-09-09 02:39:57 +08:00
parent a112893947
commit 5e47757c73
24 changed files with 1156 additions and 226 deletions

View File

@@ -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
}