refactor: config.sample.yaml.

refactor: data analyze result.
refactor: data analyze.
2025-04-07 09:48:49 +08:00 · 2025-04-01 15:35:24 +08:00 · 2025-04-01 10:13:38 +08:00
8 changed files with 283 additions and 39 deletions
--- a/.gitignore
+++ b/.gitignore
@ -1,3 +1,4 @@
 .idea
 hr_receiver.iml
 main.go.bak
+config.yaml
--- a/config.sample.yaml
+++ b/config.sample.yaml
--- a/controllers/train.go
+++ b/controllers/train.go
@ -1,14 +1,33 @@
 package controllers

 import (
+	"gonum.org/v1/gonum/floats"
+	"gonum.org/v1/gonum/stat"
+	"gonum.org/v1/gonum/stat/distuv"
+)
+
+import (
+	"errors"
 	"github.com/gin-gonic/gin"
 	"gorm.io/gorm"
 	"gorm.io/gorm/clause"
 	"hr_receiver/config"
 	"hr_receiver/models"
+	"math"
 	"net/http"
 )

+var analyzeRunTypes = []string{"6.5开始", "7开始", "8开始"} // 替换为你的具体值
+
+func contains(s string) bool {
+	for _, item := range analyzeRunTypes {
+		if item == s {
+			return true
+		}
+	}
+	return false
+}
+
 type TrainingController struct {
 	DB *gorm.DB
 }
@ -57,13 +76,12 @@ func (tc *TrainingController) CreateTrainingRecord(c *gin.Context) {
 				return err
 			}
 		}
-
-		//// 保存腰带关联关系
-		//if len(record.Belts) > 0 {
-		//	if err := tx.Model(&record).Association("Belts").Replace(record.Belts); err != nil {
-		//		return err
-		//	}
-		//}
+		if contains(record.RunType) {
+			err := tc.heartRateAnalyze(tx, record)
+			if err != nil {
+				return err
+			}
+		}

 		return nil
 	})
@ -79,25 +97,234 @@ func (tc *TrainingController) CreateTrainingRecord(c *gin.Context) {
 	})
 }

