commit e568882c6f5ae365a883ed562f08b8eaba17cb72 Author: 皓月归尘 Date: Mon Nov 11 10:57:15 2024 +0800 feat: 初始化仓库 diff --git a/.gitignore b/.gitignore new file mode 100644 index 0000000..f92bea6 --- /dev/null +++ b/.gitignore @@ -0,0 +1,83 @@ +tmp +.env +gohub +.DS_Store +.history + +# Golang # +###################### +# `go test -c` 生成的二进制文件 +*.test +# go coverage 工具 +*.out +*.prof +*.cgo1.go +*.cgo2.c +_cgo_defun.c +_cgo_gotypes.go +_cgo_export.* + +# 编译文件 # +################### +*.com +*.class +*.dll +*.exe +*.exe~ +*.o +*.so + +# 压缩包 # +############ +# Git 自带压缩,如果这些压缩包里有代码,建议解压后 commit +*.7z +*.dmg +*.gz +*.iso +*.jar +*.rar +*.tar +*.zip + +# 日志文件和数据库 # +###################### +*.log +*.sqlite +*.db + +# 临时文件 # +###################### +tmp/ +.tmp/ + +# 系统生成文件 # +###################### +.DS_Store +.DS_Store? +.AppleDouble +.LSOverride +._* +.Spotlight-V100 +.Trashes +ehthumbs.db +Thumbs.db +.TemporaryItems +.fseventsd +.VolumeIcon.icns +.com.apple.timemachine.donotpresent + +# IDE 和编辑器 # +###################### +.idea/ +/go_build_* +out/ +.vscode/ +.vscode/settings.json +*.sublime* +__debug_bin +.project + +# 前端工具链 # +###################### +.sass-cache/* +node_modules/ \ No newline at end of file diff --git a/data/config.json b/data/config.json new file mode 100644 index 0000000..d00e257 --- /dev/null +++ b/data/config.json @@ -0,0 +1,81 @@ +{ + "host": "0.0.0.0", + "port": "8080", + "startDate": "2024-09-02", + "totalWeeks": 20, + "endDate": "2025-01-19", + "timeSetting": [ + { + "index": 1, + "start": "08:00", + "end": "08:45", + "time": 0 + }, + { + "index": 2, + "start": "08:50", + "end": "09:35", + "time": 0 + }, + { + "index": 3, + "start": "09:55", + "end": "10:40", + "time": 0 + }, + { + "index": 4, + "start": "10:45", + "end": "11:30", + "time": 0 + }, + { + "index": 5, + "start": "11:35", + "end": "12:20", + "time": 0 + }, + { + "index": 6, + "start": "14:00", + "end": "14:45", + "time": 1 + }, + { + "index": 7, + "start": "14:50", + "end": "15:35", + "time": 1 + }, + { + "index": 8, + "start": "15:55", + "end": "16:40", + "time": 1 + }, + { + "index": 9, + "start": "16:45", + "end": "17:30", + "time": 1 + }, + { + "index": 10, + "start": "19:00", + "end": "19:45", + "time": 2 + }, + { + "index": 11, + "start": "19:50", + "end": "20:35", + "time": 2 + }, + { + "index": 12, + "start": "20:40", + "end": "21:25", + "time": 2 + } + ] +} diff --git a/go.mod b/go.mod new file mode 100644 index 0000000..bf43402 --- /dev/null +++ b/go.mod @@ -0,0 +1,44 @@ +module zhixiangtong + +go 1.21 + +require ( + github.com/fatih/color v1.18.0 + github.com/gin-gonic/gin v1.10.0 + github.com/google/uuid v1.6.0 + gorm.io/driver/sqlite v1.5.6 + gorm.io/gorm v1.25.11 +) + +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/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/goccy/go-json v0.10.2 // 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-colorable v0.1.13 // indirect + github.com/mattn/go-isatty v0.0.20 // indirect + github.com/mattn/go-sqlite3 v1.14.22 // 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.2 // indirect + github.com/twitchyliquid64/golang-asm v0.15.1 // indirect + github.com/ugorji/go/codec v1.2.12 // indirect + golang.org/x/arch v0.8.0 // indirect + golang.org/x/crypto v0.23.0 // indirect + golang.org/x/net v0.25.0 // indirect + golang.org/x/sys v0.25.0 // indirect + golang.org/x/text v0.15.0 // indirect + google.golang.org/protobuf v1.34.1 // indirect + gopkg.in/yaml.v3 v3.0.1 // indirect +) diff --git a/go.sum b/go.sum new file mode 100644 index 0000000..c1de6cd --- /dev/null +++ b/go.sum @@ -0,0 +1,107 @@ +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/fatih/color v1.18.0 h1:S8gINlzdQ840/4pfAwic/ZE0djQEH3wM94VfqLTZcOM= +github.com/fatih/color v1.18.0/go.mod h1:4FelSpRwEGDpQ12mAdzqdOukCy4u8WUtOY6lkT/6HfU= +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/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.5.5 h1:Khx7svrCpmxxtHBq5j2mp/xVjsi8hQMfNLvJFAlrGgU= +github.com/google/go-cmp v0.5.5/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE= +github.com/google/gofuzz v1.0.0/go.mod h1:dBl0BpW6vV/+mYPU4Po3pmUjxk6FQPldtuIdl/M65Eg= +github.com/google/uuid v1.6.0 h1:NIvaJDMOsjHA8n1jAhLSgzrAzy1Hgr+hNrb57e+94F0= +github.com/google/uuid v1.6.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo= +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/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/mattn/go-colorable v0.1.13 h1:fFA4WZxdEF4tXPZVKMLwD8oUnCTTo08duU7wxecdEvA= +github.com/mattn/go-colorable v0.1.13/go.mod h1:7S9/ev0klgBDR4GtXTXX8a3vIGJpMovkB8vQcUbaXHg= +github.com/mattn/go-isatty v0.0.16/go.mod h1:kYGgaQfpe5nmfYZH+SKPsOc2e4SrIfOl2e/yFXSvRLM= +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/mattn/go-sqlite3 v1.14.22 h1:2gZY6PC6kBnID23Tichd1K+Z0oS6nE/XwU+Vz/5o4kU= +github.com/mattn/go-sqlite3 v1.14.22/go.mod h1:Uh1q+B4BYcTPb+yiD3kU8Ct7aC0hY9fxUwlHK0RXw+Y= +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.2 h1:aYUidT7k73Pcl9nb2gScu7NSrKCSHIDE89b3+6Wq+LM= +github.com/pelletier/go-toml/v2 v2.2.2/go.mod h1:1t835xjRzz80PqgE6HHgN2JOsmgYu/h4qDAS4n929Rs= +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/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/objx v0.5.2/go.mod h1:FRsXN1f5AsAjCGJKqEizvkpNtU+EGNCLh3NxZ/8L+MA= +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.8.4/go.mod h1:sz/lmYIOXD/1dqDmKjjqLyZ2RngseejIcXlSw2iwfAo= +github.com/stretchr/testify v1.9.0 h1:HtqpIVDClZ4nwg75+f6Lvsy/wHu+3BoSGCbBAcpTsTg= +github.com/stretchr/testify v1.9.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY= +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= +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.23.0 h1:dIJU/v2J8Mdglj/8rJ6UUOM3Zc9zLZxVZwwxMooUSAI= +golang.org/x/crypto v0.23.0/go.mod h1:CKFgDieR+mRhux2Lsu27y0fO304Db0wZe70UKqHu0v8= +golang.org/x/net v0.25.0 h1:d/OCCoBEUq33pjydKrGQhw7IlUPI2Oylr+8qLx49kac= +golang.org/x/net v0.25.0/go.mod h1:JkAGAh7GEvH74S6FOH42FLoXpXbE/aqXSrIQjXgsiwM= +golang.org/x/sys v0.0.0-20220811171246-fbc7d0a398ab/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +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.25.0 h1:r+8e+loiHxRqhXVl6ML1nO3l1+oFoWbnlu2Ehimmi34= +golang.org/x/sys v0.25.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA= +golang.org/x/text v0.15.0 h1:h1V/4gjBv8v9cjcR6+AR5+/cIYK5N/WAgiv4xlsEtAk= +golang.org/x/text v0.15.0/go.mod h1:18ZOQIKpY8NJVqYksKHtTdi31H5itFRjB5/qKTNYzSU= +golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543 h1:E7g+9GITq07hpfrRu66IVDexMakfv52eLZ2CXBWiKr4= +golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0= +google.golang.org/protobuf v1.34.1 h1:9ddQBjfCyZPOHPUiPxpYESBLc+T8P3E+Vo4IbKZgFWg= +google.golang.org/protobuf v1.34.1/go.mod h1:c6P6GXX6sHbq/GpV6MGZEdwhWPcYBgnhAHhKbcUYpos= +gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405 h1:yhCVgyC4o1eVCa2tZl7eS0r+SDo693bJlVdllGtEeKM= +gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0= +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/sqlite v1.5.6 h1:fO/X46qn5NUEEOZtnjJRWRzZMe8nqJiQ9E+0hi+hKQE= +gorm.io/driver/sqlite v1.5.6/go.mod h1:U+J8craQU6Fzkcvu8oLeAQmi50TkwPEhHDEjQZXDah4= +gorm.io/gorm v1.25.11 h1:/Wfyg1B/je1hnDx3sMkX+gAlxrlZpn6X0BXRlwXlvHg= +gorm.io/gorm v1.25.11/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/main.go b/main.go new file mode 100644 index 0000000..884076e --- /dev/null +++ b/main.go @@ -0,0 +1,83 @@ +package main + +import ( + "fmt" + "github.com/fatih/color" + "github.com/gin-gonic/gin" + "log" + "os" + "runtime" + "syscall" + "unsafe" + "zhixiangtong/utils" +) + +var ( + kernel32 = syscall.NewLazyDLL("kernel32.dll") + setConsoleTitle = kernel32.NewProc("SetConsoleTitleW") + setConsoleMode = kernel32.NewProc("SetConsoleMode") +) + +const ( + ENABLE_VIRTUAL_TERMINAL_PROCESSING = 0x0004 +) + +// enableVirtualTerminalProcessing 启用 Windows 的虚拟终端处理 +func enableVirtualTerminalProcessing(fd uintptr) { + var mode uint32 + handle := syscall.Handle(fd) + syscall.GetConsoleMode(handle, &mode) + mode |= ENABLE_VIRTUAL_TERMINAL_PROCESSING + setConsoleMode.Call(uintptr(handle), uintptr(mode)) +} + +// 设置终端标题 +func setTerminalTitle(title string) { + utf16Title, _ := syscall.UTF16PtrFromString(title) + setConsoleTitle.Call(uintptr(unsafe.Pointer(utf16Title))) +} + +func main() { + // 设置终端标题 + setTerminalTitle("智享通") + if os.Getenv("TERM") == "" { + // 在 Windows 平台上启用虚拟终端处理，以支持颜色输出 + enableVirtualTerminalProcessing(os.Stdout.Fd()) + } + // 初始化配置 + dataDir := "data" + // 确保 data 目录存在 + if _, err := os.Stat(dataDir); os.IsNotExist(err) { + err := os.Mkdir(dataDir, os.ModePerm) + if err != nil { + log.Fatalf("无法创建 data 目录：%v", err) + } + } + config, err := utils.InitConfig(dataDir) + if err != nil { + log.Fatalf("初始化配置失败：%v", err) + } + + // 初始化数据库 + db, err := utils.InitDatabase(dataDir) + if err != nil { + log.Fatalf("初始化数据库失败：%v", err) + } + // 如果是 Windows 平台，强制启用颜色支持 + if runtime.GOOS == "windows" { + color.NoColor = false + } + // 设置为生产模式 + gin.SetMode(gin.ReleaseMode) + // 配置路由 + router := utils.SetupRouter(db, config) + + // 启动服务 + addr := fmt.Sprintf("%s:%s", config.Host, config.Port) + log.Printf("服务正在运行在 %s", addr) + err = router.Run(addr) + if err != nil { + log.Fatalf("服务框架启动失败 %v", err) + return + } +} diff --git a/models/building.go b/models/building.go new file mode 100644 index 0000000..b4920db --- /dev/null +++ b/models/building.go @@ -0,0 +1,20 @@ +package models + +import ( + "github.com/google/uuid" + "gorm.io/gorm" +) + +// Building 教学楼模型 +type Building struct { + ID string `gorm:"type:uuid;primaryKey" json:"id"` + Name string `gorm:"type:varchar(255);not null" json:"name"` + Campus string `gorm:"type:varchar(255);not null" json:"campus"` // 教学楼所属校区名称 + CampusID string `gorm:"type:uuid;not null" json:"campus_id"` // 教学楼所属校区ID（外键） +} + +// BeforeCreate 在创建Building之前生成UUID +func (building *Building) BeforeCreate(tx *gorm.DB) (err error) { + building.ID = uuid.New().String() + return +} diff --git a/models/campus.go b/models/campus.go new file mode 100644 index 0000000..8e3da2e --- /dev/null +++ b/models/campus.go @@ -0,0 +1,19 @@ +// Package models +package models + +import ( + "github.com/google/uuid" + "gorm.io/gorm" +) + +// Campus 校区模型 +type Campus struct { + ID string `gorm:"type:uuid;primaryKey" json:"id"` + Name string `gorm:"type:varchar(255);not null" json:"name"` +} + +// BeforeCreate 在创建Campus之前生成UUID +func (campus *Campus) BeforeCreate(tx *gorm.DB) (err error) { + campus.ID = uuid.New().String() + return +} diff --git a/models/classname.go b/models/classname.go new file mode 100644 index 0000000..a98c40d --- /dev/null +++ b/models/classname.go @@ -0,0 +1,20 @@ +package models + +import ( + "github.com/google/uuid" + "gorm.io/gorm" +) + +// Classname 班级表 +type Classname struct { + ID string `gorm:"type:uuid;primaryKey" json:"id"` + Name string `gorm:"type:varchar(255);not null" json:"name"` + Campus string `gorm:"type:varchar(255);not null" json:"campus"` // 教学楼所属校区名称 + CampusID string `gorm:"type:uuid;not null" json:"campus_id"` // 教学楼所属校区ID（外键） +} + +// BeforeCreate 在创建Classname之前生成UUID +func (classname *Classname) BeforeCreate(tx *gorm.DB) (err error) { + classname.ID = uuid.New().String() + return +} diff --git a/models/classroom.go b/models/classroom.go new file mode 100644 index 0000000..0dcf1ff --- /dev/null +++ b/models/classroom.go @@ -0,0 +1,22 @@ +package models + +import ( + "github.com/google/uuid" + "gorm.io/gorm" +) + +// Classroom 教室模型 +type Classroom struct { + ID string `gorm:"type:uuid;primaryKey" json:"id"` + Name string `gorm:"type:varchar(255);not null" json:"name"` + Build string `gorm:"type:varchar(255);not null" json:"build"` + BuildID string `gorm:"type:uuid;not null" json:"build_id"` + CampusID string `gorm:"type:uuid;not null" json:"campus_id"` // 校区ID + Campus string `gorm:"type:varchar(255);not null" json:"campus"` // 校区名称 +} + +// BeforeCreate 在创建Classroom之前生成UUID +func (classroom *Classroom) BeforeCreate(tx *gorm.DB) (err error) { + classroom.ID = uuid.New().String() + return +} diff --git a/models/course.go b/models/course.go new file mode 100644 index 0000000..4625553 --- /dev/null +++ b/models/course.go @@ -0,0 +1,37 @@ +package models + +import ( + "time" + + "github.com/google/uuid" + "gorm.io/gorm" +) + +// Course 课程模型 +type Course struct { + ID string `gorm:"type:uuid;primaryKey;column:id" json:"id"` // 课程ID + ClassName string `gorm:"type:varchar(255);not null;column:classname" json:"classname"` // 班级名称 + ClassNameID string `gorm:"type:uuid;not null;column:classname_id" json:"classname_id"` // 班级ID + Course string `gorm:"type:varchar(255);not null;column:course" json:"course"` // 课程名称 + Teacher string `gorm:"type:varchar(255);not null;column:teacher" json:"teacher"` // 授课老师 + Classroom string `gorm:"type:varchar(255);not null;column:classroom" json:"classroom"` // 教室名称 + ClassroomID string `gorm:"type:uuid;not null;column:classroom_id" json:"classroom_id"` // 教室ID + Week int `gorm:"not null;column:week" json:"week"` // 周次 + Day int `gorm:"not null;column:day" json:"day"` // 星期几 + ClassTime int `gorm:"not null;column:classTime" json:"classTime"` // 第几节课 + StartTime time.Time `gorm:"type:datetime;not null;column:startTime" json:"startTime"` // 开始时间 + StartStamp float64 `gorm:"not null;column:startStamp" json:"startStamp"` // 开始时间戳 + EndTime time.Time `gorm:"type:datetime;not null;column:endTime" json:"endTime"` // 结束时间 + EndStamp float64 `gorm:"not null;column:endStamp" json:"endStamp"` // 结束时间戳 + CampusID string `gorm:"type:uuid;not null;column:campus_id" json:"campus_id"` // 校区ID + Campus string `gorm:"type:varchar(255);not null;column:campus" json:"campus"` // 校区名称 + BuildID string `gorm:"type:uuid;not null;column:build_id" json:"build_id"` // 教学楼ID + Build string `gorm:"type:varchar(255);not null;column:build" json:"build"` // 教学楼名称 + Status int `gorm:"not null;column:status" json:"status"` // 状态 +} + +// BeforeCreate 在创建Course之前生成UUID +func (course *Course) BeforeCreate(tx *gorm.DB) (err error) { + course.ID = uuid.New().String() + return +} diff --git a/rsrc.syso b/rsrc.syso new file mode 100644 index 0000000..4f51dbd Binary files /dev/null and b/rsrc.syso differ diff --git a/utils/auth.go b/utils/auth.go new file mode 100644 index 0000000..c4fd840 --- /dev/null +++ b/utils/auth.go @@ -0,0 +1,74 @@ +package utils + +import ( + "fmt" + "github.com/fatih/color" + "github.com/gin-gonic/gin" + "net/http" + "time" +) + +func CORSMiddleware() gin.HandlerFunc { + return func(c *gin.Context) { + // 设置允许访问的域 + c.Writer.Header().Set("Access-Control-Allow-Origin", "*") + // 是否允许携带用户凭证 + c.Writer.Header().Set("Access-Control-Allow-Credentials", "true") + // 允许的自定义请求头 + c.Writer.Header().Set("Access-Control-Allow-Headers", "Content-Type, Authorization, X-Requested-With, X-CSRF-Token, Accept, Origin") + // 允许的HTTP方法 + c.Writer.Header().Set("Access-Control-Allow-Methods", "GET, POST, PUT, DELETE, OPTIONS, PATCH") + + // 处理预检请求 + if c.Request.Method == "OPTIONS" { + c.AbortWithStatus(http.StatusOK) + return + } + + // 继续处理非预检请求 + c.Next() + } +} + +func CustomLogger() gin.HandlerFunc { + return func(c *gin.Context) { + // 获取开始时间 + start := time.Now() + + // 处理请求 + c.Next() + + // 获取状态码、耗时、客户端IP、方法、路径 + statusCode := c.Writer.Status() + latency := time.Since(start) + clientIP := c.ClientIP() + method := c.Request.Method + path := c.Request.URL.Path + + // 颜色设置 + statusColor := color.New(color.FgGreen).SprintFunc() + if statusCode >= 400 && statusCode < 500 { + statusColor = color.New(color.FgYellow).SprintFunc() + } else if statusCode >= 500 { + statusColor = color.New(color.FgRed).SprintFunc() + } + + methodColor := color.New(color.FgCyan).SprintFunc() + + // 打印彩色日志 + fmt.Printf("[GIN] %v | %s | %s | %13v | %s | %s\n", + time.Now().Format("2006/01/02 - 15:04:05"), + statusColor(statusCode), + clientIP, + latency, + methodColor(method), + path, + ) + } +} +func NotFoundHandler(c *gin.Context) { + c.JSON(http.StatusNotFound, gin.H{"code": 404, "msg": "无效路径！"}) +} +func NotMethodHandler(c *gin.Context) { + c.JSON(http.StatusMethodNotAllowed, gin.H{"code": 404, "msg": "请求方法错误！"}) +} diff --git a/utils/config.go b/utils/config.go new file mode 100644 index 0000000..a1a41a1 --- /dev/null +++ b/utils/config.go @@ -0,0 +1,123 @@ +package utils + +import ( + "encoding/json" + "fmt" + "io/ioutil" + "log" + "os" + "path/filepath" + + "gorm.io/driver/sqlite" + "gorm.io/gorm" + + "zhixiangtong/models" +) + +type TimeSet struct { + Index int `json:"index"` + Start string `json:"start"` + End string `json:"end"` + Time int `json:"time"` +} + +// Config 结构体定义服务配置 +type Config struct { + Host string `json:"host"` + Port string `json:"port"` + StartDate string `json:"startDate"` + TotalWeeks int `json:"totalWeeks"` + EndDate string `json:"endDate"` + TimeSetting []TimeSet `json:"timeSetting"` +} + +// InitConfig 初始化配置文件 +func InitConfig(dataDir string) (*Config, error) { + configPath := filepath.Join(dataDir, "config.json") + + // 检查配置文件是否存在 + if _, err := os.Stat(configPath); os.IsNotExist(err) { + log.Printf("配置文件不存在，正在创建默认配置文件：%s", configPath) + // JSON 数据 + timeSettingJSON := `[{"index": 1, "start": "08:00", "end": "08:45", "time": 0}, + {"index": 2, "start": "08:50", "end": "09:35", "time": 0}, + {"index": 3, "start": "09:55", "end": "10:40", "time": 0}, + {"index": 4, "start": "10:45", "end": "11:30", "time": 0}, + {"index": 5, "start": "11:35", "end": "12:20", "time": 0}, + {"index": 6, "start": "14:00", "end": "14:45", "time": 1}, + {"index": 7, "start": "14:50", "end": "15:35", "time": 1}, + {"index": 8, "start": "15:55", "end": "16:40", "time": 1}, + {"index": 9, "start": "16:45", "end": "17:30", "time": 1}, + {"index": 10, "start": "19:00", "end": "19:45", "time": 2}, + {"index": 11, "start": "19:50", "end": "20:35", "time": 2}, + {"index": 12, "start": "20:40", "end": "21:25", "time": 2}]` + + // 将 JSON 数据解析为 TimeSet 切片 + var timeSettings []TimeSet + err := json.Unmarshal([]byte(timeSettingJSON), &timeSettings) + if err != nil { + log.Fatalf("Failed to parse JSON: %v", err) + } + // 默认配置 + defaultConfig := Config{ + Host: "localhost", + Port: "8080", + StartDate: "2024-09-02", + EndDate: "2025-01-19", + TotalWeeks: 20, + TimeSetting: timeSettings, + } + + // 将默认配置写入配置文件 + configData, err := json.MarshalIndent(defaultConfig, "", " ") + if err != nil { + return nil, fmt.Errorf("无法创建默认配置文件：%v", err) + } + + err = ioutil.WriteFile(configPath, configData, 0644) + if err != nil { + return nil, fmt.Errorf("无法写入默认配置文件：%v", err) + } + } + + // 读取配置文件 + configData, err := ioutil.ReadFile(configPath) + if err != nil { + return nil, fmt.Errorf("无法读取配置文件：%v", err) + } + + // 解析配置文件 + var config Config + err = json.Unmarshal(configData, &config) + if err != nil { + return nil, fmt.Errorf("配置文件格式错误：%v", err) + } + + log.Printf("配置文件已加载：Host=%s, Port=%s", config.Host, config.Port) + return &config, nil +} + +// InitDatabase 初始化数据库，如果不存在则创建并迁移 +func InitDatabase(dataDir string) (*gorm.DB, error) { + dbPath := filepath.Join(dataDir, "database.db") + + // 检查数据库文件是否存在 + if _, err := os.Stat(dbPath); os.IsNotExist(err) { + log.Printf("数据库文件不存在，正在创建数据库：%s", dbPath) + } + + // 连接到SQLite数据库 + db, err := gorm.Open(sqlite.Open(dbPath), &gorm.Config{}) + if err != nil { + return nil, fmt.Errorf("无法连接到数据库：%v", err) + } + + // 自动迁移数据库 + err = db.AutoMigrate(&models.Campus{}, &models.Building{}, &models.Classroom{}, &models.Course{}, &models.Classname{}) + if err != nil { + return nil, fmt.Errorf("数据库迁移失败：%v", err) + } + + log.Printf("数据库已初始化并连接成功：%s", dbPath) + return db, nil +} diff --git a/utils/router.go b/utils/router.go new file mode 100644 index 0000000..45c1e6e --- /dev/null +++ b/utils/router.go @@ -0,0 +1,402 @@ +package utils + +import ( + "fmt" + "github.com/gin-gonic/gin" + "github.com/google/uuid" + "gorm.io/gorm" + "net/http" + "sort" + "strconv" + "sync" + "time" + "zhixiangtong/models" +) + +type Setting struct { + StartDate string `json:"startDate"` + TotalWeeks int `json:"totalWeeks"` + EndDate string `json:"endDate"` + TimeSetting []TimeSet `json:"timeSetting"` +} + +// RemoveIntDuplicates 去重函数，去除整数切片中的重复元素 +func RemoveIntDuplicates(elements []int) []int { + seen := make(map[int]bool) + var result []int + + for _, v := range elements { + if _, ok := seen[v]; !ok { + result = append(result, v) + seen[v] = true + } + } + + return result +} +func RemoveStringDuplicates(stringSlice []string) []string { + keys := make(map[string]bool) + var list []string + for _, entry := range stringSlice { + if _, value := keys[entry]; !value { + keys[entry] = true + list = append(list, entry) + } + } + return list +} + +// SetupRouter 配置Gin路由 +func SetupRouter(db *gorm.DB, config *Config) *gin.Engine { + r := gin.New() + r.Use(CustomLogger()) + // 处理跨域 + r.Use(CORSMiddleware()) + // 获取所有校区 + r.GET("/getCampusList", func(c *gin.Context) { + // 使用 WaitGroup 等待 Goroutine 完成 + var wg sync.WaitGroup + wg.Add(1) + go func() { + defer wg.Done() // 在 Goroutine 完成时调用 Done() + var campuses []models.Campus + if err := db.Find(&campuses).Error; err != nil { + c.JSON(http.StatusInternalServerError, gin.H{"code": 500, "msg": "无法获取校区列表", "data": nil}) + return + } + c.JSON(http.StatusOK, gin.H{"code": 200, "msg": "获取成功！", "data": campuses}) + return + }() + wg.Wait() + }) + // 获取校区内所有教学楼 + r.GET("/getBuilds/:id", func(c *gin.Context) { + campusId := c.Param("id") + if campusId == "" { + c.JSON(http.StatusBadRequest, gin.H{"code": 422, "msg": "参数错误，校区ID为必传项！", "data": nil}) + return + } + // 使用 WaitGroup 等待 Goroutine 完成 + var wg sync.WaitGroup + wg.Add(1) + + go func() { + defer wg.Done() // 在 Goroutine 完成时调用 Done() + var buildings []models.Building + if err := db.Find(&buildings, "campus_id = ?", campusId).Error; err != nil { + c.JSON(http.StatusNotFound, gin.H{"code": 500, "msg": "获取失败！", "data": nil}) + return + } + c.JSON(http.StatusOK, gin.H{"code": 200, "msg": "获取成功！", "data": buildings}) + return + }() + wg.Wait() + }) + // 获取说所有教学楼 + r.GET("/getBuildList", func(c *gin.Context) { + // 使用 WaitGroup 等待 Goroutine 完成 + var wg sync.WaitGroup + wg.Add(1) + + go func() { + defer wg.Done() // 在 Goroutine 完成时调用 Done() + var buildings []models.Building + if err := db.Find(&buildings).Error; err != nil { + c.JSON(http.StatusInternalServerError, gin.H{"code": 500, "msg": "无法获取教学楼列表", "data": nil}) + return + } + c.JSON(http.StatusOK, gin.H{"code": 200, "msg": "获取成功！", "data": buildings}) + return + }() + wg.Wait() + }) + // 获取教室列表 + r.GET("/getClassroomList/:id", func(c *gin.Context) { + buildId := c.Param("id") + var classrooms []models.Classroom + if buildId == "" { + c.JSON(http.StatusBadRequest, gin.H{"code": 422, "msg": "参数错误，教学楼ID为必传项！", "data": nil}) + return + } + // 使用 WaitGroup 等待 Goroutine 完成 + var wg sync.WaitGroup + wg.Add(1) + + go func() { + defer wg.Done() // 在 Goroutine 完成时调用 Done() + if result := db.Find(&classrooms, "build_id = ? ", buildId); result.Error != nil { + c.JSON(http.StatusNotFound, gin.H{"code": 500, "msg": "获取失败！", "data": nil}) + return + } + c.JSON(http.StatusOK, gin.H{"code": 200, "msg": "获取成功！", "data": classrooms}) + return + }() + wg.Wait() + }) + // 获取校区班级列表 + r.GET("/getCampusClass/:id", func(c *gin.Context) { + var classNames []models.Classname + campusId := c.Param("id") + if campusId == "" { + c.JSON(http.StatusBadRequest, gin.H{"code": 422, "msg": "参数错误，校区ID为必传项！", "data": nil}) + return + } + // 使用 WaitGroup 等待 Goroutine 完成 + var wg sync.WaitGroup + wg.Add(1) + + go func() { + defer wg.Done() // 在 Goroutine 完成时调用 Done() + if result := db.Find(&classNames, "campus_id = ?", campusId); result.Error != nil { + c.JSON(http.StatusNotFound, gin.H{"code": 500, "msg": "获取失败！", "data": nil}) + return + } + c.JSON(http.StatusOK, gin.H{"code": 200, "msg": "获取成功！", "data": classNames}) + return + }() + wg.Wait() + }) + // 获取班级课表 + r.GET("/getCourse/:id", func(c *gin.Context) { + var courses []models.Course + classnameId := c.Param("id") + if classnameId == "" { + c.JSON(http.StatusBadRequest, gin.H{"code": 422, "msg": "参数错误，班级ID为必传项！", "data": nil}) + return + } + // 使用 WaitGroup 等待 Goroutine 完成 + var wg sync.WaitGroup + wg.Add(1) + + go func() { + defer wg.Done() // 在 Goroutine 完成时调用 Done() + if result := db.Find(&courses, "classname_id = ? AND status = 1", classnameId); result.Error != nil { + c.JSON(http.StatusNotFound, gin.H{"code": 422, "msg": "课程查询失败！", "data": nil}) + return + } + // 按照课程名称、教室、星期几和上课时间去重 + uniqueCourses := make(map[string]map[string]map[int][]models.Course) + for _, course := range courses { + if uniqueCourses[course.Course] == nil { + uniqueCourses[course.Course] = make(map[string]map[int][]models.Course) + } + if uniqueCourses[course.Course][course.Classroom] == nil { + uniqueCourses[course.Course][course.Classroom] = make(map[int][]models.Course) + } + uniqueCourses[course.Course][course.Classroom][course.Day] = append(uniqueCourses[course.Course][course.Classroom][course.Day], course) + } + var resultList []map[string]interface{} + for courseName, classrooms := range uniqueCourses { + for classroom, days := range classrooms { + for day, courses := range days { + var weeks []int + for _, course := range courses { + weeks = append(weeks, course.Week) + } + weeks = RemoveIntDuplicates(weeks) + sort.Ints(weeks) + var timeRange []int + for _, t := range courses { + timeRange = append(timeRange, t.ClassTime) + } + timeRange = RemoveIntDuplicates(timeRange) + sort.Ints(timeRange) + resultList = append(resultList, map[string]interface{}{ + "id": uuid.New().String(), + "course": courseName, + "classroom": classroom, + "teacher": courses[0].Teacher, + "weeks": weeks, + "day": day + 1, + "time": timeRange, + }) + } + } + } + // 返回结果 + c.JSON(http.StatusOK, gin.H{ + "code": 200, + "msg": "获取成功！", + "data": resultList, + }) + return + }() + wg.Wait() + }) + // 获取空闲教室 + r.GET("/getEmptyClassroom", func(c *gin.Context) { + // 获取请求参数 + campusID := c.Query("campus_id") + buildID := c.Query("build_id") // 可选参数 + startStamp := c.Query("startStamp") + endStamp := c.Query("endStamp") + // 参数校验 + if campusID == "" || startStamp == "" || endStamp == "" { + c.JSON(http.StatusBadRequest, gin.H{"code": 422, "msg": "参数错误，校区ID和时间戳为必传项！", "data": nil}) + return + } + // 解析时间戳参数 + startTimestamp, err := strconv.ParseInt(startStamp, 10, 64) + if err != nil { + c.JSON(http.StatusBadRequest, gin.H{"code": 400, "msg": "开始时间戳格式错误！", "data": nil}) + return + } + endTimestamp, err := strconv.ParseInt(endStamp, 10, 64) + if err != nil { + c.JSON(http.StatusBadRequest, gin.H{"code": 400, "msg": "结束时间戳格式错误！", "data": nil}) + return + } + // 使用 WaitGroup 等待 Goroutine 完成 + var wg sync.WaitGroup + wg.Add(1) + + go func() { + defer wg.Done() // 在 Goroutine 完成时调用 Done() + // 查询有课的教室 classroom1 + var classroom1 []string + query := db.Model(&models.Course{}). + Where("campus_id = ? AND startStamp >= ? AND endStamp <= ? AND status = 1", + campusID, startTimestamp, endTimestamp) + if buildID != "" { + query = query.Where("build_id = ?", buildID) + } + err = query.Pluck("classroom_id", &classroom1).Error + if err != nil { + c.JSON(http.StatusInternalServerError, gin.H{"code": 500, "msg": "查询课程数据失败！", "data": nil}) + return + } + // 查询所有教室 classroom2，根据 campus_id 和可选的 build_id 过滤 + var classroom2 []models.Classroom + query = db.Where("campus_id = ?", campusID) + if buildID != "" { + query = query.Where("build_id = ?", buildID) + } + err = query.Find(&classroom2).Error + if err != nil { + c.JSON(http.StatusInternalServerError, gin.H{"code": 500, "msg": "查询教室数据失败！", "data": nil}) + return + } + // 使用一个 map 来判断 classroom1 中的教室是否有课 + classroom1Set := make(map[string]bool) + for _, id := range classroom1 { + classroom1Set[id] = true + } + // 过滤掉有课的教室，得到无课教室列表 data + var data []models.Classroom + for _, classroom := range classroom2 { + if !classroom1Set[classroom.ID] { + data = append(data, classroom) + } + } + c.JSON(http.StatusOK, gin.H{"code": 200, "msg": "查询成功！", "data": data}) + return + }() + wg.Wait() + }) + // 获取教室课表 + r.GET("/getClassroomCourses", func(c *gin.Context) { + var courses []models.Course + classroomID := c.Query("classroom_id") + weekStr := c.Query("week") + + if classroomID == "" || weekStr == "" { + c.JSON(http.StatusBadRequest, gin.H{"code": 422, "msg": "参数错误，教室ID、周数为必传项！", "data": nil}) + return + } + + week, err := strconv.Atoi(weekStr) + if err != nil { + c.JSON(http.StatusBadRequest, gin.H{"code": 422, "msg": "参数错误，周数格式不正确！", "data": nil}) + return + } + // 使用 WaitGroup 等待 Goroutine 完成 + var wg sync.WaitGroup + wg.Add(1) + + go func() { + defer wg.Done() // 在 Goroutine 完成时调用 Done() + // 根据 classroom_id 和时间范围筛选数据 + if result := db.Where("classroom_id = ? AND week = ? AND status = 1", classroomID, week).Find(&courses); result.Error != nil { + c.JSON(http.StatusInternalServerError, gin.H{"code": 500, "msg": "课程查询失败！", "data": nil}) + return + } + + // 按照 week, day 和 classTime 聚合 + type AggregatedCourse struct { + Week int `json:"week"` + Day int `json:"day"` + ClassTime int `json:"classTime"` + Classnames []string `json:"classnames"` + Campus string `json:"campus"` + Build string `json:"build"` + StartTime time.Time `json:"startTime"` + EndTime time.Time `json:"endTime"` + StartStamp float64 `json:"startStamp"` + EndStamp float64 `json:"endStamp"` + ClassRoom string `json:"classroom"` + Course string `json:"course"` + Teacher string `json:"teacher"` + ID string `json:"id"` + } + + aggregatedData := make(map[string]AggregatedCourse) + + for _, course := range courses { + key := fmt.Sprintf("%d-%d-%d", course.Week, course.Day, course.ClassTime) + if _, exists := aggregatedData[key]; !exists { + aggregatedData[key] = AggregatedCourse{ + Week: course.Week, + Day: course.Day, + ClassTime: course.ClassTime, + Classnames: []string{}, + Campus: course.Campus, + Build: course.Build, + StartStamp: course.StartStamp, + EndStamp: course.EndStamp, + StartTime: course.StartTime, + EndTime: course.EndTime, + ClassRoom: course.Classroom, + Course: course.Course, + Teacher: course.Teacher, + ID: course.ID, + } + } + + aggCourse := aggregatedData[key] + aggCourse.Classnames = append(aggCourse.Classnames, course.ClassName) + aggregatedData[key] = aggCourse + } + + var resultList []AggregatedCourse + for _, aggCourse := range aggregatedData { + // 去重课程名称 + aggCourse.Classnames = RemoveStringDuplicates(aggCourse.Classnames) + resultList = append(resultList, aggCourse) + } + + // 返回聚合结果 + c.JSON(http.StatusOK, gin.H{ + "code": 200, + "msg": "获取成功！", + "data": resultList, + }) + return + }() + wg.Wait() + }) + // 获取学校配置 + r.GET("/getTermSetting", func(c *gin.Context) { + data := Setting{ + TimeSetting: config.TimeSetting, + StartDate: config.StartDate, + EndDate: config.EndDate, + TotalWeeks: config.TotalWeeks, + } + c.JSON(http.StatusOK, gin.H{"code": 200, "msg": "获取成功", "data": data}) + }) + // 处理请求方法错 + r.NoMethod(NotMethodHandler) + // 处理无效路径 + r.NoRoute(NotFoundHandler) + return r +} diff --git a/智享通.ico b/智享通.ico new file mode 100644 index 0000000..5f250c3 Binary files /dev/null and b/智享通.ico differ