update 401

hungjiun · Jul 3, 2023 · c649b67 · c649b67
1 parent 93262dc
commit c649b67
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/route/v2.go b/route/v2.go
@@ -90,7 +90,10 @@ func InitV2Router() http.Handler {
 		},
 		TokenLookupFuncs: []echo_middleware.ValuesExtractor{
 			func(c echo.Context) ([]string, error) {
-				return []string{c.Request().Header.Get(echo.HeaderAuthorization)}, nil
+				if len(c.Request().Header.Get(echo.HeaderAuthorization)) > 0 {
+					return []string{c.Request().Header.Get(echo.HeaderAuthorization)}, nil
+				}
+				return []string{c.QueryParam("token")}, nil
 			},
 		},
 	}))