ZiHangQin/electricity_bill_calc_service
1
0
Fork 0
forked from free-lancers/electricity_bill_calc_service
Code Pull Requests Activity

enahnce(security):改进对于令牌的获取。

Browse Source
This commit is contained in:
徐涛 2023-06-11 17:35:50 +08:00
parent 5d938da53e
commit c2a56c8253
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
security/security.go
View File

@ -14,7 +14,11 @@ import (
// ! 仅通过该中间件是不能保证上下文中一定保存有用户会话信息的。 // ! 仅通过该中间件是不能保证上下文中一定保存有用户会话信息的。
func SessionRecovery(c *fiber.Ctx) error { func SessionRecovery(c *fiber.Ctx) error {
if auth := c.Get("Authorization", ""); len(auth) > 0 { if auth := c.Get("Authorization", ""); len(auth) > 0 {
token := strings.Fields(auth)[1] authFields := strings.Fields(auth)
if len(authFields) != 2 || strings.ToLower(authFields[0]) != "bearer" || len(authFields[1]) == 0 {
return c.Next()
}
token := authFields[1]
session, err := cache.RetrieveSession(token) session, err := cache.RetrieveSession(token)
if err == nil && session != nil { if err == nil && session != nil {