TRKJ/EdgeAPI
2
0
Fork 0
mirror of https://github.com/TeaOSLab/EdgeAPI.git synced 2025-11-03 15:00:27 +08:00
Code Issues Packages Projects Releases Wiki Activity

将HTTP Header中Edge-改成X-Edge-

Browse Source
This commit is contained in:
GoEdgeLab
2021-10-19 19:49:06 +08:00
parent 686e63c0cd
commit c7214765ad
1 changed files with 17 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
internal/nodes/rest_server.go
View File

@@ -45,7 +45,13 @@ func (this *RestServer) handle(writer http.ResponseWriter, req *http.Request) {
path := req.URL.Path
// 是否显示Pretty后的JSON
shouldPretty := req.Header.Get("Edge-Response-Pretty") == "on"
shouldPretty := req.Header.Get("X-Edge-Response-Pretty") == "on"
// 兼容老的Header
var oldShouldPretty = req.Header.Get("Edge-Response-Pretty")
if len(oldShouldPretty) > 0 {
shouldPretty = oldShouldPretty == "on"
}
// 欢迎页
if path == "/" {
@@ -98,14 +104,17 @@ func (this *RestServer) handle(writer http.ResponseWriter, req *http.Request) {
if serviceName != "APIAccessTokenService" || (methodName != "GetAPIAccessToken" && methodName != "getAPIAccessToken") {
// 校验TOKEN
token := req.Header.Get("Edge-Access-Token")
token := req.Header.Get("X-Edge-Access-Token")
if len(token) == 0 {
this.writeJSON(writer, maps.Map{
"code": 400,
"data": maps.Map{},
"message": "require 'Edge-Access-Token' header",
}, shouldPretty)
return
token = req.Header.Get("Edge-Access-Token")
if len(token) == 0 {
this.writeJSON(writer, maps.Map{
"code": 400,
"data": maps.Map{},
"message": "require 'X-Edge-Access-Token' header",
}, shouldPretty)
return
}
}
accessToken, err := models.SharedAPIAccessTokenDAO.FindAccessToken(nil, token)