-func ReceiveTrainingData(c *gin.Context) {
-	var data models.TrainingData
-
-	if err := c.ShouldBindJSON(&data); err != nil {
-		c.JSON(http.StatusBadRequest, gin.H{
-			"error": "Invalid request body: " + err.Error(),
-		})
-		return
-	}
-
-	if result := config.DB.Create(&data); result.Error != nil {
-		c.JSON(http.StatusInternalServerError, gin.H{
-			"error": "Failed to save data: " + result.Error.Error(),
-		})
-		return
-	}
-
-	c.JSON(http.StatusCreated, gin.H{
-		"message": "Data saved successfully",
-		"id":      data.ID,
-	})
+// analysis_response.go
+type AnalysisResponse struct {
+	Status  string `json:"status"`  // 状态码
+	Message string `json:"message"` // 附加信息
+	Data    struct {
+		Mean      float64      `json:"mean"`      // 均值
+		StdDev    float64      `json:"stdDev"`    // 标准差
+		Histogram []HistoBin   `json:"histogram"` // 直方图数据
+		Curve     []CurvePoint `json:"curve"`     // 正态曲线数据
+	} `json:"data"`
+}
+
+type HistoBin struct {
+	BinStart float64 `json:"binStart"` // 区间起始值
+	BinEnd   float64 `json:"binEnd"`   // 区间结束值
+	Count    int     `json:"count"`    // 该区间计数
+}
+
+type CurvePoint struct {
+	X float64 `json:"x"` // X坐标
+	Y float64 `json:"y"` // Y坐标
+}
+
+// analysis_handler.go
+func (tc *TrainingController) HandleCurveAnalysis(c *gin.Context) {
+	// 获取数据库连接（根据实际项目配置调整）
+
+	// 1. 获取历史数据
+	aValues, err := collectCurveParams(tc.DB)
+	if err != nil {
+		c.JSON(500, gin.H{
+			"status":  "error",
+			"message": "数据查询失败: " + err.Error(),
+		})
+		return
+	}
+
+	// 2. 检查数据有效性
+	if len(aValues) < 10 { // 至少需要10个样本
+		c.JSON(400, gin.H{
+			"status":  "fail",
+			"message": "数据量不足，至少需要10个样本",
+		})
+		return
+	}
+
+	// 3. 计算统计量
+	mean, stddev := calculateStats(aValues)
+
+	// 4. 生成直方图数据
+	histogram := calculateHistogram(aValues, 20) // 20个分箱
+
+	// 5. 生成正态曲线
+	x, y := generateNormalCurve(mean, stddev, 100)
+
+	// 6. 构造响应
+	response := AnalysisResponse{
+		Status:  "success",
+		Message: "分析完成",
+		Data: struct {
+			Mean      float64      `json:"mean"`
+			StdDev    float64      `json:"stdDev"`
+			Histogram []HistoBin   `json:"histogram"`
+			Curve     []CurvePoint `json:"curve"`
+		}{
+			Mean:      mean,
+			StdDev:    stddev,
+			Histogram: histogram,
+			Curve:     convertToCurvePoints(x, y),
+		},
+	}
+
+	c.JSON(200, response)
+}
+
+// 直方图计算函数
+func calculateHistogram(data []float64, bins int) []HistoBin {
+	minV, maxV := floats.Min(data), floats.Max(data)
+	binWidth := (maxV - minV) / float64(bins)
+
+	counts := make([]int, bins)
+	for _, v := range data {
+		idx := int((v - minV) / binWidth)
+		if idx == bins { // 处理最大值刚好等于maxV的情况
+			idx--
+		}
+		counts[idx]++
+	}
+
+	histogram := make([]HistoBin, bins)
+	for i := 0; i < bins; i++ {
+		start := minV + float64(i)*binWidth
+		end := minV + float64(i+1)*binWidth
+		histogram[i] = HistoBin{
+			BinStart: start,
+			BinEnd:   end,
+			Count:    counts[i],
+		}
+	}
+	return histogram
+}
+
+// 转换曲线数据格式
+func convertToCurvePoints(x, y []float64) []CurvePoint {
+	points := make([]CurvePoint, len(x))
+	for i := range x {
+		points[i] = CurvePoint{
+			X: x[i],
+			Y: y[i],
+		}
+	}
+	return points
+}
+
+func (tc *TrainingController) heartRateAnalyze(tx *gorm.DB, record models.TrainRecord) error {
+	startTime := record.StartTime
+
+	// 获取所有唯一的beltID
+	var beltIDs []uint
+	tx.Model(&models.HeartRate{}).Where("train_id = ?", record.TrainId).
+		Select("DISTINCT belt_id").Pluck("belt_id", &beltIDs)
+
+	// 对每个belt计算
+	for _, bid := range beltIDs {
+		// 计算平均心率
+		ranges := getTimeRanges(startTime)
+		averages, err := calculateAverages(tx, record.TrainId, bid, ranges)
+		if err != nil {
+			return err
+		}
+
+		// 曲线拟合
+		x := []float64{2, 4, 6}
+		y := []float64{averages["2min"], averages["4min"], averages["6min"]}
+		a, _ := quadraticFit(x, y)
+
+		// 存储结果
+		analysis := models.BeltAnalysis{
+			TrainID:     record.TrainId,
+			RunType:     record.RunType,
+			BeltID:      bid,
+			Avg2min:     averages["2min"],
+			Avg4min:     averages["4min"],
+			Avg6min:     averages["6min"],
+			CurveParamA: a,
+		}
+		if err := tx.Create(&analysis).Error; err != nil {
+			return err
+		}
+	}
+	return nil
+}
+
+func collectCurveParams(tx *gorm.DB) ([]float64, error) {
+	var aValues []float64
+	// 查询所有记录的 CurveParamA 字段
+	err := tx.Model(&models.BeltAnalysis{}).Pluck("curve_param_a", &aValues).Error
+	if err != nil {
+		return nil, err
+	}
+	return aValues, nil
+}
+func calculateStats(data []float64) (mean, stddev float64) {
+	mean = stat.Mean(data, nil)
+	variance := stat.Variance(data, nil)
+	stddev = math.Sqrt(variance)
+	return
+}
+
+func generateNormalCurve(mean, stddev float64, numPoints int) (x, y []float64) {
+	normal := distuv.Normal{
+		Mu:    mean,
+		Sigma: stddev,
+	}
+
+	minV := mean - 3*stddev // 从均值-3σ开始
+	maxV := mean + 3*stddev // 到均值+3σ结束
+	step := (maxV - minV) / float64(numPoints-1)
+
+	for i := 0; i < numPoints; i++ {
+		xi := minV + float64(i)*step
+		yi := normal.Prob(xi)
+		x = append(x, xi)
+		y = append(y, yi)
+	}
+	return
+}
+func calculateAverages(tx *gorm.DB, trainID uint, beltID uint, ranges map[string]TimeRange) (map[string]float64, error) {
+	averages := make(map[string]float64)
+	for key, tr := range ranges {
+		var avg float64
+		// 使用GORM Raw SQL提高效率[6,10](@ref)
+		err := tx.Raw(`
+			SELECT COALESCE(AVG(value), 0) AS avg  -- 关键修复
+            FROM heart_rates 
+            WHERE train_id = ? 
+            AND belt_id = ? 
+            AND time BETWEEN ? AND ?`,
+			trainID, beltID, tr.Start, tr.End,
+		).Scan(&avg).Error
+		if err != nil {
+			return nil, err
+		}
+		averages[key] = avg
+	}
+	return averages, nil
+}
+
+func quadraticFit(x []float64, y []float64) (float64, error) {
+	// 使用三点计算y=ax²+b的a值（x=[2,4,6]对应分钟）
+	if len(x) != 3 || len(y) != 3 {
+		return 0, errors.New("需要三个点")
+	}
+	// 构造方程组矩阵（简化计算）
+	a := (y[2] - 2*y[1] + y[0]) / (x[2]*x[2] - 2*x[1]*x[1] + x[0]*x[0])
+	return a, nil
+}
+
+type TimeRange struct {
+	Start int64 // 毫秒时间戳起点
+	End   int64 // 毫秒时间戳终点
+}
+
+func getTimeRanges(startTime int64) map[string]TimeRange {
+	// 计算相对于训练开始时间的窗口
+	return map[string]TimeRange{
+		"2min": {Start: startTime + 120000, End: startTime + 240000}, // 第2分钟（120-240秒）
+		"4min": {Start: startTime + 240000, End: startTime + 360000},
+		"6min": {Start: startTime + 360000, End: startTime + 480000},
+	}
 }
--- a/go.mod
+++ b/go.mod
@ -4,9 +4,9 @@ go 1.23.3

 require (
 	github.com/gin-gonic/gin v1.10.0
-	github.com/golang-jwt/jwt/v4 v4.5.1
 	github.com/golang-jwt/jwt/v5 v5.2.1
 	github.com/spf13/viper v1.20.0
+	gonum.org/v1/gonum v0.16.0
 	gorm.io/driver/postgres v1.5.11
 	gorm.io/gorm v1.25.12
 )
@ -51,9 +51,9 @@ require (
 	golang.org/x/arch v0.8.0 // indirect
 	golang.org/x/crypto v0.32.0 // indirect
 	golang.org/x/net v0.33.0 // indirect
-	golang.org/x/sync v0.10.0 // indirect
+	golang.org/x/sync v0.12.0 // indirect
 	golang.org/x/sys v0.29.0 // indirect
-	golang.org/x/text v0.21.0 // indirect
+	golang.org/x/text v0.23.0 // indirect
 	google.golang.org/protobuf v1.36.1 // indirect
 	gopkg.in/yaml.v3 v3.0.1 // indirect
 )
--- a/go.sum
+++ b/go.sum
@ -31,8 +31,6 @@ github.com/go-viper/mapstructure/v2 v2.2.1 h1:ZAaOCxANMuZx5RCeg0mBdEZk7DZasvvZIx
 github.com/go-viper/mapstructure/v2 v2.2.1/go.mod h1:oJDH3BJKyqBA2TXFhDsKDGDTlndYOZ6rGS0BRZIxGhM=
 github.com/goccy/go-json v0.10.2 h1:CrxCmQqYDkv1z7lO7Wbh2HN93uovUHgrECaO5ZrCXAU=
 github.com/goccy/go-json v0.10.2/go.mod h1:6MelG93GURQebXPDq3khkgXZkazVtN9CRI+MGFi0w8I=
-github.com/golang-jwt/jwt/v4 v4.5.1 h1:JdqV9zKUdtaa9gdPlywC3aeoEsR681PlKC+4F5gQgeo=
-github.com/golang-jwt/jwt/v4 v4.5.1/go.mod h1:m21LjoU+eqJr34lmDMbreY2eSTRJ1cv77w39/MY0Ch0=
 github.com/golang-jwt/jwt/v5 v5.2.1 h1:OuVbFODueb089Lh128TAcimifWaLhJwVflnrgM17wHk=
 github.com/golang-jwt/jwt/v5 v5.2.1/go.mod h1:pqrtFR0X4osieyHYxtmOUWsAWrfe1Q5UVIyoH402zdk=
 github.com/google/go-cmp v0.6.0 h1:ofyhxvXcZhMsU5ulbFiLKl/XBFqE1GSq7atu8tAmTRI=
@ -114,14 +112,16 @@ golang.org/x/crypto v0.32.0 h1:euUpcYgM8WcP71gNpTqQCn6rC2t6ULUPiOzfWaXVVfc=
 golang.org/x/crypto v0.32.0/go.mod h1:ZnnJkOaASj8g0AjIduWNlq2NRxL0PlBrbKVyZ6V/Ugc=
 golang.org/x/net v0.33.0 h1:74SYHlV8BIgHIFC/LrYkOGIwL19eTYXQ5wc6TBuO36I=
 golang.org/x/net v0.33.0/go.mod h1:HXLR5J+9DxmrqMwG9qjGCxZ+zKXxBru04zlTvWlWuN4=
-golang.org/x/sync v0.10.0 h1:3NQrjDixjgGwUOCaF8w2+VYHv0Ve/vGYSbdkTa98gmQ=
-golang.org/x/sync v0.10.0/go.mod h1:Czt+wKu1gCyEFDUtn0jG5QVvpJ6rzVqr5aXyt9drQfk=
+golang.org/x/sync v0.12.0 h1:MHc5BpPuC30uJk597Ri8TV3CNZcTLu6B6z4lJy+g6Jw=
+golang.org/x/sync v0.12.0/go.mod h1:1dzgHSNfp02xaA81J2MS99Qcpr2w7fw1gpm99rleRqA=
 golang.org/x/sys v0.5.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
 golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
 golang.org/x/sys v0.29.0 h1:TPYlXGxvx1MGTn2GiZDhnjPA9wZzZeGKHHmKhHYvgaU=
 golang.org/x/sys v0.29.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA=
-golang.org/x/text v0.21.0 h1:zyQAAkrwaneQ066sspRyJaG9VNi/YJ1NfzcGB3hZ/qo=
-golang.org/x/text v0.21.0/go.mod h1:4IBbMaMmOPCJ8SecivzSH54+73PCFmPWxNTLm+vZkEQ=
+golang.org/x/text v0.23.0 h1:D71I7dUrlY+VX0gQShAThNGHFxZ13dGLBHQLVl1mJlY=
+golang.org/x/text v0.23.0/go.mod h1:/BLNzu4aZCJ1+kcD0DNRotWKage4q2rGVAg4o22unh4=
+gonum.org/v1/gonum v0.16.0 h1:5+ul4Swaf3ESvrOnidPp4GZbzf0mxVQpDCYUQE7OJfk=
+gonum.org/v1/gonum v0.16.0/go.mod h1:fef3am4MQ93R2HHpKnLk4/Tbh/s0+wqD5nfa6Pnwy4E=
 google.golang.org/protobuf v1.36.1 h1:yBPeRvTftaleIgM3PZ/WBIZ7XM/eEYAaEyCwvyjq/gk=
 google.golang.org/protobuf v1.36.1/go.mod h1:9fA7Ob0pmnwhb644+1+CVWFRbNajQ6iRojtC/QF5bRE=
 gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
--- a/main.go
+++ b/main.go
@ -15,7 +15,7 @@ func main() {

 	config.DB.Debug()
 	// 自动迁移模型
-	config.DB.AutoMigrate(&models.TrainRecord{}, &models.TrainingData{}, &models.Belt{}, &models.HeartRate{})
+	config.DB.AutoMigrate(&models.TrainRecord{}, &models.TrainingData{}, &models.Belt{}, &models.HeartRate{}, &models.BeltAnalysis{})

 	// 启动服务
 	r := routes.SetupRouter()
--- a/models/training.go
+++ b/models/training.go
@ -18,6 +18,20 @@ type Belt struct {
 	Name string `gorm:"size:100" json:"name"`
 }

+// 分析结果存储实体
+type BeltAnalysis struct {
+	gorm.Model
+	TrainID     uint    `gorm:"index;not null"` // 关联训练记录
+	BeltID      uint    `gorm:"index;not null"` // 腰带唯一标识
+	RunType     string  `gorm:"size:100" json:"RunType"`
+	Avg2min     float64 `gorm:"type:double precision"` // 第2分钟平均心率
+	Avg4min     float64 `gorm:"type:double precision"` // 第4分钟平均心率
+	Avg6min     float64 `gorm:"type:double precision"` // 第6分钟平均心率
+	CurveParamA float64 `gorm:"type:double precision"` // 拟合参数a值
+}
+
+// 中间计算结构（无需持久化）
+
 // 对应Flutter的HeartRate结构
 type HeartRate struct {
 	gorm.Model
@ -37,6 +51,7 @@ type TrainRecord struct {
 	StartTime    int64       `gorm:"type:bigint" json:"time"`    // 开始时间戳
 	EndTime      int64       `gorm:"type:bigint" json:"endTime"` // 结束时间戳[3](@ref)
 	Name         string      `gorm:"size:100" json:"name"`
+	RunType      string      `gorm:"size:100" json:"RunType"`
 	MaxHeartRate int         `gorm:"type:int" json:"maxHeartRate"`
 	Duration     int         `gorm:"type:int" json:"duration"` // 持续时间(秒)
 	PeopleNum    int         `gorm:"type:int" json:"peopleNum"`
--- a/routes/routes.go
+++ b/routes/routes.go
@ -15,9 +15,10 @@ func SetupRouter() *gin.Engine {

 	v1 := r.Group("/api/v1")
 	{
-		records := v1.Group("/train-records").Use(middleware.AuthMiddleware())
+		records := v1.Group("/train-records") //.Use(middleware.AuthMiddleware())
 		{
 			records.POST("", trainingController.CreateTrainingRecord)
+			records.GET("/analysis", trainingController.HandleCurveAnalysis)
 			// 可扩展其他路由：GET, PUT, DELETE等
 		}
 		auth := v1.Group("/auth")
Author	SHA1	Message	Date
laoboli	d69aaa5704	refactor: config.sample.yaml.	2025-04-07 09:48:49 +08:00
laoboli	6e8232ff7f	refactor: data analyze result.	2025-04-01 15:35:24 +08:00
laoboli	7b9e870bf9	refactor: data analyze.	2025-04-01 10:13:38 +08:00