@@ -80,69 +80,66 @@ func Login(c *gin.Context) {
8080
8181}
8282
83- //登录以后签发jwt
83+ // 登录以后签发jwt
8484func tokenNext (c * gin.Context , user model.SysUser ) {
8585 j := & middleware.JWT {
86- []byte (global .GVA_CONFIG .JWT .SigningKey ), // 唯一签名
86+ SigningKey : []byte (global .GVA_CONFIG .JWT .SigningKey ), // 唯一签名
8787 }
8888 clams := request.CustomClaims {
8989 UUID : user .UUID ,
9090 ID : user .ID ,
9191 NickName : user .NickName ,
9292 AuthorityId : user .AuthorityId ,
9393 StandardClaims : jwt.StandardClaims {
94- NotBefore : int64 ( time .Now ().Unix () - 1000 ) , // 签名生效时间
95- ExpiresAt : int64 ( time .Now ().Unix () + 60 * 60 * 24 * 7 ) , // 过期时间 一周
96- Issuer : "qmPlus" , // 签名的发行者
94+ NotBefore : time .Now ().Unix () - 1000 , // 签名生效时间
95+ ExpiresAt : time .Now ().Unix () + 60 * 60 * 24 * 7 , // 过期时间 一周
96+ Issuer : "qmPlus" , // 签名的发行者
9797 },
9898 }
9999 token , err := j .CreateToken (clams )
100100 if err != nil {
101101 response .FailWithMessage ("获取token失败" , c )
102+ return
103+ }
104+ if ! global .GVA_CONFIG .System .UseMultipoint {
105+ response .OkWithData (resp.LoginResponse {
106+ User : user ,
107+ Token : token ,
108+ ExpiresAt : clams .StandardClaims .ExpiresAt * 1000 ,
109+ }, c )
110+ return
111+ }
112+ var loginJwt model.JwtBlacklist
113+ loginJwt .Jwt = token
114+ err , jwtStr := service .GetRedisJWT (user .Username )
115+ if err == redis .Nil {
116+ if err := service .SetRedisJWT (loginJwt , user .Username ); err != nil {
117+ response .FailWithMessage ("设置登录状态失败" , c )
118+ return
119+ }
120+ response .OkWithData (resp.LoginResponse {
121+ User : user ,
122+ Token : token ,
123+ ExpiresAt : clams .StandardClaims .ExpiresAt * 1000 ,
124+ }, c )
125+ } else if err != nil {
126+ response .FailWithMessage (fmt .Sprintf ("%v" , err ), c )
102127 } else {
103- if global .GVA_CONFIG .System .UseMultipoint {
104- var loginJwt model.JwtBlacklist
105- loginJwt .Jwt = token
106- err , jwtStr := service .GetRedisJWT (user .Username )
107- if err == redis .Nil {
108- err2 := service .SetRedisJWT (loginJwt , user .Username )
109- if err2 != nil {
110- response .FailWithMessage ("设置登录状态失败" , c )
111- } else {
112- response .OkWithData (resp.LoginResponse {
113- User : user ,
114- Token : token ,
115- ExpiresAt : clams .StandardClaims .ExpiresAt * 1000 ,
116- }, c )
117- }
118- } else if err != nil {
119- response .FailWithMessage (fmt .Sprintf ("%v" , err ), c )
120- } else {
121- var blackJWT model.JwtBlacklist
122- blackJWT .Jwt = jwtStr
123- err3 := service .JsonInBlacklist (blackJWT )
124- if err3 != nil {
125- response .FailWithMessage ("jwt作废失败" , c )
126- } else {
127- err2 := service .SetRedisJWT (loginJwt , user .Username )
128- if err2 != nil {
129- response .FailWithMessage ("设置登录状态失败" , c )
130- } else {
131- response .OkWithData (resp.LoginResponse {
132- User : user ,
133- Token : token ,
134- ExpiresAt : clams .StandardClaims .ExpiresAt * 1000 ,
135- }, c )
136- }
137- }
138- }
139- } else {
140- response .OkWithData (resp.LoginResponse {
141- User : user ,
142- Token : token ,
143- ExpiresAt : clams .StandardClaims .ExpiresAt * 1000 ,
144- }, c )
128+ var blackJWT model.JwtBlacklist
129+ blackJWT .Jwt = jwtStr
130+ if err := service .JsonInBlacklist (blackJWT ); err != nil {
131+ response .FailWithMessage ("jwt作废失败" , c )
132+ return
133+ }
134+ if err := service .SetRedisJWT (loginJwt , user .Username ); err != nil {
135+ response .FailWithMessage ("设置登录状态失败" , c )
136+ return
145137 }
138+ response .OkWithData (resp.LoginResponse {
139+ User : user ,
140+ Token : token ,
141+ ExpiresAt : clams .StandardClaims .ExpiresAt * 1000 ,
142+ }, c )
146143 }
147144}
148145
@@ -189,21 +186,21 @@ type UserHeaderImg struct {
189186// @Router /user/uploadHeaderImg [post]
190187func UploadHeaderImg (c * gin.Context ) {
191188 claims , _ := c .Get ("claims" )
192- //获取头像文件
189+ // 获取头像文件
193190 // 这里我们通过断言获取 claims内的所有内容
194191 waitUse := claims .(* request.CustomClaims )
195192 uuid := waitUse .UUID
196193 _ , header , err := c .Request .FormFile ("headerImg" )
197- //便于找到用户 以后从jwt中取
194+ // 便于找到用户 以后从jwt中取
198195 if err != nil {
199196 response .FailWithMessage (fmt .Sprintf ("上传文件失败,%v" , err ), c )
200197 } else {
201- //文件上传后拿到文件路径
198+ // 文件上传后拿到文件路径
202199 err , filePath , _ := utils .Upload (header )
203200 if err != nil {
204201 response .FailWithMessage (fmt .Sprintf ("接收返回值失败,%v" , err ), c )
205202 } else {
206- //修改数据库后得到修改后的user并且返回供前端使用
203+ // 修改数据库后得到修改后的user并且返回供前端使用
207204 err , user := service .UploadHeaderImg (uuid , filePath )
208205 if err != nil {
209206 response .FailWithMessage (fmt .Sprintf ("修改数据库链接失败,%v" , err ), c )
0 commit comments