106 lines
2.8 KiB
Go
106 lines
2.8 KiB
Go
package main
|
||
|
||
import (
|
||
"fmt"
|
||
"log"
|
||
"time"
|
||
|
||
"github.com/Nanqipro/YunQue-Tech-Projects/ai_english_learning/serve/config"
|
||
"github.com/Nanqipro/YunQue-Tech-Projects/ai_english_learning/serve/internal/database"
|
||
"github.com/Nanqipro/YunQue-Tech-Projects/ai_english_learning/serve/internal/models"
|
||
"github.com/Nanqipro/YunQue-Tech-Projects/ai_english_learning/serve/internal/utils"
|
||
)
|
||
|
||
func main() {
|
||
// 加载配置
|
||
config.LoadConfig()
|
||
if config.GlobalConfig == nil {
|
||
log.Fatal("Failed to load configuration")
|
||
}
|
||
|
||
// 初始化数据库
|
||
database.InitDatabase()
|
||
defer database.CloseDatabase()
|
||
|
||
db := database.GetDB()
|
||
|
||
// 测试用户数据
|
||
testUsers := []struct {
|
||
Username string
|
||
Email string
|
||
Password string
|
||
Nickname string
|
||
}{
|
||
{
|
||
Username: "testuser",
|
||
Email: "test@example.com",
|
||
Password: "Test@123",
|
||
Nickname: "测试用户",
|
||
},
|
||
{
|
||
Username: "student1",
|
||
Email: "student1@example.com",
|
||
Password: "Student@123",
|
||
Nickname: "学生一号",
|
||
},
|
||
{
|
||
Username: "student2",
|
||
Email: "student2@example.com",
|
||
Password: "Student@123",
|
||
Nickname: "学生二号",
|
||
},
|
||
{
|
||
Username: "teacher",
|
||
Email: "teacher@example.com",
|
||
Password: "Teacher@123",
|
||
Nickname: "教师账号",
|
||
},
|
||
}
|
||
|
||
fmt.Println("开始创建测试用户...")
|
||
|
||
for _, userData := range testUsers {
|
||
// 检查用户是否已存在
|
||
var existingUser models.User
|
||
result := db.Where("username = ? OR email = ?", userData.Username, userData.Email).First(&existingUser)
|
||
|
||
if result.Error == nil {
|
||
fmt.Printf("用户 %s 已存在，跳过创建\n", userData.Username)
|
||
continue
|
||
}
|
||
|
||
// 加密密码
|
||
hashedPassword, err := utils.HashPassword(userData.Password)
|
||
if err != nil {
|
||
log.Printf("Failed to hash password for %s: %v", userData.Username, err)
|
||
continue
|
||
}
|
||
|
||
// 创建用户
|
||
user := models.User{
|
||
Username: userData.Username,
|
||
Email: userData.Email,
|
||
PasswordHash: hashedPassword,
|
||
Nickname: &userData.Nickname,
|
||
Status: "active",
|
||
CreatedAt: time.Now(),
|
||
UpdatedAt: time.Now(),
|
||
}
|
||
|
||
if err := db.Create(&user).Error; err != nil {
|
||
log.Printf("Failed to create user %s: %v", userData.Username, err)
|
||
continue
|
||
}
|
||
|
||
fmt.Printf("✓ 成功创建用户: %s (邮箱: %s, 密码: %s)\n",
|
||
userData.Username, userData.Email, userData.Password)
|
||
}
|
||
|
||
fmt.Println("\n测试用户创建完成！")
|
||
fmt.Println("\n可用的测试账号：")
|
||
fmt.Println("1. 用户名: testuser 邮箱: test@example.com 密码: Test@123")
|
||
fmt.Println("2. 用户名: student1 邮箱: student1@example.com 密码: Student@123")
|
||
fmt.Println("3. 用户名: student2 邮箱: student2@example.com 密码: Student@123")
|
||
fmt.Println("4. 用户名: teacher 邮箱: teacher@example.com 密码: Teacher@123")
|
||
}
|