yml/hfb_sys
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

SSE 推送替代轮训

Browse Source
This commit is contained in:
yml2213
2026-05-27 06:18:55 +08:00
parent 1458fc279c
commit a002987784
9 changed files with 493 additions and 42 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/internal/middleware/auth.go
+13 -6
View File
@@ -16,11 +16,19 @@ const (
ContextUsername = "username"
)
func extractToken(c *gin.Context) string {
header := c.GetHeader("Authorization")
tokenText := strings.TrimSpace(strings.TrimPrefix(header, "Bearer "))
if tokenText != "" && tokenText != header {
return tokenText
}
return c.Query("token")
}
func Auth(jwtManager *auth.JWTManager) gin.HandlerFunc {
return func(c *gin.Context) {
header := c.GetHeader("Authorization")
tokenText := strings.TrimSpace(strings.TrimPrefix(header, "Bearer "))
if tokenText == "" || tokenText == header {
tokenText := extractToken(c)
if tokenText == "" {
response.Unauthorized(c, "缺少访问令牌")
c.Abort()
return
@@ -41,9 +49,8 @@ func Auth(jwtManager *auth.JWTManager) gin.HandlerFunc {
func AdminAuth(jwtManager *auth.JWTManager) gin.HandlerFunc {
return func(c *gin.Context) {
header := c.GetHeader("Authorization")
tokenText := strings.TrimSpace(strings.TrimPrefix(header, "Bearer "))
if tokenText == "" || tokenText == header {
tokenText := extractToken(c)
if tokenText == "" {
response.Unauthorized(c, "缺少后台访问令牌")
c.Abort()
return