diff --git a/dao/user.go b/dao/user.go index 0ec58d9..486d034 100644 --- a/dao/user.go +++ b/dao/user.go @@ -17,6 +17,7 @@ type User struct { Redis bool `gorm:"column:redis"` Run bool `gorm:"column:run"` Upload bool `gorm:"column:upload"` + Avatar string `gorm:"column:avatar"` CreateTime string `gorm:"column:create_time"` UpdateTime string `gorm:"column:update_time"` } @@ -75,3 +76,7 @@ func FindUserByEmail(email string) User { func UpdateUserByID(id int, name, password, email string) { DB.Model(&User{}).Where("id = ?", id).Updates(User{Name: name, Password: password, Email: email}) } + +func UpdateUserByID2(id int, req proto.UpdateUserInfoReq) { + DB.Debug().Model(&User{}).Where("id = ?", id).Updates(User{Name: req.Username, Age: req.Age, Role: req.Role, Run: req.Run, Redis: req.Redis, Upload: req.Upload, Avatar: req.Avatar, Gender: req.Gender}) +} diff --git a/handler/user.go b/handler/user.go index bc53b8d..11023d0 100644 --- a/handler/user.go +++ b/handler/user.go @@ -25,6 +25,7 @@ func SetUpUserGroup(router *gin.Engine) { userGroup.POST("/confirm", ConfirmQRLogin) userGroup.POST("/search", SearchHandler) userGroup.POST("/info", GetUserInfo) + userGroup.POST("/update", UpdateUserInfo) } type RLReq struct { @@ -45,13 +46,53 @@ type SearchReq struct { Keyword string `json:"keyword" form:"keyword"` ID int `json:"id" form:"id"` } +type GetUserInfoReq struct { + ID int `json:"id" form:"id"` +} func GetUserInfo(c *gin.Context) { + var req_data GetUserInfoReq id, _ := c.Get("id") user_id := int(id.(float64)) - user := dao.FindUserByID2(user_id) - user.Password = "" //不返回密码 - c.JSON(200, gin.H{"code": proto.SuccessCode, "message": "success", "data": user}) + if err := c.ShouldBind(&req_data); err == nil { + var user dao.User + if req_data.ID == user_id { + user = dao.FindUserByID2(user_id) + user.Password = "" //不返回密码 + } else { + //判断当前用户是否有权限查看 + cur_user := dao.FindUserByID2(user_id) + if cur_user.Role == "admin" { + user = dao.FindUserByID2(req_data.ID) + user.Password = "" //不返回密码 + } else { + c.JSON(200, gin.H{"code": proto.PermissionDenied, "message": "无权查看", "data": "2"}) + return + } + } + c.JSON(200, gin.H{"code": proto.SuccessCode, "message": "success", "data": user}) + } else { + c.JSON(200, gin.H{"code": proto.ParameterError, "message": err, "data": "2"}) + return + } +} + +func UpdateUserInfo(c *gin.Context) { + var req_data proto.UpdateUserInfoReq + id, _ := c.Get("id") + user_id := int(id.(float64)) + if err := c.ShouldBind(&req_data); err == nil { + rid, err2 := service.UpdateUser(user_id, req_data) + if err2 != nil { + c.JSON(200, gin.H{"code": proto.OperationFailed, "message": "failed", "data": "2"}) + return + } + c.JSON(200, gin.H{"code": proto.SuccessCode, "message": "success", "data": rid}) + } else { + c.JSON(200, gin.H{"code": proto.ParameterError, "message": err, "data": "2"}) + return + } + } func GetScanUUID(c *gin.Context) { diff --git a/main.go b/main.go index 52466a9..48c4966 100644 --- a/main.go +++ b/main.go @@ -86,9 +86,11 @@ func JWTAuthMiddleware() gin.HandlerFunc { } //如果请求为login或register,则不需要验证token - if strings.Contains(c.Request.URL.Path, "/login") || strings.Contains(c.Request.URL.Path, "/register") || strings.Contains(c.Request.URL.Path, "/uuid") || strings.Contains(c.Request.URL.Path, "/gqr") || strings.Contains(c.Request.URL.Path, "/cid/callback") { - c.Next() - return + for k, _ := range proto.Url_map { + if strings.Contains(c.Request.URL.Path, k) { + c.Next() + return + } } if tokenString == "" { //c.AbortWithStatus(200) diff --git a/proto/conf.go b/proto/conf.go index 427fa4b..15b5053 100644 --- a/proto/conf.go +++ b/proto/conf.go @@ -9,6 +9,7 @@ import ( var Config ConfigStruct var SigningKey = []byte{} +var Url_map = map[string]bool{"/login": true, "/register": true, "/uuid": true, "/gqr": true, "/cid/callback": true} const ( MYSQL_USER = "video_t2" diff --git a/proto/status.go b/proto/status.go index 6554ee8..b259edf 100644 --- a/proto/status.go +++ b/proto/status.go @@ -18,8 +18,9 @@ const ( TokenParseError = 19 // Token解析错误 // 用户名密码相关错误码 - UsernameOrPasswordError = 6 // 用户名或密码错误 - UsernameExists = 7 // 用户名已存在 + UsernameOrPasswordError = 6 // 用户名或密码错误 + UsernameExists = 7 // 用户名已存在 + PermissionDenied = 21 // 权限不足 // Redis相关错误码 RedisSetError = 8 // 设置redis错误 diff --git a/proto/user_req.go b/proto/user_req.go new file mode 100644 index 0000000..be15ea6 --- /dev/null +++ b/proto/user_req.go @@ -0,0 +1,13 @@ +package proto + +type UpdateUserInfoReq struct { + ID int `json:"id" form:"id"` //用户id + Username string `json:"username" form:"username"` //用户名 + Age int `json:"age" form:"age"` //年龄 + Role string `json:"role" form:"role"` //角色 + Gender string `json:"gender" form:"gender"` //性别 + Redis bool `json:"redis" form:"redis"` //是否刷新redis + Upload bool `json:"upload" form:"upload"` //是否上传头像 + Run bool `json:"run" form:"run"` //是否运行 + Avatar string `json:"avatar" form:"avatar"` //头像 +} diff --git a/service/userService.go b/service/userService.go index 3ce3274..d090431 100644 --- a/service/userService.go +++ b/service/userService.go @@ -41,3 +41,16 @@ func GetUserByID(id int) []proto.User { func GetUserByNameLike(name string) []proto.User { return dao.FindUserByNameLike(name) } + +func UpdateUser(user_id int, req proto.UpdateUserInfoReq) (int, error) { + cur_user := dao.FindUserByID2(user_id) + if user_id == req.ID { + dao.UpdateUserByID2(user_id, req) + return user_id, nil + } else if cur_user.Role == "admin" { + dao.UpdateUserByID2(req.ID, req) + return req.ID, nil + } else { + return 0, nil + } +} diff --git a/worker/redis.go b/worker/redis.go index 709790c..b716737 100644 --- a/worker/redis.go +++ b/worker/redis.go @@ -346,3 +346,29 @@ func GetRedisBitmap(key string, offset int64) int { } return int(val) } + +// 发布订阅者模式-发布消息 +func Publish(channel string, message string, expire time.Duration) { + ctx := context.Background() + err := redisClient.Publish(ctx, channel, message).Err() + if err != nil { + fmt.Println("Error publishing message: %v", err) + } + err = redisClient.Expire(ctx, channel, expire).Err() + if err != nil { + fmt.Println("Error setting key: %v", err) + } +} + +// 发布订阅者模式-订阅消息 +func Subscribe(channel string) []string { + ctx := context.Background() + pubsub := redisClient.Subscribe(ctx, channel) + ch := pubsub.Channel() + defer pubsub.Close() + var messages []string + for msg := range ch { + messages = append(messages, msg.Payload) + } + return messages +}