From d746bc39354841c38d1bc42e4aa7700c10687c1a Mon Sep 17 00:00:00 2001 From: laoboli <1293528695@qq.com> Date: Wed, 19 Mar 2025 14:55:12 +0800 Subject: [PATCH] struct --- .gitignore | 3 + config.yaml | 6 + config/config.go | 45 ++++ controllers/train.go | 31 +++ go.mod | 58 +++++ go.sum | 136 ++++++++++++ hr_receiver.iml | 9 + main.go | 22 ++ main.go.bak | 516 +++++++++++++++++++++++++++++++++++++++++++ models/training.go | 11 + routes/routes.go | 17 ++ 11 files changed, 854 insertions(+) create mode 100644 .gitignore create mode 100644 config.yaml create mode 100644 config/config.go create mode 100644 controllers/train.go create mode 100644 go.mod create mode 100644 go.sum create mode 100644 hr_receiver.iml create mode 100644 main.go create mode 100644 main.go.bak create mode 100644 models/training.go create mode 100644 routes/routes.go diff --git a/.gitignore b/.gitignore new file mode 100644 index 0000000..cac686b --- /dev/null +++ b/.gitignore @@ -0,0 +1,3 @@ +.idea +hr_receiver.iml +main.go.bak diff --git a/config.yaml b/config.yaml new file mode 100644 index 0000000..e744097 --- /dev/null +++ b/config.yaml @@ -0,0 +1,6 @@ +database: + host: localhost + port: 5432 + user: postgres + password: root + name: training_db diff --git a/config/config.go b/config/config.go new file mode 100644 index 0000000..3f6fb33 --- /dev/null +++ b/config/config.go @@ -0,0 +1,45 @@ +package config + +import ( + "gorm.io/driver/postgres" +) +import ( + "github.com/spf13/viper" + "gorm.io/gorm" +) + +var DB *gorm.DB + +type AppConfig struct { + DB struct { + Host string `yaml:"host"` + Port string `yaml:"port"` + User string `yaml:"user"` + Password string `yaml:"password"` + Name string `yaml:"name"` + } `yaml:"database"` +} + +func InitConfig() { + viper.AddConfigPath("./") + viper.SetConfigName("config") + viper.SetConfigType("yaml") + if err := viper.ReadInConfig(); err != nil { + panic("Failed to read config: " + err.Error()) + } +} + +func ConnectDB() { + dsn := "host=" + viper.GetString("database.host") + + " user=" + viper.GetString("database.user") + + " password=" + viper.GetString("database.password") + + " dbname=" + viper.GetString("database.name") + + " port=" + viper.GetString("database.port") + + " sslmode=disable" + + var err error + DB, err = gorm.Open(postgres.Open(dsn), &gorm.Config{}) + if err != nil { + panic("Failed to connect database: " + err.Error()) + } +} diff --git a/controllers/train.go b/controllers/train.go new file mode 100644 index 0000000..f402094 --- /dev/null +++ b/controllers/train.go @@ -0,0 +1,31 @@ +package controllers + +import ( + "github.com/gin-gonic/gin" + "hr_receiver/config" + "hr_receiver/models" + "net/http" +) + +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, + }) +} diff --git a/go.mod b/go.mod new file mode 100644 index 0000000..f59f430 --- /dev/null +++ b/go.mod @@ -0,0 +1,58 @@ +module hr_receiver + +go 1.23.3 + +require ( + github.com/gin-gonic/gin v1.10.0 + github.com/lib/pq v1.10.9 + github.com/spf13/viper v1.20.0 + gorm.io/driver/postgres v1.5.11 + gorm.io/gorm v1.25.12 +) + +require ( + github.com/bytedance/sonic v1.11.6 // indirect + github.com/bytedance/sonic/loader v0.1.1 // indirect + 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/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 + github.com/go-playground/universal-translator v0.18.1 // indirect + github.com/go-playground/validator/v10 v10.20.0 // indirect + github.com/go-viper/mapstructure/v2 v2.2.1 // indirect + github.com/goccy/go-json v0.10.2 // indirect + github.com/jackc/pgpassfile v1.0.0 // indirect + github.com/jackc/pgservicefile v0.0.0-20221227161230-091c0ba34f0a // indirect + github.com/jackc/pgx/v5 v5.5.5 // indirect + github.com/jackc/puddle/v2 v2.2.1 // indirect + github.com/jinzhu/inflection v1.0.0 // indirect + github.com/jinzhu/now v1.1.5 // indirect + github.com/json-iterator/go v1.1.12 // indirect + github.com/klauspost/cpuid/v2 v2.2.7 // indirect + github.com/leodido/go-urn v1.4.0 // indirect + github.com/mattn/go-isatty v0.0.20 // indirect + github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd // indirect + github.com/modern-go/reflect2 v1.0.2 // indirect + github.com/pelletier/go-toml/v2 v2.2.3 // indirect + github.com/rogpeppe/go-internal v1.13.1 // indirect + github.com/sagikazarmark/locafero v0.7.0 // indirect + github.com/sourcegraph/conc v0.3.0 // indirect + github.com/spf13/afero v1.12.0 // indirect + github.com/spf13/cast v1.7.1 // indirect + github.com/spf13/pflag v1.0.6 // indirect + github.com/subosito/gotenv v1.6.0 // indirect + github.com/twitchyliquid64/golang-asm v0.15.1 // indirect + github.com/ugorji/go/codec v1.2.12 // indirect + go.uber.org/atomic v1.9.0 // indirect + go.uber.org/multierr v1.9.0 // indirect + 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/sys v0.29.0 // indirect + golang.org/x/text v0.21.0 // indirect + google.golang.org/protobuf v1.36.1 // indirect + gopkg.in/yaml.v3 v3.0.1 // indirect +) diff --git a/go.sum b/go.sum new file mode 100644 index 0000000..1cd3b79 --- /dev/null +++ b/go.sum @@ -0,0 +1,136 @@ +github.com/bytedance/sonic v1.11.6 h1:oUp34TzMlL+OY1OUWxHqsdkgC/Zfc85zGqw9siXjrc0= +github.com/bytedance/sonic v1.11.6/go.mod h1:LysEHSvpvDySVdC2f87zGWf6CIKJcAvqab1ZaiQtds4= +github.com/bytedance/sonic/loader v0.1.1 h1:c+e5Pt1k/cy5wMveRDyk2X4B9hF4g7an8N3zCYjJFNM= +github.com/bytedance/sonic/loader v0.1.1/go.mod h1:ncP89zfokxS5LZrJxl5z0UJcsk4M4yY2JpfqGeCtNLU= +github.com/cloudwego/base64x v0.1.4 h1:jwCgWpFanWmN8xoIUHa2rtzmkd5J2plF/dnLS6Xd/0Y= +github.com/cloudwego/base64x v0.1.4/go.mod h1:0zlkT4Wn5C6NdauXdJRhSKRlJvmclQ1hhJgA0rcu/8w= +github.com/cloudwego/iasm v0.2.0 h1:1KNIy1I1H9hNNFEEH3DVnI4UujN+1zjpuk6gwHLTssg= +github.com/cloudwego/iasm v0.2.0/go.mod h1:8rXZaNYT2n95jn+zTI1sDr+IgcD2GVs0nlbbQPiEFhY= +github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= +github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c= +github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= +github.com/frankban/quicktest v1.14.6 h1:7Xjx+VpznH+oBnejlPUj8oUpdxnVs4f8XU8WnHkI4W8= +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/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= +github.com/gin-contrib/sse v0.1.0/go.mod h1:RHrZQHXnP2xjPF+u1gW/2HnVO7nvIa9PG3Gm+fLHvGI= +github.com/gin-gonic/gin v1.10.0 h1:nTuyha1TYqgedzytsKYqna+DfLos46nTv2ygFy86HFU= +github.com/gin-gonic/gin v1.10.0/go.mod h1:4PMNQiOhvDRa013RKVbsiNwoyezlm2rm0uX/T7kzp5Y= +github.com/go-playground/assert/v2 v2.2.0 h1:JvknZsQTYeFEAhQwI4qEt9cyV5ONwRHC+lYKSsYSR8s= +github.com/go-playground/assert/v2 v2.2.0/go.mod h1:VDjEfimB/XKnb+ZQfWdccd7VUvScMdVu0Titje2rxJ4= +github.com/go-playground/locales v0.14.1 h1:EWaQ/wswjilfKLTECiXz7Rh+3BjFhfDFKv/oXslEjJA= +github.com/go-playground/locales v0.14.1/go.mod h1:hxrqLVvrK65+Rwrd5Fc6F2O76J/NuW9t0sjnWqG1slY= +github.com/go-playground/universal-translator v0.18.1 h1:Bcnm0ZwsGyWbCzImXv+pAJnYK9S473LQFuzCbDbfSFY= +github.com/go-playground/universal-translator v0.18.1/go.mod h1:xekY+UJKNuX9WP91TpwSH2VMlDf28Uj24BCp08ZFTUY= +github.com/go-playground/validator/v10 v10.20.0 h1:K9ISHbSaI0lyB2eWMPJo+kOS/FBExVwjEviJTixqxL8= +github.com/go-playground/validator/v10 v10.20.0/go.mod h1:dbuPbCMFw/DrkbEynArYaCwl3amGuJotoKCe95atGMM= +github.com/go-viper/mapstructure/v2 v2.2.1 h1:ZAaOCxANMuZx5RCeg0mBdEZk7DZasvvZIxtHqx8aGss= +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/google/go-cmp v0.6.0 h1:ofyhxvXcZhMsU5ulbFiLKl/XBFqE1GSq7atu8tAmTRI= +github.com/google/go-cmp v0.6.0/go.mod h1:17dUlkBOakJ0+DkrSSNjCkIjxS6bF9zb3elmeNGIjoY= +github.com/google/gofuzz v1.0.0/go.mod h1:dBl0BpW6vV/+mYPU4Po3pmUjxk6FQPldtuIdl/M65Eg= +github.com/jackc/pgpassfile v1.0.0 h1:/6Hmqy13Ss2zCq62VdNG8tM1wchn8zjSGOBJ6icpsIM= +github.com/jackc/pgpassfile v1.0.0/go.mod h1:CEx0iS5ambNFdcRtxPj5JhEz+xB6uRky5eyVu/W2HEg= +github.com/jackc/pgservicefile v0.0.0-20221227161230-091c0ba34f0a h1:bbPeKD0xmW/Y25WS6cokEszi5g+S0QxI/d45PkRi7Nk= +github.com/jackc/pgservicefile v0.0.0-20221227161230-091c0ba34f0a/go.mod h1:5TJZWKEWniPve33vlWYSoGYefn3gLQRzjfDlhSJ9ZKM= +github.com/jackc/pgx/v5 v5.5.5 h1:amBjrZVmksIdNjxGW/IiIMzxMKZFelXbUoPNb+8sjQw= +github.com/jackc/pgx/v5 v5.5.5/go.mod h1:ez9gk+OAat140fv9ErkZDYFWmXLfV+++K0uAOiwgm1A= +github.com/jackc/puddle/v2 v2.2.1 h1:RhxXJtFG022u4ibrCSMSiu5aOq1i77R3OHKNJj77OAk= +github.com/jackc/puddle/v2 v2.2.1/go.mod h1:vriiEXHvEE654aYKXXjOvZM39qJ0q+azkZFrfEOc3H4= +github.com/jinzhu/inflection v1.0.0 h1:K317FqzuhWc8YvSVlFMCCUb36O/S9MCKRDI7QkRKD/E= +github.com/jinzhu/inflection v1.0.0/go.mod h1:h+uFLlag+Qp1Va5pdKtLDYj+kHp5pxUVkryuEj+Srlc= +github.com/jinzhu/now v1.1.5 h1:/o9tlHleP7gOFmsnYNz3RGnqzefHA47wQpKrrdTIwXQ= +github.com/jinzhu/now v1.1.5/go.mod h1:d3SSVoowX0Lcu0IBviAWJpolVfI5UJVZZ7cO71lE/z8= +github.com/json-iterator/go v1.1.12 h1:PV8peI4a0ysnczrg+LtxykD8LfKY9ML6u2jnxaEnrnM= +github.com/json-iterator/go v1.1.12/go.mod h1:e30LSqwooZae/UwlEbR2852Gd8hjQvJoHmT4TnhNGBo= +github.com/klauspost/cpuid/v2 v2.0.9/go.mod h1:FInQzS24/EEf25PyTYn52gqo7WaD8xa0213Md/qVLRg= +github.com/klauspost/cpuid/v2 v2.2.7 h1:ZWSB3igEs+d0qvnxR/ZBzXVmxkgt8DdzP6m9pfuVLDM= +github.com/klauspost/cpuid/v2 v2.2.7/go.mod h1:Lcz8mBdAVJIBVzewtcLocK12l3Y+JytZYpaMropDUws= +github.com/knz/go-libedit v1.10.1/go.mod h1:MZTVkCWyz0oBc7JOWP3wNAzd002ZbM/5hgShxwh4x8M= +github.com/kr/pretty v0.3.1 h1:flRD4NNwYAUpkphVc1HcthR4KEIFJ65n8Mw5qdRn3LE= +github.com/kr/pretty v0.3.1/go.mod h1:hoEshYVHaxMs3cyo3Yncou5ZscifuDolrwPKZanG3xk= +github.com/kr/text v0.2.0 h1:5Nx0Ya0ZqY2ygV366QzturHI13Jq95ApcVaJBhpS+AY= +github.com/kr/text v0.2.0/go.mod h1:eLer722TekiGuMkidMxC/pM04lWEeraHUUmBw8l2grE= +github.com/leodido/go-urn v1.4.0 h1:WT9HwE9SGECu3lg4d/dIA+jxlljEa1/ffXKmRjqdmIQ= +github.com/leodido/go-urn v1.4.0/go.mod h1:bvxc+MVxLKB4z00jd1z+Dvzr47oO32F/QSNjSBOlFxI= +github.com/lib/pq v1.10.9 h1:YXG7RB+JIjhP29X+OtkiDnYaXQwpS4JEWq7dtCCRUEw= +github.com/lib/pq v1.10.9/go.mod h1:AlVN5x4E4T544tWzH6hKfbfQvm3HdbOxrmggDNAPY9o= +github.com/mattn/go-isatty v0.0.20 h1:xfD0iDuEKnDkl03q4limB+vH+GxLEtL/jb4xVJSWWEY= +github.com/mattn/go-isatty v0.0.20/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y= +github.com/modern-go/concurrent v0.0.0-20180228061459-e0a39a4cb421/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q= +github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd h1:TRLaZ9cD/w8PVh93nsPXa1VrQ6jlwL5oN8l14QlcNfg= +github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q= +github.com/modern-go/reflect2 v1.0.2 h1:xBagoLtFs94CBntxluKeaWgTMpvLxC4ur3nMaC9Gz0M= +github.com/modern-go/reflect2 v1.0.2/go.mod h1:yWuevngMOJpCy52FWWMvUC8ws7m/LJsjYzDa0/r8luk= +github.com/pelletier/go-toml/v2 v2.2.3 h1:YmeHyLY8mFWbdkNWwpr+qIL2bEqT0o95WSdkNHvL12M= +github.com/pelletier/go-toml/v2 v2.2.3/go.mod h1:MfCQTFTvCcUyyvvwm1+G6H/jORL20Xlb6rzQu9GuUkc= +github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM= +github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4= +github.com/rogpeppe/go-internal v1.13.1 h1:KvO1DLK/DRN07sQ1LQKScxyZJuNnedQ5/wKSR38lUII= +github.com/rogpeppe/go-internal v1.13.1/go.mod h1:uMEvuHeurkdAXX61udpOXGD/AzZDWNMNyH2VO9fmH0o= +github.com/sagikazarmark/locafero v0.7.0 h1:5MqpDsTGNDhY8sGp0Aowyf0qKsPrhewaLSsFaodPcyo= +github.com/sagikazarmark/locafero v0.7.0/go.mod h1:2za3Cg5rMaTMoG/2Ulr9AwtFaIppKXTRYnozin4aB5k= +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= +github.com/spf13/afero v1.12.0/go.mod h1:ZTlWwG4/ahT8W7T0WQ5uYmjI9duaLQGy3Q2OAl4sk/4= +github.com/spf13/cast v1.7.1 h1:cuNEagBQEHWN1FnbGEjCXL2szYEXqfJPbP2HNUaca9Y= +github.com/spf13/cast v1.7.1/go.mod h1:ancEpBxwJDODSW/UG4rDrAqiKolqNNh2DX3mk86cAdo= +github.com/spf13/pflag v1.0.6 h1:jFzHGLGAlb3ruxLB8MhbI6A8+AQX/2eW4qeyNZXNp2o= +github.com/spf13/pflag v1.0.6/go.mod h1:McXfInJRrz4CZXVZOBLb0bTZqETkiAhM9Iw0y3An2Bg= +github.com/spf13/viper v1.20.0 h1:zrxIyR3RQIOsarIrgL8+sAvALXul9jeEPa06Y0Ph6vY= +github.com/spf13/viper v1.20.0/go.mod h1:P9Mdzt1zoHIG8m2eZQinpiBjo6kCmZSKBClNNqjJvu4= +github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME= +github.com/stretchr/objx v0.4.0/go.mod h1:YvHI0jy2hoMjB+UWwv71VJQ9isScKT/TqJzVSSt89Yw= +github.com/stretchr/objx v0.5.0/go.mod h1:Yh+to48EsGEfYuaHDzXPcE3xhTkx73EhmCGUpEOglKo= +github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI= +github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg= +github.com/stretchr/testify v1.7.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg= +github.com/stretchr/testify v1.8.0/go.mod h1:yNjHg4UonilssWZ8iaSj1OCr/vHnekPRkoO+kdMU+MU= +github.com/stretchr/testify v1.8.1/go.mod h1:w2LPCIKwWwSfY2zedu0+kehJoqGctiVI29o6fzry7u4= +github.com/stretchr/testify v1.10.0 h1:Xv5erBjTwe/5IxqUQTdXv5kgmIvbHo3QQyRwhJsOfJA= +github.com/stretchr/testify v1.10.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY= +github.com/subosito/gotenv v1.6.0 h1:9NlTDc1FTs4qu0DDq7AEtTPNw6SVm7uBMsUCUjABIf8= +github.com/subosito/gotenv v1.6.0/go.mod h1:Dk4QP5c2W3ibzajGcXpNraDfq2IrhjMIvMSWPKKo0FU= +github.com/twitchyliquid64/golang-asm v0.15.1 h1:SU5vSMR7hnwNxj24w34ZyCi/FmDZTkS4MhqMhdFk5YI= +github.com/twitchyliquid64/golang-asm v0.15.1/go.mod h1:a1lVb/DtPvCB8fslRZhAngC2+aY1QWCk3Cedj/Gdt08= +github.com/ugorji/go/codec v1.2.12 h1:9LC83zGrHhuUA9l16C9AHXAqEV/2wBQ4nkvumAE65EE= +github.com/ugorji/go/codec v1.2.12/go.mod h1:UNopzCgEMSXjBc6AOMqYvWC1ktqTAfzJZUZgYf6w6lg= +go.uber.org/atomic v1.9.0 h1:ECmE8Bn/WFTYwEW/bpKD3M8VtR/zQVbavAoalC1PYyE= +go.uber.org/atomic v1.9.0/go.mod h1:fEN4uk6kAWBTFdckzkM89CLk9XfWZrxpCo0nPH17wJc= +go.uber.org/multierr v1.9.0 h1:7fIwc/ZtS0q++VgcfqFDxSBZVv/Xo49/SYnDFupUwlI= +go.uber.org/multierr v1.9.0/go.mod h1:X2jQV1h+kxSjClGpnseKVIxpmcjrj7MNnI0bnlfKTVQ= +golang.org/x/arch v0.0.0-20210923205945-b76863e36670/go.mod h1:5om86z9Hs0C8fWVUuoMHwpExlXzs5Tkyp9hOrfG7pp8= +golang.org/x/arch v0.8.0 h1:3wRIsP3pM4yUptoR96otTUOXI367OS0+c9eeRi9doIc= +golang.org/x/arch v0.8.0/go.mod h1:FEVrYAQjsQXMVJ1nsMoVVXPZg6p2JE2mx8psSWTDQys= +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/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= +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= +gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c h1:Hei/4ADfdWqJk1ZMxUNpqntNwaWcugrBjAiHlqqRiVk= +gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c/go.mod h1:JHkPIbrfpd72SG/EVd6muEfDQjcINNoR0C8j2r3qZ4Q= +gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM= +gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA= +gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM= +gorm.io/driver/postgres v1.5.11 h1:ubBVAfbKEUld/twyKZ0IYn9rSQh448EdelLYk9Mv314= +gorm.io/driver/postgres v1.5.11/go.mod h1:DX3GReXH+3FPWGrrgffdvCk3DQ1dwDPdmbenSkweRGI= +gorm.io/gorm v1.25.12 h1:I0u8i2hWQItBq1WfE0o2+WuL9+8L21K9e2HHSTE/0f8= +gorm.io/gorm v1.25.12/go.mod h1:xh7N7RHfYlNc5EmcI/El95gXusucDrQnHXe0+CgWcLQ= +nullprogram.com/x/optparse v1.0.0/go.mod h1:KdyPE+Igbe0jQUrVfMqDMeJQIJZEuyV7pjYmp6pbG50= +rsc.io/pdf v0.1.1/go.mod h1:n8OzWcQ6Sp37PL01nO98y4iUCRdTGarVfzxY20ICaU4= diff --git a/hr_receiver.iml b/hr_receiver.iml new file mode 100644 index 0000000..eacc75a --- /dev/null +++ b/hr_receiver.iml @@ -0,0 +1,9 @@ + + + + + + + + + \ No newline at end of file diff --git a/main.go b/main.go new file mode 100644 index 0000000..98e4fd3 --- /dev/null +++ b/main.go @@ -0,0 +1,22 @@ +package main + +import ( + "hr_receiver/config" + "hr_receiver/models" + "hr_receiver/routes" +) + +func main() { + // 初始化配置 + config.InitConfig() + + // 连接数据库 + config.ConnectDB() + + // 自动迁移模型 + config.DB.AutoMigrate(&models.TrainingData{}) + + // 启动服务 + r := routes.SetupRouter() + r.Run(":8080") +} diff --git a/main.go.bak b/main.go.bak new file mode 100644 index 0000000..56057e9 --- /dev/null +++ b/main.go.bak @@ -0,0 +1,516 @@ +// main.go +package main + +import ( + "errors" + "fmt" + _ "fmt" + "github.com/gin-gonic/gin" + "github.com/lib/pq" + "gorm.io/driver/postgres" + "gorm.io/gorm/clause" + "log" + "net/http" + "net/url" + "regexp" + "sort" + "time" + + "gorm.io/gorm" +) + +// 调整后的结构体定义 +type CreateTrainingRequest struct { + TID uint `json:"tid" binding:"required"` + StartTime int64 `json:"startTime" binding:"required,min=1609459200000"` // 2021-01-01起 + EndTime int64 `json:"endTime" binding:"required,min=1609459200000"` + Name string `json:"name" binding:"required"` + MaxHeartRate int `json:"maxHeartRate" binding:"required,min=30,max=250"` + Duration int `json:"duration" binding:"required"` // 单位：秒 + PeopleNum int `json:"peopleNum" binding:"required"` + Evaluation string `json:"evaluation" binding:"required"` + BeltAddrs []string `json:"beltAddrs" binding:"required"` +} + +type HeartRateUploadRequest struct { + TrainID uint `json:"trainId" binding:"required"` + Data []HeartRateData `json:"data" binding:"required,dive"` +} + +type HeartRateData struct { + BeltAddr string `json:"beltAddr" binding:"required"` + Timestamp int64 `json:"timestamp" binding:"required,min=1609459200000"` + Value int `json:"value" binding:"required,min=30,max=250"` + LastValue int `json:"lastValue" binding:"required,min=30,max=250"` +} + +// 数据库模型 +type TrainingRecord struct { + TID uint `gorm:"primaryKey;column:tid" json:"tid"` + StartTime time.Time `gorm:"not null" json:"-"` + EndTime time.Time `gorm:"not null" json:"-"` + Name string `gorm:"type:varchar(255);default:'训练'" json:"name"` + MaxHeartRate int `gorm:"not null" json:"maxHeartRate"` + Duration int `gorm:"not null" json:"duration"` + PeopleNum int `gorm:"not null" json:"peopleNum"` + Evaluation string `gorm:"type:varchar(255);default:'适中'" json:"evaluation"` + BeltAddresses []string `gorm:"type:text[]" json:"beltAddrs"` + + // 添加毫秒时间戳字段（仅用于JSON序列化） + StartTimestamp int64 `gorm:"-" json:"startTime"` + EndTimestamp int64 `gorm:"-" json:"endTime"` +} + +type HeartRate struct { + Time time.Time `gorm:"primaryKey;type:timestamptz" json:"time"` + TrainID uint `gorm:"primaryKey;index" json:"train_id"` + BeltAddr string `gorm:"type:varchar(255);not null;index" json:"belt_addr"` + Value int `gorm:"check:value BETWEEN 30 AND 250" json:"value"` + LastValue int `gorm:"check:last_value BETWEEN 30 AND 250" json:"last_value"` +} + +// 统计分析响应结构 +type TrainingReport struct { + TrainID uint `json:"train_id"` + AvgHeartRate float64 `json:"avg_heart_rate"` + MaxHeartRate int `json:"max_heart_rate"` + DangerSeconds int `json:"danger_seconds"` + BeltStats map[string]Stats `json:"belt_stats"` + TimeSeries []TimePoint `json:"time_series,omitempty"` +} + +type Stats struct { + Avg float64 `json:"avg"` + Max int `json:"max"` + DangerCount int `json:"danger_count"` +} + +type TimePoint struct { + Time time.Time `json:"time"` + AvgValue float64 `json:"avg_value"` + DangerCount int `json:"danger_count"` +} + +// 实现自定义序列化逻辑 +func (t *TrainingRecord) AfterFind(tx *gorm.DB) (err error) { + t.StartTimestamp = t.StartTime.UnixNano() / int64(time.Millisecond) + t.EndTimestamp = t.EndTime.UnixNano() / int64(time.Millisecond) + return +} + +func (t *TrainingRecord) BeforeCreate(tx *gorm.DB) (err error) { + t.StartTime = time.Unix(0, t.StartTimestamp*int64(time.Millisecond)).UTC() + t.EndTime = time.Unix(0, t.EndTimestamp*int64(time.Millisecond)).UTC() + return +} + +// 更新处理函数 +//func createTraining(c *gin.Context) { +// var req CreateTrainingRequest +// if err := c.ShouldBindJSON(&req); err != nil { +// c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()}) +// return +// } +// +// record := TrainingRecord{ +// TID: req.TID, +// StartTimestamp: req.StartTime, +// EndTimestamp: req.EndTime, +// Name: req.Name, +// MaxHeartRate: req.MaxHeartRate, +// Duration: req.Duration, +// PeopleNum: req.PeopleNum, +// Evaluation: req.Evaluation, +// BeltAddresses: req.BeltAddrs, +// } +// +// if err := db.Create(&record).Error; err != nil { +// if isDuplicateKeyError(err) { +// c.JSON(http.StatusConflict, gin.H{"error": "训练记录已存在"}) +// return +// } +// c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()}) +// return +// } +// +// c.JSON(http.StatusCreated, record) +//} + +// 统一时间处理函数 +func parseTimestamp(ms int64) (time.Time, error) { + if ms < 1609459200000 { // 2021-01-01 00:00:00 UTC + return time.Time{}, fmt.Errorf("无效的时间戳") + } + return time.Unix(0, ms*int64(time.Millisecond)).UTC(), nil +} + +// 更新心率数据处理 +func processHeartRateData(req *HeartRateUploadRequest) ([]HeartRate, error) { + var rates []HeartRate + + for _, d := range req.Data { + t, err := parseTimestamp(d.Timestamp) + if err != nil { + return nil, fmt.Errorf("无效的时间戳: %d", d.Timestamp) + } + + rates = append(rates, HeartRate{ + TrainID: req.TrainID, + BeltAddr: d.BeltAddr, + Time: t, + Value: d.Value, + LastValue: d.LastValue, + }) + } + + // 按时间排序 + sort.Slice(rates, func(i, j int) bool { + return rates[i].Time.Before(rates[j].Time) + }) + + return rates, nil +} + +// 更新自动迁移逻辑 +func autoMigrate(db *gorm.DB) { + db.Set("gorm:table_options", " comment '训练记录表'").AutoMigrate(&TrainingRecord{}) + db.Set("gorm:table_options", " comment '心率数据表'").AutoMigrate(&HeartRate{}) + + // 创建复合索引 + db.Exec(` + CREATE INDEX IF NOT EXISTS idx_heart_rates_main + ON heart_rates (train_id, belt_addr, time DESC) + `) +} + +var db *gorm.DB + +const ( + defaultDB = "postgres" // 用于创建新数据库的默认数据库 +) + +func initDB() { + // 解析原始DSN + dsn := "host=localhost user=postgres password=root dbname=training port=5432 sslmode=disable" + parsedDSN, err := url.Parse(dsn) + if err != nil { + log.Fatal("Invalid DSN:", err) + } + query := parsedDSN.Query() + query.Set("dbname", defaultDB) + query.Set("sslmode", "disable") + query.Set("port", "5432") + query.Set("host", "localhost") + query.Set("user", "postgres") + query.Set("password", "root") + defaultDSN := fmt.Sprintf("host=%s user=%s password=%s dbname=%s port=%s sslmode=%s", + query.Get("host"), + query.Get("user"), + query.Get("password"), + query.Get("dbname"), + query.Get("port"), + query.Get("sslmode"), + ) + //defaultDSNUrl, _ := url.Parse(defaultDSN) + + // 提取数据库名称 + dbName := "training" + + // 第一步：尝试连接目标数据库 + db, err := gorm.Open(postgres.Open(dsn), &gorm.Config{}) + if err != nil { + // 检查是否是数据库不存在的错误（PostgreSQL错误码3D000） + if isDatabaseNotExistError(err) { + log.Printf("Database %q does not exist, attempting to create...", dbName) + createDatabase(defaultDSN, dbName) + } else { + log.Fatal("Database connection failed:", err) + } + } else { + log.Printf("Database %q already exists", dbName) + sqlDB, _ := db.DB() + sqlDB.Close() + } + + // 再次连接目标数据库 + db, err = gorm.Open(postgres.Open(dsn), &gorm.Config{ + CreateBatchSize: 1000, + }) + if err != nil { + log.Fatal("Final database connection failed:", err) + } + + // 自动迁移表结构 + autoMigrate(db) +} + +// 检查是否为数据库不存在错误 +func isDatabaseNotExistError(err error) bool { + if err == nil { + return false + } + errStr := err.Error() + return containsErrorCode(errStr, "3D000") +} + +func containsErrorCode(errStr, code string) bool { + // 使用正则表达式检查错误字符串中是否包含指定的错误码 + re := regexp.MustCompile(`\b` + code + `\b`) + return re.MatchString(errStr) +} + +// 创建新数据库 +func createDatabase(parsedDSN string, dbName string) { + // 使用默认数据库连接 + //parsedDSN.Path = "/" + defaultDB + defaultDSN := parsedDSN + + // 创建数据库 + db, err := gorm.Open(postgres.Open(defaultDSN), &gorm.Config{}) + if err != nil { + log.Fatal("Connect to default database failed:", err) + } + + // 需要超级用户权限才能创建数据库 + createSQL := fmt.Sprintf("CREATE DATABASE \"%s\"", dbName) + if err := db.Exec(createSQL).Error; err != nil { + log.Fatal("Create database failed:", err) + } + + sqlDB, _ := db.DB() + sqlDB.Close() + log.Printf("Database %q created successfully", dbName) +} + +// 自动迁移表结构 +//func autoMigrate(db *gorm.DB) { +// err := db.AutoMigrate( +// &TrainingRecord{}, +// &HeartRate{}, +// ) +// if err != nil { +// log.Fatal("Auto migrate failed:", err) +// } +// +// // 添加索引（生产环境建议使用迁移工具） +// db.Exec("CREATE INDEX IF NOT EXISTS idx_heart_rates_train_time ON heart_rates (train_id, time)") +// log.Println("Database schema initialized successfully") +//} + +// ...（保持之前的import和结构体定义不变） +// +// func initDB() { +// dsn := "host=localhost user=postgres password=root dbname=training port=5432 sslmode=disable" +// var err error +// db, err = gorm.Open(postgres.Open(dsn), &gorm.Config{ +// CreateBatchSize: 1000, +// }) +// if err != nil { +// log.Fatal("Failed to connect to database:", err) +// } +// +// // 自动迁移（生产环境建议使用迁移工具） +// db.AutoMigrate(&TrainingRecord{}, &HeartRate{}) +// } +func setupRouter() *gin.Engine { + r := gin.Default() + + // 训练记录路由组 + trainingGroup := r.Group("/api/trainings") + { + trainingGroup.POST("create", createTraining) + trainingGroup.POST("/:id/heart-rates", uploadHeartRates) + //trainingGroup.GET("/:id/report", getTrainingReport) + //trainingGroup.GET("/:id/time-series", getTimeSeriesData) + } + + return r +} + +// 更新处理函数 +func createTraining(c *gin.Context) { + var req CreateTrainingRequest + if err := c.ShouldBindJSON(&req); err != nil { + c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()}) + return + } + + record := TrainingRecord{ + TID: req.TID, + StartTimestamp: req.StartTime, + EndTimestamp: req.EndTime, + Name: req.Name, + MaxHeartRate: req.MaxHeartRate, + Duration: req.Duration, + PeopleNum: req.PeopleNum, + Evaluation: req.Evaluation, + BeltAddresses: req.BeltAddrs, + } + + if err := db.Create(&record).Error; err != nil { + if isDuplicateKeyError(err) { + c.JSON(http.StatusConflict, gin.H{"error": "训练记录已存在"}) + return + } + c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()}) + return + } + + c.JSON(http.StatusCreated, record) +} + +// 判断是否为唯一键冲突错误 +func isDuplicateKeyError(err error) bool { + if err == nil { + return false + } + + var pqErr *pq.Error + if errors.As(err, &pqErr) { + return pqErr.Code == "23505" // 唯一键冲突错误码 + } + + return false +} + +// 更新上传心率处理 +func uploadHeartRates(c *gin.Context) { + var req HeartRateUploadRequest + if err := c.ShouldBindJSON(&req); err != nil { + c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()}) + return + } + + // 时区处理（接受时区参数，默认为UTC） + loc := time.UTC + if tz := c.Query("tz"); tz != "" { + if l, err := time.LoadLocation(tz); err == nil { + loc = l + } + } + + heartRates := make([]HeartRate, 0, len(req.Data)) + for _, d := range req.Data { + t, err := parseTimestamp(d.Timestamp) + if err != nil { + c.JSON(http.StatusBadRequest, gin.H{ + "error": fmt.Sprintf("invalid timestamp: %v", d.Timestamp), + }) + return + } + + // 转换为指定时区 + t = t.In(loc) + + heartRates = append(heartRates, HeartRate{ + TrainID: req.TrainID, + BeltAddr: d.BeltAddr, + Time: t, + Value: d.Value, + LastValue: d.LastValue, + }) + } + + // 批量插入优化 + err := db.Clauses( + clause.OnConflict{ + Columns: []clause.Column{{Name: "train_id"}, {Name: "time"}}, + DoNothing: true, + }, + clause.Returning{}, + ).CreateInBatches(heartRates, 1000).Error + + if err != nil { + log.Printf("Batch insert error: %v", err) + c.JSON(http.StatusInternalServerError, gin.H{"error": "数据存储失败"}) + return + } + + c.JSON(http.StatusCreated, gin.H{ + "message": fmt.Sprintf("成功存储 %d 条心率数据", len(heartRates)), + "timezone": loc.String(), + }) +} + +// 获取训练报告 +func getTrainingReport(c *gin.Context) { + trainID := parseUint(c.Param("id")) + threshold := c.DefaultQuery("threshold", "120") // 默认危险阈值120 + + var report TrainingReport + report.BeltStats = make(map[string]Stats) + + // 获取基础统计信息 + baseQuery := db.Model(&HeartRate{}).Where("train_id = ?", trainID) + + // 整体平均和最大心率 + baseQuery.Select("AVG(value) as avg, MAX(value) as max"). + Row().Scan(&report.AvgHeartRate, &report.MaxHeartRate) + + // 危险时长计算（假设5秒一个数据点） + var dangerCount int64 + baseQuery.Where("value >= ?", threshold).Count(&dangerCount) + report.DangerSeconds = int(dangerCount) * 5 + + // 各腰带统计 + rows, err := db.Model(&HeartRate{}). + Select("belt_addr, AVG(value) as avg, MAX(value) as max, COUNT(*) filter (where value >= ?) as danger", threshold). + Where("train_id = ?", trainID). + Group("belt_addr"). + Rows() + + if err != nil { + c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()}) + return + } + defer rows.Close() + + for rows.Next() { + var addr string + var stat Stats + rows.Scan(&addr, &stat.Avg, &stat.Max, &stat.DangerCount) + report.BeltStats[addr] = stat + } + + report.TrainID = trainID + c.JSON(http.StatusOK, report) +} + +// 获取时间序列数据 +func getTimeSeriesData(c *gin.Context) { + trainID := parseUint(c.Param("id")) + interval := c.DefaultQuery("interval", "1m") // 默认1分钟间隔 + + var points []TimePoint + err := db.Raw(` + SELECT + date_trunc(?, time) as time, + AVG(value) as avg_value, + COUNT(*) FILTER (WHERE value >= 120) as danger_count + FROM heart_rates + WHERE train_id = ? + GROUP BY 1 + ORDER BY 1 + `, interval, trainID).Scan(&points).Error + + if err != nil { + c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()}) + return + } + + c.JSON(http.StatusOK, points) +} + +// 辅助函数：字符串转uint +func parseUint(s string) uint { + var n uint + fmt.Sscanf(s, "%d", &n) + return n +} + +// 补全main函数 +func main() { + initDB() + r := setupRouter() + log.Fatal(r.Run(":8081")) +} diff --git a/models/training.go b/models/training.go new file mode 100644 index 0000000..f9ed52c --- /dev/null +++ b/models/training.go @@ -0,0 +1,11 @@ +package models + +import "gorm.io/gorm" + +type TrainingData struct { + gorm.Model + Features string `gorm:"type:JSONB; not null" json:"features"` + Label string `gorm:"type:varchar(255)" json:"label"` + Probability float64 `gorm:"type:decimal(5,4)" json:"probability"` + SessionID string `gorm:"index" json:"session_id"` +} diff --git a/routes/routes.go b/routes/routes.go new file mode 100644 index 0000000..afab4f4 --- /dev/null +++ b/routes/routes.go @@ -0,0 +1,17 @@ +package routes + +import ( + "github.com/gin-gonic/gin" + "hr_receiver/controllers" +) + +func SetupRouter() *gin.Engine { + r := gin.Default() + + api := r.Group("/api/v1") + { + api.POST("/training", controllers.ReceiveTrainingData) + } + + return r +}