Compare commits

..

8 Commits

Author SHA1 Message Date
crosstyan 9a1dedf425 v1 2026-05-09 12:00:34 +08:00
lly d7885c442f fix: phone and email nullable. 2026-02-03 15:52:26 +08:00
lly 85ec3cba4a feat: account. 2026-02-03 15:43:42 +08:00
lly 1c38601fe0 fix: util. 2026-01-21 16:44:44 +08:00
lly 876916010f feat: count. 2026-01-21 16:39:19 +08:00
lly 1a252a12be feat: count. 2025-10-22 15:25:42 +08:00
lly 8d8dd26a2c feat: calculate when upload. 2025-08-05 10:55:19 +08:00
lly a14c553736 refactor: port. 2025-08-05 10:50:39 +08:00
20 changed files with 36088 additions and 11 deletions
Executable
+10
View File
@@ -0,0 +1,10 @@
#!/bin/bash
# docker exec hr_data_analyzer_db_1 mysqldump -uroot -proot training_db > train.sql
# docker exec hr_data_analyzer_db_1 pg_dump -U postgres training_db > train.sql
docker exec -e PGPASSWORD=root hr_data_analyzer_db_1 pg_dump \
-U postgres \
--data-only \
--inserts \
-t step_heart_rates -t step_stride_freqs -t step_train_records \
training_db > data_only.sql
-6
View File
@@ -1,6 +0,0 @@
database:
host: localhost #when use docker change to "db"
port: 5432
user: postgres
password: root
name: training_db
+270
View File
@@ -0,0 +1,270 @@
// controllers/ai.go
package controllers
import (
"context" // 在此处添加 context 导入
"fmt"
"github.com/gin-gonic/gin"
"github.com/sashabaranov/go-openai"
"hr_receiver/util"
"io"
"io/ioutil"
"log"
"mime/multipart"
"net/http"
"os"
"strings"
)
// 配置文件 (与 main.go 保持一致)
const (
BaseURL = "https://tokenhub.tencentmaas.com/v1/"
APIKey = "sk-KJNOFMltNzhSKh2IxW3G3MKmZF3q2RrOlvSk497CfTHp1Z4u" // 请替换为实际的 API Key
Model = "deepseek-v4-flash"
)
// readDocxContent 读取 .docx 文件并将其转换为结构化文本
// 修改为先保存临时文件再读取
func readDocxContent(fileHeader *multipart.FileHeader) (string, error) {
// 1. 创建临时文件
tempFile, err := os.CreateTemp("", "upload_*.docx")
if err != nil {
return "", fmt.Errorf("failed to create temporary file: %w", err)
}
defer os.Remove(tempFile.Name()) // 确保函数结束时删除临时文件
defer tempFile.Close()
// 2. 打开上传的文件流
src, err := fileHeader.Open()
if err != nil {
return "", fmt.Errorf("failed to open uploaded file: %w", err)
}
defer src.Close()
// 3. 将上传的文件内容复制到临时文件
_, err = io.Copy(tempFile, src)
if err != nil {
return "", fmt.Errorf("failed to copy file to temporary location: %w", err)
}
// 4. 获取临时文件的完整路径
tempFilePath := tempFile.Name()
str, err := util.DocxToStructuredPrompt(tempFilePath)
if err != nil {
return "", fmt.Errorf("failed to parse docx with go-docx: %w", err)
}
// 注意:表格、图片等复杂元素的处理可能需要更复杂的逻辑,这里仅处理简单文本
return str, nil
}
// readCSVContent 读取 .csv 文件内容
// 修改为先保存临时文件再读取
// readCSVContent 读取 .csv 文件内容
// 修改为先保存临时文件再读取,并增加了数据压缩逻辑以解决 token 超长问题
func readCSVContent(fileHeader *multipart.FileHeader) (string, error) {
// 1. 创建临时文件
tempFile, err := os.CreateTemp("", "upload_*.csv")
if err != nil {
return "", fmt.Errorf("failed to create temporary file: %w", err)
}
defer os.Remove(tempFile.Name()) // 确保函数结束时删除临时文件
defer tempFile.Close()
// 2. 打开上传的文件流
src, err := fileHeader.Open()
if err != nil {
return "", fmt.Errorf("failed to open uploaded file: %w", err)
}
defer src.Close()
// 3. 将上传的文件内容复制到临时文件
_, err = io.Copy(tempFile, src)
if err != nil {
return "", fmt.Errorf("failed to copy file to temporary location: %w", err)
}
// 4. 读取临时文件内容
content, err := ioutil.ReadFile(tempFile.Name())
if err != nil {
return "", fmt.Errorf("failed to read CSV content from temporary file: %w", err)
}
lines := strings.Split(string(content), "\n")
var compressedLines []string
for i, line := range lines {
// 1. 必须保留第一行(表头),让 AI 知道每一列是什么
if i == 0 {
compressedLines = append(compressedLines, line)
continue
}
// 2. 跳过空行
if strings.TrimSpace(line) == "" {
continue
}
// 3. 抽样逻辑:每 4 行保留 1 行
// i=1 是数据第1行i=2 是数据第2行...
// (i-1)%4 == 0 意味着数据第1, 5, 9, 13... 行会被保留
if (i-1)%4 == 0 {
compressedLines = append(compressedLines, line)
}
}
// --- 修改逻辑结束 ---
// 将处理后的行重新组合成字符串
resultContent := strings.Join(compressedLines, "\n")
// --- 新增逻辑结束 ---
return resultContent, nil
}
// buildAnalysisPrompt 构建发送给 AI 的提示词
func buildAnalysisPrompt(teachingPlanContent, heartRateContent string) string {
return fmt.Sprintf(`请根据以下体育课堂的教案和心率监测数据,生成一份详细的课堂分析报告:
## 教案内容:
%s
## 心率监测数据:
%s
这是一份幼儿园体育课的教案和课程心率监测数据,请帮对照分析课程教学效果,运动量和运动负荷情况是否科学,并提出课程设计的优化方案。
优化方案参考如下格式,教学过程需要详细一些:
# 幼儿体育教案(华侨大学版本)
| 项目 | 内容 |
| ------------ | -------------------------------- |
| **课程名** | |
| **年段** | 小 中 大 |
| **教师姓名** | |
| **时间** | 年 月 日 |
| **地点** | |
| **人数** | 男: 女: |
| **时长** | 分钟 |
| **天气预报** | 晴 雨 阴;温度 ℃ |
| **器材准备** | |
## 教学目标
| 类型 | 目标 |
| -------- | ------------ |
| **体能目标** | |
| **技能目标** | |
| **情感目标** | |
## 教学过程
| 阶段 | 阶段 | 项目名称 | 引导语及教学方法 | 队形/站位/留意点 | 目标心率区间 | 时间(分) |
| ---------- | -------- | ----------------------------- | ------------------------ | --------------------- | ------------ | ---------- |
| **准备部分** | 热身 | | | | | 3 |
| | 注意力游戏 | | | | | 3 |
| **正课部分** | 基本素质练习及常规意识培养环节 | | | | | 5 |
| | 复习环节 | | | | | 5 |
| | 新授环节 | | | | | 8 |
| **结束部分** | 社会性及情感目标游戏 | | | | | 4 |
| | 整理放松 | | | | | 2 |
请以专业体育教师的视角,提供详细的数据分析和教学建议。请直接输出报告内容,不要包含“好的”、“收到”、“作为一名...”等任何开场白或客套话。`, teachingPlanContent, heartRateContent)
}
// callAIForAnalysis 调用大模型进行分析
func callAIForAnalysis(prompt string) (string, error) {
sizeInBytes := len(prompt)
sizeInKB := float64(sizeInBytes) / 1024.0
// 在日志中打印大小,保留两位小数
log.Printf("=== 发送给 AI 的内容大小: %.2f KB (%d 字节) ===", sizeInKB, sizeInBytes)
config := openai.DefaultConfig(APIKey)
config.BaseURL = BaseURL
client := openai.NewClientWithConfig(config)
resp, err := client.CreateChatCompletion(
context.Background(),
openai.ChatCompletionRequest{
Model: Model,
Messages: []openai.ChatCompletionMessage{
{
Role: openai.ChatMessageRoleUser,
Content: prompt,
},
},
Temperature: 0.6, // 可调整
TopP: 0.6, // 可调整
MaxTokens: 4000, // 根据需要调整
},
)
if err != nil {
return "", fmt.Errorf("API call failed: %w", err)
}
if len(resp.Choices) == 0 {
return "", fmt.Errorf("no choices returned from API")
}
return resp.Choices[0].Message.Content, nil
}
// AnalyzeByAI Gin 控制器方法
func (tc *TrainingController) AnalyzeByAI(c *gin.Context) {
// 1. 解析多部分表单请求
form, err := c.MultipartForm()
if err != nil {
log.Printf("Error parsing multipart form: %v", err)
c.JSON(http.StatusBadRequest, gin.H{"error": fmt.Sprintf("Failed to parse form: %v", err)})
return
}
// 2. 获取文件列表
docxFiles := form.File["teaching_plan"] // 假设前端字段名为 'teaching_plan'
csvFiles := form.File["heart_rate_data"] // 假设前端字段名为 'heart_rate_data'
if len(docxFiles) == 0 || len(csvFiles) == 0 {
c.JSON(http.StatusBadRequest, gin.H{"error": "Missing required files: teaching_plan (.docx) or heart_rate_data (.csv)"})
return
}
// 3. 读取文件内容
// 注意:这里我们只取第一个上传的文件
teachingPlanFileHeader := docxFiles[0]
heartRateFileHeader := csvFiles[0]
teachingPlanContent, err := readDocxContent(teachingPlanFileHeader)
if err != nil {
log.Printf("Error reading teaching plan file (%s): %v", teachingPlanFileHeader.Filename, err)
c.JSON(http.StatusInternalServerError, gin.H{"error": fmt.Sprintf("Failed to process teaching plan file: %v", err)})
return
}
heartRateContent, err := readCSVContent(heartRateFileHeader)
if err != nil {
log.Printf("Error reading heart rate file (%s): %v", heartRateFileHeader.Filename, err)
c.JSON(http.StatusInternalServerError, gin.H{"error": fmt.Sprintf("Failed to process heart rate file: %v", err)})
return
}
// 4. 构建 Prompt
prompt := buildAnalysisPrompt(teachingPlanContent, heartRateContent)
// 5. 调用 AI 分析
analysisResult, err := callAIForAnalysis(prompt)
if err != nil {
log.Printf("Error calling AI for analysis: %v", err)
c.JSON(http.StatusInternalServerError, gin.H{"error": fmt.Sprintf("AI analysis failed: %v", err)})
return
}
//outputFile := ".md"
//ioutil.WriteFile(outputFile, []byte(analysisResult), 0644)
// 6. 返回结果
// 方式一:返回 JSON 结构
c.JSON(http.StatusOK, gin.H{
"status": "success",
"data": analysisResult,
})
}
+117
View File
@@ -0,0 +1,117 @@
package controllers
import (
"hr_receiver/config"
"hr_receiver/models"
"hr_receiver/util"
"net/http"
"github.com/gin-gonic/gin"
)
type LoginRequest struct {
Username string `json:"username" binding:"required"`
Password string `json:"password" binding:"required"`
}
type RegisterRequest struct {
Username string `json:"username" form:"username"`
Password string `json:"password" form:"password"`
}
type AuthResponse struct {
Token string `json:"token"`
User models.User `json:"user"`
}
// Register 用户注册
func Register(c *gin.Context) {
var req RegisterRequest
if err := c.ShouldBindJSON(&req); err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
return
}
// 检查用户名是否已存在
var existingUser models.User
if result := config.DB.Where("username = ?", req.Username).First(&existingUser); result.Error == nil {
c.JSON(http.StatusConflict, gin.H{"error": "Username already exists"})
return
}
// 创建新用户
user := models.User{
Username: req.Username,
Password: req.Password, // BeforeCreate钩子会自动加密
}
if result := config.DB.Create(&user); result.Error != nil {
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to create user"})
return
}
// 生成Token
token, err := util.GenerateToken(user.ID, user.Username)
if err != nil {
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to generate token"})
return
}
c.JSON(http.StatusCreated, AuthResponse{
Token: token,
User: user,
})
}
// Login 用户登录
func Login(c *gin.Context) {
var req LoginRequest
if err := c.ShouldBindJSON(&req); err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
return
}
// 查找用户
var user models.User
result := config.DB.Where("username = ?", req.Username).First(&user)
if result.Error != nil {
c.JSON(http.StatusUnauthorized, gin.H{"error": "Invalid username or password"})
return
}
// 验证密码
if !user.CheckPassword(req.Password) {
c.JSON(http.StatusUnauthorized, gin.H{"error": "Invalid username or password"})
return
}
// 生成JWT Token
token, err := util.GenerateToken(user.ID, user.Username)
if err != nil {
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to generate token"})
return
}
c.JSON(http.StatusOK, AuthResponse{
Token: token,
User: user,
})
}
// GetProfile 获取用户信息(需要认证)
func GetProfile(c *gin.Context) {
userID, exists := c.Get("userID")
if !exists {
c.JSON(http.StatusUnauthorized, gin.H{"error": "Unauthorized"})
return
}
var user models.User
if result := config.DB.First(&user, userID); result.Error != nil {
c.JSON(http.StatusNotFound, gin.H{"error": "User not found"})
return
}
c.JSON(http.StatusOK, user)
}
+40 -2
View File
@@ -33,6 +33,12 @@ func (tc *StepTrainingController) CreateTrainingRecord(c *gin.Context) {
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
return
}
username, exists := c.Get("username")
if !exists {
c.JSON(http.StatusUnauthorized, gin.H{"error": "无法获取用户信息,请重新登录"})
return
}
record.Username = username.(string)
// 使用事务保存数据[4](@ref)
err := tc.DB.Transaction(func(tx *gorm.DB) error {
@@ -79,6 +85,33 @@ func (tc *StepTrainingController) CreateTrainingRecord(c *gin.Context) {
return nil
})
// ====== 新增部分:启动异步回归计算 ======
go func() {
// 查询完整数据(需要关联的心率和步频数据)
var fullRecord models.StepTrainRecord
if err := tc.DB.
Where("train_id = ?", record.TrainId).
Preload("HeartRates", "heart_rate_type = ?", 1). // 只要有效心率
Preload("StrideFreqs", "predict_value = ?", 1). // 只要有效步频
First(&fullRecord).Error; err != nil {
log.Printf("训练记录%d查询失败无法计算回归: %v", record.TrainId, err)
return
}
// 检查数据是否满足计算条件
if len(fullRecord.HeartRates) == 0 || len(fullRecord.StrideFreqs) == 0 {
log.Printf("训练记录%d缺少心率或步频数据跳过回归计算", record.TrainId)
return
}
// 计算并保存回归结果
if _, err := tc.GetOrCalculateRegression(fullRecord.TrainId); err != nil {
log.Printf("训练记录%d回归计算失败: %v", fullRecord.TrainId, err)
} else {
log.Printf("训练记录%d回归结果已保存", fullRecord.TrainId)
}
}()
if err != nil {
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
return
@@ -96,6 +129,11 @@ func (tc *StepTrainingController) GetTrainingRecords(c *gin.Context) {
PageNum int `form:"pageNum,default=1"` // 页码,默认第一页
PageSize int `form:"pageSize,default=10"` // 每页数量默认10条
}
username, exists := c.Get("username")
if !exists {
c.JSON(http.StatusUnauthorized, gin.H{"error": "无法获取用户信息,请重新登录"})
return
}
var params PaginationParams
if err := c.ShouldBindQuery(&params); err != nil {
@@ -120,13 +158,13 @@ func (tc *StepTrainingController) GetTrainingRecords(c *gin.Context) {
)
// 获取总记录数
if err := tc.DB.Model(&models.StepTrainRecord{}).Count(&totalRows).Error; err != nil {
if err := tc.DB.Model(&models.StepTrainRecord{}).Where("username = ?", username).Count(&totalRows).Error; err != nil {
c.JSON(http.StatusInternalServerError, gin.H{"error": "获取记录总数失败"})
return
}
// 查询分页数据(按开始时间倒序排列)
result := tc.DB.
result := tc.DB.Where("username = ?", username).
Order("start_time DESC"). // 按开始时间倒序
Offset(offset).
Limit(params.PageSize).
+7951
View File
File diff suppressed because it is too large Load Diff
+1 -1
View File
@@ -4,7 +4,7 @@ services:
app:
build: .
ports:
- "8180:8080"
- "127.0.0.1:8180:8080"
depends_on:
db:
condition: service_healthy
+3
View File
@@ -3,9 +3,11 @@ module hr_receiver
go 1.23.3
require (
github.com/fumiama/go-docx v0.0.0-20250506085032-0c30fd09304b
github.com/gin-gonic/gin v1.10.0
github.com/golang-jwt/jwt/v5 v5.2.1
github.com/sajari/regression v1.0.1
github.com/sashabaranov/go-openai v1.41.2
github.com/spf13/viper v1.20.0
gonum.org/v1/gonum v0.16.0
gorm.io/driver/postgres v1.5.11
@@ -18,6 +20,7 @@ require (
github.com/cloudwego/base64x v0.1.4 // indirect
github.com/cloudwego/iasm v0.2.0 // indirect
github.com/fsnotify/fsnotify v1.8.0 // indirect
github.com/fumiama/imgsz v0.0.2 // indirect
github.com/gabriel-vasile/mimetype v1.4.3 // indirect
github.com/gin-contrib/sse v0.1.0 // indirect
github.com/go-playground/locales v0.14.1 // indirect
+6
View File
@@ -13,6 +13,10 @@ github.com/frankban/quicktest v1.14.6 h1:7Xjx+VpznH+oBnejlPUj8oUpdxnVs4f8XU8WnHk
github.com/frankban/quicktest v1.14.6/go.mod h1:4ptaffx2x8+WTWXmUCuVU6aPUX1/Mz7zb5vbUoiM6w0=
github.com/fsnotify/fsnotify v1.8.0 h1:dAwr6QBTBZIkG8roQaJjGof0pp0EeF+tNV7YBP3F/8M=
github.com/fsnotify/fsnotify v1.8.0/go.mod h1:8jBTzvmWwFyi3Pb8djgCCO5IBqzKJ/Jwo8TRcHyHii0=
github.com/fumiama/go-docx v0.0.0-20250506085032-0c30fd09304b h1:/mxSugRc4SgN7XgBtT19dAJ7cAXLTbPmlJLJE4JjRkE=
github.com/fumiama/go-docx v0.0.0-20250506085032-0c30fd09304b/go.mod h1:ssRF0IaB1hCcKIObp3FkZOsjTcAHpgii70JelNb4H8M=
github.com/fumiama/imgsz v0.0.2 h1:fAkC0FnIscdKOXwAxlyw3EUba5NzxZdSxGaq3Uyfxak=
github.com/fumiama/imgsz v0.0.2/go.mod h1:dR71mI3I2O5u6+PCpd47M9TZptzP+39tRBcbdIkoqM4=
github.com/gabriel-vasile/mimetype v1.4.3 h1:in2uUcidCuFcDKtdcBxlR0rJ1+fsokWf+uqxgUFjbI0=
github.com/gabriel-vasile/mimetype v1.4.3/go.mod h1:d8uq/6HKRL6CGdk+aubisF/M5GcPfT7nKyLpA0lbSSk=
github.com/gin-contrib/sse v0.1.0 h1:Y/yl/+YNO8GZSjAhjMsSuLt29uWRFHdHYUb5lYOV9qE=
@@ -77,6 +81,8 @@ github.com/sagikazarmark/locafero v0.7.0 h1:5MqpDsTGNDhY8sGp0Aowyf0qKsPrhewaLSsF
github.com/sagikazarmark/locafero v0.7.0/go.mod h1:2za3Cg5rMaTMoG/2Ulr9AwtFaIppKXTRYnozin4aB5k=
github.com/sajari/regression v1.0.1 h1:iTVc6ZACGCkoXC+8NdqH5tIreslDTT/bXxT6OmHR5PE=
github.com/sajari/regression v1.0.1/go.mod h1:NeG/XTW1lYfGY7YV/Z0nYDV/RGh3wxwd1yW46835flM=
github.com/sashabaranov/go-openai v1.41.2 h1:vfPRBZNMpnqu8ELsclWcAvF19lDNgh1t6TVfFFOPiSM=
github.com/sashabaranov/go-openai v1.41.2/go.mod h1:lj5b/K+zjTSFxVLijLSTDZuP7adOgerWeFyZLUhAKRg=
github.com/sourcegraph/conc v0.3.0 h1:OQTbbt6P72L20UqAkXXuLOj79LfEanQ+YQFNpLA9ySo=
github.com/sourcegraph/conc v0.3.0/go.mod h1:Sdozi7LEKbFPqYX2/J+iBAM6HpqSLTASQIKqDmF7Mt0=
github.com/spf13/afero v1.12.0 h1:UcOPyRBYczmFn6yvphxkn9ZEOY65cpwGKb5mL36mrqs=
+2 -1
View File
@@ -24,9 +24,10 @@ func main() {
&models.StepHeartRate{},
&models.StepStrideFreq{},
&models.RegressionResult{},
&models.User{},
)
// 启动服务
r := routes.SetupRouter()
r.Run(":8000")
r.Run(":8080")
}
+42
View File
@@ -0,0 +1,42 @@
package middleware
import (
"hr_receiver/util"
"net/http"
"strings"
"github.com/gin-gonic/gin"
)
func JWTAuth() gin.HandlerFunc {
return func(c *gin.Context) {
authHeader := c.GetHeader("Authorization")
if authHeader == "" {
c.JSON(http.StatusUnauthorized, gin.H{"error": "Authorization header required"})
c.Abort()
return
}
// Bearer Token格式
parts := strings.SplitN(authHeader, " ", 2)
if !(len(parts) == 2 && parts[0] == "Bearer") {
c.JSON(http.StatusUnauthorized, gin.H{"error": "Authorization header format must be Bearer {token}"})
c.Abort()
return
}
// 解析Token
claims, err := util.ParseToken(parts[1])
if err != nil {
c.JSON(http.StatusUnauthorized, gin.H{"error": "Invalid or expired token"})
c.Abort()
return
}
// 将用户信息存入上下文
c.Set("userID", claims.UserID)
c.Set("username", claims.Username)
c.Next()
}
}
+2
View File
@@ -7,6 +7,7 @@ type StepStrideFreq struct {
TrainId uint `gorm:"column:train_id; index" json:"trainId"` // 外键关联训练记录[4](@ref)
Time int64 `gorm:"type:bigint" json:"time"` // 保持与前端一致的毫秒时间戳[3](@ref)
Value int `gorm:"type:int" json:"value"`
Count int `gorm:"type:int" json:"count"`
PredictValue int `gorm:"type:int" json:"predictValue"`
Identifier string `gorm:"uniqueIndex;type:varchar(255)" json:"identifier"`
}
@@ -24,6 +25,7 @@ type StepHeartRate struct {
// 对应Flutter的TrainRecord结构
type StepTrainRecord struct {
gorm.Model
Username string `gorm:"size:50" json:"username"` // 对应Dart的tid字段
TrainId uint `gorm:"uniqueIndex" json:"tid"` // 对应Dart的tid字段
StartTime int64 `gorm:"type:bigint" json:"time"` // 开始时间戳
EndTime int64 `gorm:"type:bigint" json:"endTime"` // 结束时间戳[3](@ref)
+40
View File
@@ -0,0 +1,40 @@
package models
import (
"golang.org/x/crypto/bcrypt"
"gorm.io/gorm"
)
type User struct {
ID uint `gorm:"primaryKey" json:"id"`
Username string `gorm:"uniqueIndex;not null" json:"username"`
Email *string `gorm:"uniqueIndex;" json:"email"`
Phone *string `gorm:"uniqueIndex;" json:"phone"`
Password string `gorm:"not null" json:"-"`
CreatedAt int64 `json:"created_at"`
UpdatedAt int64 `json:"updated_at"`
}
// HashPassword 密码加密
func (u *User) HashPassword(password string) error {
bytes, err := bcrypt.GenerateFromPassword([]byte(password), 14)
if err != nil {
return err
}
u.Password = string(bytes)
return nil
}
// CheckPassword 验证密码
func (u *User) CheckPassword(password string) bool {
err := bcrypt.CompareHashAndPassword([]byte(u.Password), []byte(password))
return err == nil
}
// BeforeCreate 创建前钩子
func (u *User) BeforeCreate(tx *gorm.DB) (err error) {
if u.Password != "" {
return u.HashPassword(u.Password)
}
return nil
}
+7 -1
View File
@@ -20,9 +20,10 @@ func SetupRouter() *gin.Engine {
{
records.POST("", trainingController.CreateTrainingRecord)
records.GET("/analysis", trainingController.HandleCurveAnalysis)
records.POST("/analysis-by-ai", trainingController.AnalyzeByAI)
// 可扩展其他路由GET, PUT, DELETE等
}
steps := v1.Group("/step") //.Use(middleware.AuthMiddleware())
steps := v1.Group("/step").Use(middleware.JWTAuth())
{
steps.POST("", stepTrainController.CreateTrainingRecord)
steps.GET("train-records", stepTrainController.GetTrainingRecords)
@@ -30,6 +31,11 @@ func SetupRouter() *gin.Engine {
steps.GET("train-rank/:trainId", stepTrainController.GetTrainingRank)
// 可扩展其他路由GET, PUT, DELETE等
}
public := v1.Group("")
{
public.POST("/register", controllers.Register)
public.POST("/login", controllers.Login)
}
auth := v1.Group("/auth")
{
auth.GET("/token", func(c *gin.Context) {
Regular → Executable
View File
+188
View File
@@ -0,0 +1,188 @@
package main
import (
"context"
"fmt"
"io/ioutil"
"log"
"os"
"strings"
"time"
"github.com/sashabaranov/go-openai"
)
// 配置文件
const (
BaseURL = "https://api.lkeap.cloud.tencent.com/v1"
APIKey = "sk-Y4zjnwulSuSlf60mrzwCxq2ipktHSs4jZHgWeQOArWuWJEOd" // 替换为实际的API Key
Model = "deepseek-v3" // 推荐使用terminus版本
)
// 读取文件内容
func readFileContent(filename string) (string, error) {
content, err := ioutil.ReadFile(filename)
if err != nil {
return "", fmt.Errorf("读取文件 %s 失败: %v", filename, err)
}
return string(content), nil
}
// 构建分析提示词
func buildAnalysisPrompt(teachingPlanContent, heartRateContent string) string {
return fmt.Sprintf(`请根据以下体育课堂的教案和心率监测数据,生成一份详细的课堂分析报告:
## 教案内容:
%s
## 心率监测数据:
%s
这是一份幼儿园体育课的教案和课程心率监测数据,请帮对照分析课程教学效果,运动量和运动负荷情况是否科学,并提出课程设计的优化方案。
优化方案参考如下格式,教学过程需要详细一些:
# 幼儿体育教案(华侨大学版本)
| 项目 | 内容 |
| ------------ | -------------------------------- |
| **课程名** | |
| **年段** | 小 中 大 |
| **教师姓名** | |
| **时间** | 年 月 日 |
| **地点** | |
| **人数** | 男: 女: |
| **时长** | 分钟 |
| **天气预报** | 晴 雨 阴;温度 ℃ |
| **器材准备** | |
## 教学目标
| 类型 | 目标 |
| -------- | ------------ |
| **体能目标** | |
| **技能目标** | |
| **情感目标** | |
## 教学过程
| 阶段 | 阶段 | 项目名称 | 引导语及教学方法 | 队形/站位/留意点 | 目标心率区间 | 时间(分) |
| ---------- | -------- | ----------------------------- | ------------------------ | --------------------- | ------------ | ---------- |
| **准备部分** | 热身 | | | | | 3 |
| | 注意力游戏 | | | | | 3 |
| **正课部分** | 基本素质练习及常规意识培养环节 | | | | | 5 |
| | 复习环节 | | | | | 5 |
| | 新授环节 | | | | | 8 |
| **结束部分** | 社会性及情感目标游戏 | | | | | 4 |
| | 整理放松 | | | | | 2 |
请以专业体育教师的视角,提供详细的数据分析和教学建议。`, teachingPlanContent, heartRateContent)
}
// 调用大模型进行分析
func analyzeClassData(teachingPlanFile, heartRateFile string) (string, error) {
// 读取文件内容
teachingPlanContent, err := readFileContent(teachingPlanFile)
if err != nil {
return "", err
}
heartRateContent, err := readFileContent(heartRateFile)
if err != nil {
return "", err
}
// 构建客户端
config := openai.DefaultConfig(APIKey)
config.BaseURL = BaseURL
client := openai.NewClientWithConfig(config)
// 构建提示词
prompt := buildAnalysisPrompt(teachingPlanContent, heartRateContent)
// 调用API
resp, err := client.CreateChatCompletion(
context.Background(),
openai.ChatCompletionRequest{
Model: Model,
Messages: []openai.ChatCompletionMessage{
{
Role: openai.ChatMessageRoleUser,
Content: prompt,
},
},
Temperature: 0.6, // 使用默认值
TopP: 0.6, // 使用默认值
MaxTokens: 5000, // 适当限制输出长度
},
)
if err != nil {
return "", fmt.Errorf("API调用失败: %v", err)
}
if len(resp.Choices) == 0 {
return "", fmt.Errorf("未收到有效响应")
}
return resp.Choices[0].Message.Content, nil
}
// 保存分析结果到文件
func saveAnalysisResult(result, outputFile string) error {
// 添加时间戳和分隔符
timestamp := "生成时间: " + getCurrentTime()
separator := strings.Repeat("=", 80)
formattedResult := fmt.Sprintf("%s\n%s\n%s\n%s\n%s\n",
separator, timestamp, separator, result, separator)
return ioutil.WriteFile(outputFile, []byte(formattedResult), 0644)
}
// 获取当前时间(简化版)
func getCurrentTime() string {
// 实际使用时可以导入time包
return time.Now().Format("2006-01-02 15:04:05") // 替换为 time.Now().Format("2006-01-02 15:04:05")
}
func main() {
// 文件路径配置
teachingPlanFile := "D:\\projects\\IdeaProjects\\hr_receiver\\test\\b.md"
heartRateFile := "D:\\projects\\IdeaProjects\\hr_receiver\\test\\b.csv"
outputFile := "小班.md"
// 检查文件是否存在
if _, err := os.Stat(teachingPlanFile); os.IsNotExist(err) {
log.Fatalf("教案文件不存在: %s", teachingPlanFile)
}
if _, err := os.Stat(heartRateFile); os.IsNotExist(err) {
log.Fatalf("心率数据文件不存在: %s", heartRateFile)
}
fmt.Println("开始分析体育课堂数据...")
fmt.Printf("教案文件: %s\n", teachingPlanFile)
fmt.Printf("心率数据: %s\n", heartRateFile)
// 进行分析
result, err := analyzeClassData(teachingPlanFile, heartRateFile)
if err != nil {
log.Fatalf("分析失败: %v", err)
}
// 保存结果
err = saveAnalysisResult(result, outputFile)
if err != nil {
log.Fatalf("保存结果失败: %v", err)
}
fmt.Printf("分析完成!结果已保存到: %s\n", outputFile)
fmt.Println("\n分析报告摘要:")
fmt.Println(strings.Repeat("-", 50))
// 显示前200字符作为预览
preview := result
if len(result) > 200 {
preview = result[:200] + "..."
}
fmt.Println(preview)
}
+86
View File
@@ -0,0 +1,86 @@
package main
import (
"fmt"
"os"
"strings"
docx "github.com/fumiama/go-docx"
)
func DocxToStructuredPrompt(filename string) (string, error) {
f, err := os.Open(filename)
if err != nil {
return "", err
}
defer f.Close()
fi, err := f.Stat()
if err != nil {
return "", err
}
doc, err := docx.Parse(f, fi.Size())
if err != nil {
return "", err
}
var sb strings.Builder
sb.WriteString(fmt.Sprintf("# 文件:%s\n\n", filename))
for _, item := range doc.Document.Body.Items {
switch v := item.(type) {
case *docx.Paragraph:
// 直接用 fmt.Sprint 利用庫的 Stringer
text := fmt.Sprint(v)
text = strings.TrimSpace(text)
if text != "" {
sb.WriteString(text + "\n\n")
}
case *docx.Table:
sb.WriteString("## 表格\n")
// 先印表頭(可選)
sb.WriteString("| ")
// 假設第一行是表頭(很多文件如此),或全部當內容
for i, row := range v.TableRows {
var cells []string
for _, cell := range row.TableCells {
// 這裡是重點cell 本身沒有 String(),但可以遍歷它的 Paragraphs
var cellText strings.Builder
for _, p := range cell.Paragraphs {
cellText.WriteString(fmt.Sprint(p))
cellText.WriteString(" ")
}
cells = append(cells, strings.TrimSpace(cellText.String()))
}
sb.WriteString(strings.Join(cells, " | "))
sb.WriteString(" |\n")
// 如果想加 markdown 表頭分隔線(只在第一行後加)
if i == 0 {
sb.WriteString("| " + strings.Repeat("--- | ", len(cells)) + "\n")
}
}
sb.WriteString("\n")
default:
// 忽略圖片、頁首等
}
}
return sb.String(), nil
}
func main1() {
// 測試用
prompt, err := docxToStructuredPrompt("D:\\myDocument\\tencent\\weChat\\WeChat Files\\wxid_pv6rg3z2l28y22\\FileStorage\\File\\2026-01\\(改)小班体育活动《蚂蚁运粮》(泉秀实幼吴思莹).docx")
if err != nil {
fmt.Println("錯誤:", err)
return
}
fmt.Println(prompt)
}
+27191
View File
File diff suppressed because it is too large Load Diff
+76
View File
@@ -0,0 +1,76 @@
package util
import (
"fmt"
"os"
"strings"
docx "github.com/fumiama/go-docx"
)
func DocxToStructuredPrompt(filename string) (string, error) {
f, err := os.Open(filename)
if err != nil {
return "", err
}
defer f.Close()
fi, err := f.Stat()
if err != nil {
return "", err
}
doc, err := docx.Parse(f, fi.Size())
if err != nil {
return "", err
}
var sb strings.Builder
sb.WriteString(fmt.Sprintf("# 文件:%s\n\n", filename))
for _, item := range doc.Document.Body.Items {
switch v := item.(type) {
case *docx.Paragraph:
// 直接用 fmt.Sprint 利用庫的 Stringer
text := fmt.Sprint(v)
text = strings.TrimSpace(text)
if text != "" {
sb.WriteString(text + "\n\n")
}
case *docx.Table:
sb.WriteString("## 表格\n")
// 先印表頭(可選)
sb.WriteString("| ")
// 假設第一行是表頭(很多文件如此),或全部當內容
for i, row := range v.TableRows {
var cells []string
for _, cell := range row.TableCells {
// 這裡是重點cell 本身沒有 String(),但可以遍歷它的 Paragraphs
var cellText strings.Builder
for _, p := range cell.Paragraphs {
cellText.WriteString(fmt.Sprint(p))
cellText.WriteString(" ")
}
cells = append(cells, strings.TrimSpace(cellText.String()))
}
sb.WriteString(strings.Join(cells, " | "))
sb.WriteString(" |\n")
// 如果想加 markdown 表頭分隔線(只在第一行後加)
if i == 0 {
sb.WriteString("| " + strings.Repeat("--- | ", len(cells)) + "\n")
}
}
sb.WriteString("\n")
default:
// 忽略圖片、頁首等
}
}
return sb.String(), nil
}
+56
View File
@@ -0,0 +1,56 @@
package util
import (
"errors"
"time"
"github.com/golang-jwt/jwt/v5"
)
var ApiSecret = "your-super-secret-key" // 预共享密钥
type Claims struct {
UserID uint `json:"user_id"`
Username string `json:"username"`
jwt.RegisteredClaims
}
// GenerateToken 生成JWT Token
func GenerateToken(userID uint, username string) (string, error) {
expirationTime := time.Now().Add(24 * 30 * time.Hour) // Token有效期24小时
//expirationTime := time.Now().Add(1 * time.Second) // Token有效期24小时
claims := &Claims{
UserID: userID,
Username: username,
RegisteredClaims: jwt.RegisteredClaims{
ExpiresAt: jwt.NewNumericDate(expirationTime),
IssuedAt: jwt.NewNumericDate(time.Now()),
NotBefore: jwt.NewNumericDate(time.Now()),
Issuer: "your-app-name",
},
}
token := jwt.NewWithClaims(jwt.SigningMethodHS256, claims)
tokenString, err := token.SignedString([]byte(ApiSecret))
return tokenString, err
}
// ParseToken 解析JWT Token
func ParseToken(tokenStr string) (*Claims, error) {
claims := &Claims{}
token, err := jwt.ParseWithClaims(tokenStr, claims, func(token *jwt.Token) (interface{}, error) {
return []byte(ApiSecret), nil
})
if err != nil {
return nil, err
}
if !token.Valid {
return nil, errors.New("invalid token")
}
return claims, nil
}