直接完成
This commit is contained in:
當前提交
b924e94c63
2
.gitignore
vendored
Normal file
2
.gitignore
vendored
Normal file
@ -0,0 +1,2 @@
|
|||||||
|
.idea
|
||||||
|
integrity-checkin
|
||||||
104
dao/checkin.go
Normal file
104
dao/checkin.go
Normal file
@ -0,0 +1,104 @@
|
|||||||
|
package dao
|
||||||
|
|
||||||
|
import (
|
||||||
|
"errors"
|
||||||
|
"gorm.io/gorm"
|
||||||
|
"integrity-checkin/data/dto"
|
||||||
|
"integrity-checkin/data/po"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
type CheckinDAO struct {
|
||||||
|
Tx *gorm.DB
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c CheckinDAO) FindUserActivityByDay(userID int, activityID int, day string) *po.UserActivity {
|
||||||
|
var userActivity po.UserActivity
|
||||||
|
err := c.Tx.Take(&userActivity, "user_id=? and activity_id=? and day=?", userID, activityID, day).Error
|
||||||
|
if err == gorm.ErrRecordNotFound {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
return &userActivity
|
||||||
|
}
|
||||||
|
func (c CheckinDAO) CreateCheckin(userID int, activityID int, image string) (*po.UserActivity, error) {
|
||||||
|
if activity := c.FindActivityByID(activityID); activity == nil {
|
||||||
|
return nil, errors.New("activityID不存在")
|
||||||
|
}
|
||||||
|
userActivity := po.UserActivity{
|
||||||
|
UserID: userID,
|
||||||
|
UserActivityForm: po.UserActivityForm{
|
||||||
|
ActivityID: activityID,
|
||||||
|
Image: image,
|
||||||
|
},
|
||||||
|
Status: 1,
|
||||||
|
Day: time.Now().Format("20060102"),
|
||||||
|
}
|
||||||
|
err := c.Tx.Create(&userActivity).Error
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
return &userActivity, nil
|
||||||
|
}
|
||||||
|
func (c CheckinDAO) FindActivityByID(id int) *po.Activity {
|
||||||
|
var activity po.Activity
|
||||||
|
err := c.Tx.Take(&activity, id).Error
|
||||||
|
if err == gorm.ErrRecordNotFound {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
return &activity
|
||||||
|
}
|
||||||
|
func (c CheckinDAO) FindUserActivityByID(id int) *po.UserActivity {
|
||||||
|
var userActivity po.UserActivity
|
||||||
|
err := c.Tx.Take(&userActivity, id).Error
|
||||||
|
if err == gorm.ErrRecordNotFound {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
return &userActivity
|
||||||
|
}
|
||||||
|
func (c CheckinDAO) SaveUserActivity(userActivity *po.UserActivity) {
|
||||||
|
err := c.Tx.Save(userActivity).Error
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
func (c CheckinDAO) FindAllActivities() []po.Activity {
|
||||||
|
var arr []po.Activity
|
||||||
|
err := c.Tx.Find(&arr).Error
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
return arr
|
||||||
|
}
|
||||||
|
func getUserActivityExtendedSQL(where string) string {
|
||||||
|
return "select ua.*,a.category activity_category," +
|
||||||
|
"a.name activity_name from user_activity ua left join activities a on a.id=ua.activity_id " + where +
|
||||||
|
" order by created_at desc"
|
||||||
|
}
|
||||||
|
func (c CheckinDAO) FindAllUserActivities() []dto.UserActivityExtended {
|
||||||
|
var arr []dto.UserActivityExtended
|
||||||
|
err := c.Tx.Raw(getUserActivityExtendedSQL("")).Find(&arr).Error
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
return arr
|
||||||
|
}
|
||||||
|
func (c CheckinDAO) FindUserActivityExtendedByID(id int) *dto.UserActivityExtended {
|
||||||
|
var userActivityExtended dto.UserActivityExtended
|
||||||
|
err := c.Tx.Raw(getUserActivityExtendedSQL("where id=?"), id).Take(&userActivityExtended).Error
|
||||||
|
if err == gorm.ErrRecordNotFound {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
return &userActivityExtended
|
||||||
|
}
|
||||||
22
dao/dao.go
Normal file
22
dao/dao.go
Normal file
@ -0,0 +1,22 @@
|
|||||||
|
package dao
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"gorm.io/driver/mysql"
|
||||||
|
"gorm.io/gorm"
|
||||||
|
)
|
||||||
|
|
||||||
|
var DB *gorm.DB
|
||||||
|
|
||||||
|
func Setup() {
|
||||||
|
db, err := gorm.Open(mysql.Open(fmt.Sprintf("%v:%v@tcp(%v:%v)/%v?charset=utf8mb4&parseTime=True&loc=Local",
|
||||||
|
"integrity",
|
||||||
|
"integrity",
|
||||||
|
"202.194.15.130",
|
||||||
|
3306,
|
||||||
|
"integrity")), &gorm.Config{})
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
DB = db
|
||||||
|
}
|
||||||
51
dao/user.go
Normal file
51
dao/user.go
Normal file
@ -0,0 +1,51 @@
|
|||||||
|
package dao
|
||||||
|
|
||||||
|
import (
|
||||||
|
"gorm.io/gorm"
|
||||||
|
"integrity-checkin/data/po"
|
||||||
|
)
|
||||||
|
|
||||||
|
type UserDAO struct {
|
||||||
|
Tx *gorm.DB
|
||||||
|
}
|
||||||
|
|
||||||
|
func (u UserDAO) FindUserByUsername(username string) *po.User {
|
||||||
|
var user po.User
|
||||||
|
err := u.Tx.Take(&user, "username=?", username).Error
|
||||||
|
if err == gorm.ErrRecordNotFound {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
return &user
|
||||||
|
}
|
||||||
|
func (u UserDAO) FindUserByID(id int) *po.User {
|
||||||
|
var user po.User
|
||||||
|
err := u.Tx.Take(&user, id).Error
|
||||||
|
if err == gorm.ErrRecordNotFound {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
return &user
|
||||||
|
}
|
||||||
|
func (u UserDAO) SaveUser(user *po.User) {
|
||||||
|
err := u.Tx.Save(user).Error
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
func (u UserDAO) CreateUser(username string, realName string) *po.User {
|
||||||
|
user := po.User{
|
||||||
|
Username: username,
|
||||||
|
RealName: realName,
|
||||||
|
RoleID: 1,
|
||||||
|
}
|
||||||
|
err := u.Tx.Create(&user).Error
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
return &user
|
||||||
|
}
|
||||||
17
data/dto/user.go
Normal file
17
data/dto/user.go
Normal file
@ -0,0 +1,17 @@
|
|||||||
|
package dto
|
||||||
|
|
||||||
|
import "integrity-checkin/data/po"
|
||||||
|
|
||||||
|
type UserClaims struct {
|
||||||
|
UserID int `json:"user_id"`
|
||||||
|
RoleID int `json:"role_id"`
|
||||||
|
}
|
||||||
|
type SDULoginData struct {
|
||||||
|
RealName string
|
||||||
|
Username string
|
||||||
|
}
|
||||||
|
type UserActivityExtended struct {
|
||||||
|
po.UserActivity
|
||||||
|
ActivityCategory string `json:"activity_category"`
|
||||||
|
ActivityName string `json:"activity_name"`
|
||||||
|
}
|
||||||
5
data/itf/itf.go
Normal file
5
data/itf/itf.go
Normal file
@ -0,0 +1,5 @@
|
|||||||
|
package itf
|
||||||
|
|
||||||
|
type IValidator interface {
|
||||||
|
Validate() error
|
||||||
|
}
|
||||||
44
data/po/checkin.go
Normal file
44
data/po/checkin.go
Normal file
@ -0,0 +1,44 @@
|
|||||||
|
package po
|
||||||
|
|
||||||
|
import (
|
||||||
|
"errors"
|
||||||
|
"strings"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
type Activity struct {
|
||||||
|
ID int `json:"id"`
|
||||||
|
Category string `json:"category"`
|
||||||
|
Name string `json:"name"`
|
||||||
|
Credit int `json:"credit"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a Activity) TableName() string {
|
||||||
|
return "activities"
|
||||||
|
}
|
||||||
|
|
||||||
|
type UserActivity struct {
|
||||||
|
ID int `json:"id"`
|
||||||
|
UserID int `json:"user_id"`
|
||||||
|
UserActivityForm
|
||||||
|
Status int `json:"status"`
|
||||||
|
Day string `json:"day"`
|
||||||
|
CreatedAt time.Time `json:"created_at"`
|
||||||
|
UpdatedAt time.Time `json:"updated_at"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (u UserActivity) TableName() string {
|
||||||
|
return "user_activity"
|
||||||
|
}
|
||||||
|
|
||||||
|
type UserActivityForm struct {
|
||||||
|
ActivityID int `json:"activity_id" form:"activity_id" binding:"required"`
|
||||||
|
Image string `json:"image" form:"image" binding:"required,url"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (u UserActivityForm) Validate() error {
|
||||||
|
if !strings.HasPrefix(u.Image, "https://m.360buyimg.com") {
|
||||||
|
return errors.New("不是京东图床链接")
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
9
data/po/user.go
Normal file
9
data/po/user.go
Normal file
@ -0,0 +1,9 @@
|
|||||||
|
package po
|
||||||
|
|
||||||
|
type User struct {
|
||||||
|
ID int `json:"id"`
|
||||||
|
Username string `json:"username"`
|
||||||
|
RealName string `json:"real_name"`
|
||||||
|
Credit int `json:"credit"`
|
||||||
|
RoleID int `json:"role_id"`
|
||||||
|
}
|
||||||
36
go.mod
Normal file
36
go.mod
Normal file
@ -0,0 +1,36 @@
|
|||||||
|
module integrity-checkin
|
||||||
|
|
||||||
|
go 1.19
|
||||||
|
|
||||||
|
require (
|
||||||
|
github.com/PuerkitoBio/goquery v1.8.0 // indirect
|
||||||
|
github.com/andybalholm/cascadia v1.3.1 // indirect
|
||||||
|
github.com/ekyoung/gin-nice-recovery v0.0.0-20160510022553-1654dca486db // indirect
|
||||||
|
github.com/gin-contrib/sse v0.1.0 // indirect
|
||||||
|
github.com/gin-gonic/gin v1.8.1 // indirect
|
||||||
|
github.com/go-errors/errors v1.4.2 // indirect
|
||||||
|
github.com/go-playground/locales v0.14.0 // indirect
|
||||||
|
github.com/go-playground/universal-translator v0.18.0 // indirect
|
||||||
|
github.com/go-playground/validator/v10 v10.11.1 // indirect
|
||||||
|
github.com/go-resty/resty/v2 v2.7.0 // indirect
|
||||||
|
github.com/go-sql-driver/mysql v1.6.0 // indirect
|
||||||
|
github.com/goccy/go-json v0.9.11 // indirect
|
||||||
|
github.com/golang-jwt/jwt v3.2.2+incompatible // 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/leodido/go-urn v1.2.1 // indirect
|
||||||
|
github.com/mattn/go-isatty v0.0.16 // 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.0.6 // indirect
|
||||||
|
github.com/ugorji/go/codec v1.2.7 // indirect
|
||||||
|
golang.org/x/crypto v0.3.0 // indirect
|
||||||
|
golang.org/x/net v0.2.0 // indirect
|
||||||
|
golang.org/x/sys v0.2.0 // indirect
|
||||||
|
golang.org/x/text v0.4.0 // indirect
|
||||||
|
google.golang.org/protobuf v1.28.1 // indirect
|
||||||
|
gopkg.in/yaml.v2 v2.4.0 // indirect
|
||||||
|
gorm.io/driver/mysql v1.4.4 // indirect
|
||||||
|
gorm.io/gorm v1.24.2 // indirect
|
||||||
|
)
|
||||||
113
go.sum
Normal file
113
go.sum
Normal file
@ -0,0 +1,113 @@
|
|||||||
|
github.com/PuerkitoBio/goquery v1.8.0 h1:PJTF7AmFCFKk1N6V6jmKfrNH9tV5pNE6lZMkG0gta/U=
|
||||||
|
github.com/PuerkitoBio/goquery v1.8.0/go.mod h1:ypIiRMtY7COPGk+I/YbZLbxsxn9g5ejnI2HSMtkjZvI=
|
||||||
|
github.com/andybalholm/cascadia v1.3.1 h1:nhxRkql1kdYCc8Snf7D5/D3spOX+dBgjA6u8x004T2c=
|
||||||
|
github.com/andybalholm/cascadia v1.3.1/go.mod h1:R4bJ1UQfqADjvDa4P6HZHLh/3OxWWEqc0Sk8XGwHqvA=
|
||||||
|
github.com/creack/pty v1.1.9/go.mod h1:oKZEueFk5CKHvIhNR5MUki03XCEU+Q6VDXinZuGJ33E=
|
||||||
|
github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
||||||
|
github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
||||||
|
github.com/ekyoung/gin-nice-recovery v0.0.0-20160510022553-1654dca486db h1:oZ4U9IqO8NS+61OmGTBi8vopzqTRxwQeogyBHdrhjbc=
|
||||||
|
github.com/ekyoung/gin-nice-recovery v0.0.0-20160510022553-1654dca486db/go.mod h1:Pk7/9x6tyChFTkahDvLBQMlvdsWvfC+yU8HTT5VD314=
|
||||||
|
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.8.1 h1:4+fr/el88TOO3ewCmQr8cx/CtZ/umlIRIs5M4NTNjf8=
|
||||||
|
github.com/gin-gonic/gin v1.8.1/go.mod h1:ji8BvRH1azfM+SYow9zQ6SZMvR8qOMZHmsCuWR9tTTk=
|
||||||
|
github.com/go-errors/errors v1.4.2 h1:J6MZopCL4uSllY1OfXM374weqZFFItUbrImctkmUxIA=
|
||||||
|
github.com/go-errors/errors v1.4.2/go.mod h1:sIVyrIiJhuEF+Pj9Ebtd6P/rEYROXFi3BopGUQ5a5Og=
|
||||||
|
github.com/go-playground/assert/v2 v2.0.1/go.mod h1:VDjEfimB/XKnb+ZQfWdccd7VUvScMdVu0Titje2rxJ4=
|
||||||
|
github.com/go-playground/locales v0.14.0 h1:u50s323jtVGugKlcYeyzC0etD1HifMjqmJqb8WugfUU=
|
||||||
|
github.com/go-playground/locales v0.14.0/go.mod h1:sawfccIbzZTqEDETgFXqTho0QybSa7l++s0DH+LDiLs=
|
||||||
|
github.com/go-playground/universal-translator v0.18.0 h1:82dyy6p4OuJq4/CByFNOn/jYrnRPArHwAcmLoJZxyho=
|
||||||
|
github.com/go-playground/universal-translator v0.18.0/go.mod h1:UvRDBj+xPUEGrFYl+lu/H90nyDXpg0fqeB/AQUGNTVA=
|
||||||
|
github.com/go-playground/validator/v10 v10.11.1 h1:prmOlTVv+YjZjmRmNSF3VmspqJIxJWXmqUsHwfTRRkQ=
|
||||||
|
github.com/go-playground/validator/v10 v10.11.1/go.mod h1:i+3WkQ1FvaUjjxh1kSvIA4dMGDBiPU55YFDl0WbKdWU=
|
||||||
|
github.com/go-resty/resty/v2 v2.7.0 h1:me+K9p3uhSmXtrBZ4k9jcEAfJmuC8IivWHwaLZwPrFY=
|
||||||
|
github.com/go-resty/resty/v2 v2.7.0/go.mod h1:9PWDzw47qPphMRFfhsyk0NnSgvluHcljSMVIq3w7q0I=
|
||||||
|
github.com/go-sql-driver/mysql v1.6.0 h1:BCTh4TKNUYmOmMUcQ3IipzF5prigylS7XXjEkfCHuOE=
|
||||||
|
github.com/go-sql-driver/mysql v1.6.0/go.mod h1:DCzpHaOWr8IXmIStZouvnhqoel9Qv2LBy8hT2VhHyBg=
|
||||||
|
github.com/goccy/go-json v0.9.11 h1:/pAaQDLHEoCq/5FFmSKBswWmK6H0e8g4159Kc/X/nqk=
|
||||||
|
github.com/goccy/go-json v0.9.11/go.mod h1:6MelG93GURQebXPDq3khkgXZkazVtN9CRI+MGFi0w8I=
|
||||||
|
github.com/golang-jwt/jwt v3.2.2+incompatible h1:IfV12K8xAKAnZqdXVzCZ+TOjboZ2keLg81eXfW3O+oY=
|
||||||
|
github.com/golang-jwt/jwt v3.2.2+incompatible/go.mod h1:8pz2t5EyA70fFQQSrl6XZXzqecmYZeUEB8OUGHkxJ+I=
|
||||||
|
github.com/golang/protobuf v1.5.0/go.mod h1:FsONVRAS9T7sI+LIUmWTfcYkHO4aIWwzhcaSAoJOfIk=
|
||||||
|
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/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.4/go.mod h1:d3SSVoowX0Lcu0IBviAWJpolVfI5UJVZZ7cO71lE/z8=
|
||||||
|
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/kr/pretty v0.1.0/go.mod h1:dAy3ld7l9f0ibDNOQOHHMYYIIbhfbHSm3C4ZsoJORNo=
|
||||||
|
github.com/kr/pretty v0.2.1/go.mod h1:ipq/a2n7PKx3OHsz4KJII5eveXtPO4qwEXGdVfWzfnI=
|
||||||
|
github.com/kr/pretty v0.3.0/go.mod h1:640gp4NfQd8pI5XOwp5fnNeVWj67G7CFk/SaSQn7NBk=
|
||||||
|
github.com/kr/pty v1.1.1/go.mod h1:pFQYn66WHrOpPYNljwOMqo10TkYh1fy3cYio2l3bCsQ=
|
||||||
|
github.com/kr/text v0.1.0/go.mod h1:4Jbv+DJW3UT/LiOwJeYQe1efqtUx/iVham/4vfdArNI=
|
||||||
|
github.com/kr/text v0.2.0/go.mod h1:eLer722TekiGuMkidMxC/pM04lWEeraHUUmBw8l2grE=
|
||||||
|
github.com/leodido/go-urn v1.2.1 h1:BqpAaACuzVSgi/VLzGZIobT2z4v53pjosyNd9Yv6n/w=
|
||||||
|
github.com/leodido/go-urn v1.2.1/go.mod h1:zt4jvISO2HfUBqxjfIshjdMTYS56ZS/qv49ictyFfxY=
|
||||||
|
github.com/mattn/go-isatty v0.0.16 h1:bq3VjFmv/sOjHtdEhmkEV4x1AJtvUvOJ2PFAZ5+peKQ=
|
||||||
|
github.com/mattn/go-isatty v0.0.16/go.mod h1:kYGgaQfpe5nmfYZH+SKPsOc2e4SrIfOl2e/yFXSvRLM=
|
||||||
|
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.0.6 h1:nrzqCb7j9cDFj2coyLNLaZuJTLjWjlaz6nvTvIwycIU=
|
||||||
|
github.com/pelletier/go-toml/v2 v2.0.6/go.mod h1:eumQOmlWiOPt5WriQQqoM5y18pDHwha2N+QD+EUNTek=
|
||||||
|
github.com/pkg/diff v0.0.0-20210226163009-20ebb0f2a09e/go.mod h1:pJLUxLENpZxwdsKMEsNbx1VGcRFpLqf3715MtcvvzbA=
|
||||||
|
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
|
||||||
|
github.com/rogpeppe/go-internal v1.6.1/go.mod h1:xXDCJY+GAPziupqXw64V24skbSoqbTEfhy4qGm1nDQc=
|
||||||
|
github.com/rogpeppe/go-internal v1.8.0/go.mod h1:WmiCO8CzOY8rg0OYDC4/i/2WRWAB6poM+XZ2dLUbcbE=
|
||||||
|
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.6.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
|
||||||
|
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/ugorji/go v1.2.7 h1:qYhyWUUd6WbiM+C6JZAUkIJt/1WrjzNHY9+KCIjVqTo=
|
||||||
|
github.com/ugorji/go v1.2.7/go.mod h1:nF9osbDWLy6bDVv/Rtoh6QgnvNDpmCalQV5urGCCS6M=
|
||||||
|
github.com/ugorji/go/codec v1.2.7 h1:YPXUKf7fYbp/y8xloBqZOw2qaVggbfwMlI8WM3wZUJ0=
|
||||||
|
github.com/ugorji/go/codec v1.2.7/go.mod h1:WGN1fab3R1fzQlVQTkfxVtIBhWDRqOviHU95kRgeqEY=
|
||||||
|
golang.org/x/crypto v0.0.0-20211215153901-e495a2d5b3d3/go.mod h1:IxCIyHEi3zRg3s0A5j5BB6A9Jmi73HwBIUl50j+osU4=
|
||||||
|
golang.org/x/crypto v0.3.0 h1:a06MkbcxBrEFc0w0QIZWXrH/9cCX6KJyWbBOIwAn+7A=
|
||||||
|
golang.org/x/crypto v0.3.0/go.mod h1:hebNnKkNXi2UzZN1eVRvBB7co0a+JxK6XbPiWVs/3J4=
|
||||||
|
golang.org/x/net v0.0.0-20210916014120-12bc252f5db8/go.mod h1:9nx3DQGgdP8bBQD5qxJ1jj9UTztislL4KSBs9R2vV5Y=
|
||||||
|
golang.org/x/net v0.0.0-20211029224645-99673261e6eb/go.mod h1:9nx3DQGgdP8bBQD5qxJ1jj9UTztislL4KSBs9R2vV5Y=
|
||||||
|
golang.org/x/net v0.0.0-20211112202133-69e39bad7dc2/go.mod h1:9nx3DQGgdP8bBQD5qxJ1jj9UTztislL4KSBs9R2vV5Y=
|
||||||
|
golang.org/x/net v0.2.0 h1:sZfSu1wtKLGlWI4ZZayP0ck9Y73K1ynO6gqzTdBVdPU=
|
||||||
|
golang.org/x/net v0.2.0/go.mod h1:KqCZLdyyvdV855qA2rE3GC2aiw5xGR5TEjj8smXukLY=
|
||||||
|
golang.org/x/sys v0.0.0-20201119102817-f84b799fce68/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
||||||
|
golang.org/x/sys v0.0.0-20210423082822-04245dca01da/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
||||||
|
golang.org/x/sys v0.0.0-20210615035016-665e8c7367d1/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||||
|
golang.org/x/sys v0.0.0-20210806184541-e5e7981a1069/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||||
|
golang.org/x/sys v0.0.0-20220811171246-fbc7d0a398ab/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||||
|
golang.org/x/sys v0.2.0 h1:ljd4t30dBnAvMZaQCevtY0xLLD0A+bRZXbgLMLU1F/A=
|
||||||
|
golang.org/x/sys v0.2.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||||
|
golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo=
|
||||||
|
golang.org/x/text v0.3.6/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=
|
||||||
|
golang.org/x/text v0.3.7/go.mod h1:u+2+/6zg+i71rQMx5EYifcz6MCKuco9NR6JIITiCfzQ=
|
||||||
|
golang.org/x/text v0.4.0 h1:BrVqGRd7+k1DiOgtnFvAkoQEWQvBc25ouMJM6429SFg=
|
||||||
|
golang.org/x/text v0.4.0/go.mod h1:mrYo+phRRbMaCq/xk9113O4dZlRixOauAjOtrjsXDZ8=
|
||||||
|
golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
|
||||||
|
golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
|
||||||
|
google.golang.org/protobuf v1.26.0-rc.1/go.mod h1:jlhhOSvTdKEhbULTjvd4ARK9grFBp09yW+WbY/TyQbw=
|
||||||
|
google.golang.org/protobuf v1.28.1 h1:d0NfwRgPtno5B1Wa6L2DAG+KivqkdutMf1UhdNx175w=
|
||||||
|
google.golang.org/protobuf v1.28.1/go.mod h1:HV8QOd/L58Z+nl8r43ehVNZIU/HEI6OcFqwMG9pJV4I=
|
||||||
|
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
|
||||||
|
gopkg.in/check.v1 v1.0.0-20180628173108-788fd7840127/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
|
||||||
|
gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c/go.mod h1:JHkPIbrfpd72SG/EVd6muEfDQjcINNoR0C8j2r3qZ4Q=
|
||||||
|
gopkg.in/errgo.v2 v2.1.0/go.mod h1:hNsd1EY+bozCKY1Ytp96fpM3vjJbqLJn88ws8XvfDNI=
|
||||||
|
gopkg.in/yaml.v2 v2.4.0 h1:D8xgwECY7CYvx+Y2n4sBz93Jn9JRvxdiyyo8CTfuKaY=
|
||||||
|
gopkg.in/yaml.v2 v2.4.0/go.mod h1:RDklbk79AGWmwhnvt/jBztapEOGDOx6ZbXqjP6csGnQ=
|
||||||
|
gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
||||||
|
gopkg.in/yaml.v3 v3.0.0-20210107192922-496545a6307b/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
||||||
|
gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
||||||
|
gorm.io/driver/mysql v1.4.4 h1:MX0K9Qvy0Na4o7qSC/YI7XxqUw5KDw01umqgID+svdQ=
|
||||||
|
gorm.io/driver/mysql v1.4.4/go.mod h1:BCg8cKI+R0j/rZRQxeKis/forqRwRSYOR8OM3Wo6hOM=
|
||||||
|
gorm.io/gorm v1.23.8/go.mod h1:l2lP/RyAtc1ynaTjFksBde/O8v9oOGIApu2/xRitmZk=
|
||||||
|
gorm.io/gorm v1.24.2 h1:9wR6CFD+G8nOusLdvkZelOEhpJVwwHzpQOUM+REd6U0=
|
||||||
|
gorm.io/gorm v1.24.2/go.mod h1:DVrVomtaYTbqs7gB/x2uVvqnXzv0nqjB396B8cG4dBA=
|
||||||
35
main.go
Normal file
35
main.go
Normal file
@ -0,0 +1,35 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
nice "github.com/ekyoung/gin-nice-recovery"
|
||||||
|
"github.com/gin-gonic/gin"
|
||||||
|
"integrity-checkin/dao"
|
||||||
|
"integrity-checkin/pkg/app"
|
||||||
|
"integrity-checkin/pkg/httpclient"
|
||||||
|
"integrity-checkin/router"
|
||||||
|
"integrity-checkin/service"
|
||||||
|
"math/rand"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
Setup()
|
||||||
|
httpclient.Setup()
|
||||||
|
engine := gin.New()
|
||||||
|
engine.Use(gin.Logger(),
|
||||||
|
nice.Recovery(func(c *gin.Context, err interface{}) {
|
||||||
|
aw := app.NewWrapper(c)
|
||||||
|
aw.Error("内部错误，请重试：" + fmt.Sprintf("%v", err)).SendJSON()
|
||||||
|
}))
|
||||||
|
dao.Setup()
|
||||||
|
service.Setup()
|
||||||
|
router.Setup(engine)
|
||||||
|
err := engine.Run(fmt.Sprintf(":%v", "8964"))
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
func Setup() {
|
||||||
|
rand.Seed(time.Now().UnixNano())
|
||||||
|
}
|
||||||
17
middleware/cors.go
Normal file
17
middleware/cors.go
Normal file
@ -0,0 +1,17 @@
|
|||||||
|
package middleware
|
||||||
|
|
||||||
|
import "github.com/gin-gonic/gin"
|
||||||
|
|
||||||
|
func Cors() 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", "*")
|
||||||
|
c.Writer.Header().Set("Access-Control-Allow-Methods", "GET, HEAD, POST, PUT, DELETE, OPTIONS, PATCH")
|
||||||
|
if c.Request.Method == "OPTIONS" {
|
||||||
|
c.AbortWithStatus(204)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.Next()
|
||||||
|
}
|
||||||
|
}
|
||||||
31
middleware/jwt.go
Normal file
31
middleware/jwt.go
Normal file
@ -0,0 +1,31 @@
|
|||||||
|
package middleware
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"github.com/gin-gonic/gin"
|
||||||
|
"integrity-checkin/pkg/app"
|
||||||
|
"integrity-checkin/pkg/util"
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
|
func JWT(minRoleID int) gin.HandlerFunc {
|
||||||
|
return func(c *gin.Context) {
|
||||||
|
aw := app.NewWrapper(c)
|
||||||
|
authHeader := c.GetHeader("Authorization")
|
||||||
|
if strings.HasPrefix(authHeader, "Bearer ") {
|
||||||
|
authHeader = authHeader[7:]
|
||||||
|
}
|
||||||
|
claims, err := util.ParseJWT(authHeader)
|
||||||
|
if err != nil {
|
||||||
|
aw.Error("该接口需要登录").SendJSON()
|
||||||
|
aw.Ctx.Abort()
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if claims.RoleID < minRoleID {
|
||||||
|
aw.Error(fmt.Sprintf("该接口需要角色大于或等于：%v（请尝试退出后重登录）", minRoleID)).SendJSON()
|
||||||
|
aw.Ctx.Abort()
|
||||||
|
return
|
||||||
|
}
|
||||||
|
aw.Ctx.Set("userClaims", claims)
|
||||||
|
}
|
||||||
|
}
|
||||||
11
pkg/app/handler_func.go
Normal file
11
pkg/app/handler_func.go
Normal file
@ -0,0 +1,11 @@
|
|||||||
|
package app
|
||||||
|
|
||||||
|
import "github.com/gin-gonic/gin"
|
||||||
|
|
||||||
|
func HandlerFunc(handler func(aw *Wrapper) Result) func(c *gin.Context) {
|
||||||
|
return func(c *gin.Context) {
|
||||||
|
aw := NewWrapper(c)
|
||||||
|
res := handler(aw)
|
||||||
|
c.JSON(200, res)
|
||||||
|
}
|
||||||
|
}
|
||||||
72
pkg/app/wrapper.go
Normal file
72
pkg/app/wrapper.go
Normal file
@ -0,0 +1,72 @@
|
|||||||
|
package app
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/gin-gonic/gin"
|
||||||
|
"integrity-checkin/pkg/util"
|
||||||
|
)
|
||||||
|
|
||||||
|
type Result struct {
|
||||||
|
Code int `json:"code"`
|
||||||
|
Msg string `json:"msg,omitempty"`
|
||||||
|
Data interface{} `json:"data,omitempty"`
|
||||||
|
wrapper *Wrapper
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r Result) SendJSON() {
|
||||||
|
r.wrapper.Ctx.JSON(200, r)
|
||||||
|
}
|
||||||
|
|
||||||
|
type Wrapper struct {
|
||||||
|
Ctx *gin.Context
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewWrapper(c *gin.Context) *Wrapper {
|
||||||
|
return &Wrapper{Ctx: c}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (w Wrapper) OK() Result {
|
||||||
|
return Result{
|
||||||
|
Code: 0,
|
||||||
|
Msg: "",
|
||||||
|
Data: nil,
|
||||||
|
wrapper: &w,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
func (w Wrapper) Success(data interface{}) Result {
|
||||||
|
return Result{
|
||||||
|
Code: 0,
|
||||||
|
Msg: "",
|
||||||
|
Data: data,
|
||||||
|
wrapper: &w,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
func (w Wrapper) Error(msg string) Result {
|
||||||
|
return Result{
|
||||||
|
Code: -1,
|
||||||
|
Msg: msg,
|
||||||
|
Data: nil,
|
||||||
|
wrapper: &w,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
func (w Wrapper) ErrorWithCode(code int, msg string) Result {
|
||||||
|
return Result{
|
||||||
|
Code: code,
|
||||||
|
Msg: msg,
|
||||||
|
Data: nil,
|
||||||
|
wrapper: &w,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
func (w Wrapper) GetIP() string {
|
||||||
|
return w.Ctx.ClientIP()
|
||||||
|
}
|
||||||
|
func (w Wrapper) ExtractUserClaims() *util.UserClaims {
|
||||||
|
raw, exist := w.Ctx.Get("userClaims")
|
||||||
|
if !exist {
|
||||||
|
panic("userClaims not exists")
|
||||||
|
}
|
||||||
|
uc, ok := raw.(*util.UserClaims)
|
||||||
|
if !ok {
|
||||||
|
panic("userClaims failed to convert")
|
||||||
|
}
|
||||||
|
return uc
|
||||||
|
}
|
||||||
67
pkg/httpclient/resty.go
Normal file
67
pkg/httpclient/resty.go
Normal file
@ -0,0 +1,67 @@
|
|||||||
|
package httpclient
|
||||||
|
|
||||||
|
import (
|
||||||
|
"errors"
|
||||||
|
"github.com/go-resty/resty/v2"
|
||||||
|
"integrity-checkin/data/dto"
|
||||||
|
"regexp"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
var Client *resty.Client
|
||||||
|
|
||||||
|
func Setup() {
|
||||||
|
Client = resty.New().SetTimeout(10 * time.Second)
|
||||||
|
}
|
||||||
|
func SDULogin(username string, password string) (*dto.SDULoginData, error) {
|
||||||
|
resp, err := Client.R().SetFormData(map[string]string{
|
||||||
|
"username": username,
|
||||||
|
"password": password,
|
||||||
|
}).Post("https://pass.sdu.edu.cn/cas/restlet/tickets")
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
result1 := string(resp.Body())
|
||||||
|
if resp.StatusCode()/100 != 2 {
|
||||||
|
return nil, errors.New(result1)
|
||||||
|
}
|
||||||
|
resp, err = Client.R().SetBody("service=https://service.sdu.edu.cn/tp_up/view?m=up").
|
||||||
|
Post("https://pass.sdu.edu.cn/cas/restlet/tickets/" + result1)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
result2 := string(resp.Body())
|
||||||
|
if resp.StatusCode()/100 != 2 {
|
||||||
|
return nil, errors.New(result2)
|
||||||
|
}
|
||||||
|
resp, err = Client.R().SetQueryParams(map[string]string{
|
||||||
|
"ticket": result2,
|
||||||
|
"service": "https://service.sdu.edu.cn/tp_up/view?m=up",
|
||||||
|
}).Get("https://pass.sdu.edu.cn/cas/serviceValidate")
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
result3 := string(resp.Body())
|
||||||
|
if resp.StatusCode()/100 != 2 {
|
||||||
|
return nil, errors.New(result3)
|
||||||
|
}
|
||||||
|
realName := ""
|
||||||
|
username = ""
|
||||||
|
if arr := regexp.MustCompile("<cas:USER_NAME>(.*?)<").FindStringSubmatch(result3); len(arr) >= 2 {
|
||||||
|
realName = arr[1]
|
||||||
|
} else {
|
||||||
|
return nil, errors.New(result3)
|
||||||
|
}
|
||||||
|
if arr := regexp.MustCompile("<cas:ID_NUMBER>(.*?)<").FindStringSubmatch(result3); len(arr) >= 2 {
|
||||||
|
username = arr[1]
|
||||||
|
} else {
|
||||||
|
return nil, errors.New(result3)
|
||||||
|
}
|
||||||
|
if realName == "" || username == "" {
|
||||||
|
return nil, errors.New(result3)
|
||||||
|
}
|
||||||
|
return &dto.SDULoginData{
|
||||||
|
RealName: realName,
|
||||||
|
Username: username,
|
||||||
|
}, nil
|
||||||
|
}
|
||||||
42
pkg/util/permission.go
Normal file
42
pkg/util/permission.go
Normal file
@ -0,0 +1,42 @@
|
|||||||
|
package util
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/golang-jwt/jwt"
|
||||||
|
)
|
||||||
|
|
||||||
|
type UserClaims struct {
|
||||||
|
UserID int `json:"user_id"`
|
||||||
|
RoleID int `json:"role_id"`
|
||||||
|
jwt.StandardClaims
|
||||||
|
}
|
||||||
|
|
||||||
|
const (
|
||||||
|
JWTSecret = "fee516aa-8f5b-4248-a80c-bf467d40fd62"
|
||||||
|
)
|
||||||
|
|
||||||
|
func ParseJWT(token string) (*UserClaims, error) {
|
||||||
|
tokenClaims, err := jwt.ParseWithClaims(token, &UserClaims{}, func(token *jwt.Token) (interface{}, error) {
|
||||||
|
return []byte(JWTSecret), nil
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if tokenClaims != nil {
|
||||||
|
if claims, ok := tokenClaims.Claims.(*UserClaims); ok && tokenClaims.Valid {
|
||||||
|
return claims, nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
func GenerateJWT(userID int, roleID int) string {
|
||||||
|
token := jwt.NewWithClaims(jwt.SigningMethodHS256, UserClaims{
|
||||||
|
UserID: userID,
|
||||||
|
RoleID: roleID,
|
||||||
|
StandardClaims: jwt.StandardClaims{},
|
||||||
|
})
|
||||||
|
str, err := token.SignedString([]byte(JWTSecret))
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
return str
|
||||||
|
}
|
||||||
27
router/router.go
Normal file
27
router/router.go
Normal file
@ -0,0 +1,27 @@
|
|||||||
|
package router
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/gin-gonic/gin"
|
||||||
|
"integrity-checkin/middleware"
|
||||||
|
"integrity-checkin/pkg/app"
|
||||||
|
"integrity-checkin/service"
|
||||||
|
)
|
||||||
|
|
||||||
|
func Setup(engine *gin.Engine) {
|
||||||
|
user := engine.Group("/user")
|
||||||
|
{
|
||||||
|
hub := service.ExUserService
|
||||||
|
// 测试方法，获取signed jwt
|
||||||
|
user.POST("/login", app.HandlerFunc(hub.Login))
|
||||||
|
}
|
||||||
|
checkin := engine.Group("/checkin")
|
||||||
|
{
|
||||||
|
hub := service.ExCheckinService
|
||||||
|
checkin.GET("/list_activities", app.HandlerFunc(hub.ListActivities))
|
||||||
|
checkin.Use(middleware.JWT(1))
|
||||||
|
checkin.POST("/submit", app.HandlerFunc(hub.Submit))
|
||||||
|
checkin.POST("/set_status", middleware.JWT(2), app.HandlerFunc(hub.SetStatus))
|
||||||
|
checkin.GET("/get_user_activity", middleware.JWT(2), app.HandlerFunc(hub.GetUserActivity))
|
||||||
|
checkin.GET("/list_user_activities", middleware.JWT(2), app.HandlerFunc(hub.ListUserActivities))
|
||||||
|
}
|
||||||
|
}
|
||||||
73
service/checkin.go
Normal file
73
service/checkin.go
Normal file
@ -0,0 +1,73 @@
|
|||||||
|
package service
|
||||||
|
|
||||||
|
import (
|
||||||
|
"integrity-checkin/data/po"
|
||||||
|
"integrity-checkin/pkg/app"
|
||||||
|
"strconv"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
type CheckinService struct {
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c CheckinService) Submit(aw *app.Wrapper) app.Result {
|
||||||
|
uc := aw.ExtractUserClaims()
|
||||||
|
var req po.UserActivityForm
|
||||||
|
if err := aw.Ctx.ShouldBind(&req); err != nil {
|
||||||
|
return aw.Error(err.Error())
|
||||||
|
}
|
||||||
|
if err := req.Validate(); err != nil {
|
||||||
|
return aw.Error(err.Error())
|
||||||
|
}
|
||||||
|
today := time.Now().Format("20060102")
|
||||||
|
if checkinDAO.FindUserActivityByDay(uc.UserID, req.ActivityID, today) != nil {
|
||||||
|
return aw.Error("您今天已参加过该项目，请明天再来")
|
||||||
|
}
|
||||||
|
userActivity, err := checkinDAO.CreateCheckin(uc.UserID, req.ActivityID, req.Image)
|
||||||
|
if err != nil {
|
||||||
|
return aw.Error(err.Error())
|
||||||
|
}
|
||||||
|
return aw.Success(userActivity)
|
||||||
|
}
|
||||||
|
func (c CheckinService) SetStatus(aw *app.Wrapper) app.Result {
|
||||||
|
type SetStatusReq struct {
|
||||||
|
ID int `form:"id" binding:"required"`
|
||||||
|
Status int `form:"status" binding:"required"`
|
||||||
|
Credit int `form:"credit" binding:"required,gte=1"`
|
||||||
|
}
|
||||||
|
var req SetStatusReq
|
||||||
|
if err := aw.Ctx.ShouldBind(&req); err != nil {
|
||||||
|
return aw.Error(err.Error())
|
||||||
|
}
|
||||||
|
userActivity := checkinDAO.FindUserActivityByID(req.ID)
|
||||||
|
if userActivity == nil {
|
||||||
|
return aw.Error("打卡不存在")
|
||||||
|
}
|
||||||
|
if userActivity.Status != 1 {
|
||||||
|
return aw.Error("已经审核过此打卡")
|
||||||
|
}
|
||||||
|
if req.Status == 1 {
|
||||||
|
return aw.Error("只能是 2通过，3拒绝")
|
||||||
|
}
|
||||||
|
if req.Status == 2 { // pass
|
||||||
|
user := userDAO.FindUserByID(userActivity.UserID)
|
||||||
|
user.Credit += req.Credit
|
||||||
|
userDAO.SaveUser(user)
|
||||||
|
}
|
||||||
|
userActivity.Status = req.Status
|
||||||
|
checkinDAO.SaveUserActivity(userActivity)
|
||||||
|
return aw.Success(userActivity)
|
||||||
|
}
|
||||||
|
func (c CheckinService) ListActivities(aw *app.Wrapper) app.Result {
|
||||||
|
return aw.Success(checkinDAO.FindAllActivities())
|
||||||
|
}
|
||||||
|
func (c CheckinService) ListUserActivities(aw *app.Wrapper) app.Result {
|
||||||
|
return aw.Success(checkinDAO.FindAllUserActivities())
|
||||||
|
}
|
||||||
|
func (c CheckinService) GetUserActivity(aw *app.Wrapper) app.Result {
|
||||||
|
id, err := strconv.Atoi(aw.Ctx.Query("id"))
|
||||||
|
if err != nil {
|
||||||
|
return aw.Error(err.Error())
|
||||||
|
}
|
||||||
|
return aw.Success(checkinDAO.FindUserActivityByID(id))
|
||||||
|
}
|
||||||
21
service/service.go
Normal file
21
service/service.go
Normal file
@ -0,0 +1,21 @@
|
|||||||
|
package service
|
||||||
|
|
||||||
|
import "integrity-checkin/dao"
|
||||||
|
|
||||||
|
var (
|
||||||
|
userDAO dao.UserDAO
|
||||||
|
checkinDAO dao.CheckinDAO
|
||||||
|
)
|
||||||
|
|
||||||
|
var (
|
||||||
|
ExUserService UserService
|
||||||
|
ExCheckinService CheckinService
|
||||||
|
)
|
||||||
|
|
||||||
|
func Setup() {
|
||||||
|
userDAO = dao.UserDAO{Tx: dao.DB}
|
||||||
|
checkinDAO = dao.CheckinDAO{Tx: dao.DB}
|
||||||
|
|
||||||
|
ExUserService = UserService{}
|
||||||
|
ExCheckinService = CheckinService{}
|
||||||
|
}
|
||||||
30
service/user.go
Normal file
30
service/user.go
Normal file
@ -0,0 +1,30 @@
|
|||||||
|
package service
|
||||||
|
|
||||||
|
import (
|
||||||
|
"integrity-checkin/pkg/app"
|
||||||
|
"integrity-checkin/pkg/httpclient"
|
||||||
|
"integrity-checkin/pkg/util"
|
||||||
|
)
|
||||||
|
|
||||||
|
type UserService struct {
|
||||||
|
}
|
||||||
|
|
||||||
|
func (u UserService) Login(aw *app.Wrapper) app.Result {
|
||||||
|
type LoginReq struct {
|
||||||
|
Username string `form:"username" binding:"required"`
|
||||||
|
Password string `form:"password" binding:"required"`
|
||||||
|
}
|
||||||
|
var req LoginReq
|
||||||
|
if err := aw.Ctx.ShouldBind(&req); err != nil {
|
||||||
|
return aw.Error(err.Error())
|
||||||
|
}
|
||||||
|
sduLoginData, err := httpclient.SDULogin(req.Username, req.Password)
|
||||||
|
if err != nil {
|
||||||
|
return aw.Error(err.Error())
|
||||||
|
}
|
||||||
|
user := userDAO.FindUserByUsername(sduLoginData.Username)
|
||||||
|
if user == nil {
|
||||||
|
user = userDAO.CreateUser(sduLoginData.Username, sduLoginData.RealName)
|
||||||
|
}
|
||||||
|
return aw.Success(util.GenerateJWT(user.ID, user.RoleID))
|
||||||
|
}
|
||||||
載入中…
x
新增問題並參考
Block a user