first commit

go_backend/tools/generate_password.go

@@ -0,0 +1,42 @@
+package main
+
+import (
+	"fmt"
+	"golang.org/x/crypto/bcrypt"
+)
+
+// GeneratePassword 生成bcrypt加密密码
+func GeneratePassword(password string) (string, error) {
+	hashedPassword, err := bcrypt.GenerateFromPassword([]byte(password), bcrypt.DefaultCost)
+	if err != nil {
+		return "", err
+	}
+	return string(hashedPassword), nil
+}
+
+func main() {
+	// 为测试数据生成加密密码
+	passwords := []string{
+		"admin123",  // 企业管理员密码
+		"user123",   // 普通用户密码
+	}
+
+	fmt.Println("生成加密密码：")
+	fmt.Println("=====================================")
+	
+	for i, pwd := range passwords {
+		hashed, err := GeneratePassword(pwd)
+		if err != nil {
+			fmt.Printf("生成密码失败: %v\n", err)
+			continue
+		}
+		fmt.Printf("%d. 原始密码: %s\n", i+1, pwd)
+		fmt.Printf("   加密后: %s\n\n", hashed)
+	}
+	
+	fmt.Println("=====================================")
+	fmt.Println("使用说明：")
+	fmt.Println("1. 复制上面的加密密码")
+	fmt.Println("2. 在 test_data_ai_wht.sql 中替换对应的密码占位符")
+	fmt.Println("3. 执行 SQL 文件导入测试数据")
+}

go_backend/tools/generate_token.go

@@ -0,0 +1,39 @@
+package main
+
+import (
+	"ai_xhs/config"
+	"ai_xhs/utils"
+	"flag"
+	"fmt"
+	"log"
+)
+
+func main() {
+	// 解析命令行参数
+	env := flag.String("env", "dev", "运行环境: dev, prod")
+	employeeID := flag.Int("id", 1, "员工ID")
+	flag.Parse()
+
+	// 加载配置
+	if err := config.LoadConfig(*env); err != nil {
+		log.Fatalf("配置加载失败: %v", err)
+	}
+
+	// 生成Token
+	token, err := utils.GenerateToken(*employeeID)
+	if err != nil {
+		log.Fatalf("生成Token失败: %v", err)
+	}
+
+	fmt.Println("========================================")
+	fmt.Printf("环境: %s\n", *env)
+	fmt.Printf("员工ID: %d\n", *employeeID)
+	fmt.Println("========================================")
+	fmt.Printf("JWT Token:\n%s\n", token)
+	fmt.Println("========================================")
+	fmt.Println("\n使用方式:")
+	fmt.Println("在请求头中添加: Authorization: Bearer " + token)
+	fmt.Println("\ncURL示例:")
+	fmt.Printf("curl -H \"Authorization: Bearer %s\" http://localhost:8080/api/employee/profile\n", token)
+	fmt.Println("========================================")
+}