fanhuadesenlinnn/cloudpods
1
0
Fork 0
mirror of https://hubproxy.babadafafafafa.cn/https://github.com/yunionio/cloudpods.git synced 2026-09-20 08:03:53 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix(apigateway): export timeout (#13828)

Browse Source
This commit is contained in:
屈轩
2022-03-31 13:51:12 +08:00
committed by GitHub
parent 1378c8a358
commit 042ab4eb02
2 changed files with 16 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
pkg/apigateway/handler/dispatcher.go
View File

@@ -19,6 +19,7 @@ import (
"fmt"
"net/http"
"path"
"time"
"yunion.io/x/onecloud/pkg/appsrv"
)
@@ -51,7 +52,17 @@ func (h SHandler) Bind(app *appsrv.Application) {
if h.MiddlewareFunc != nil {
f = h.MiddlewareFunc(f)
}
app.AddHandler(h.Method, h.Prefix, f)
segs := appsrv.SplitPath(h.Prefix)
hi := appsrv.NewHandlerInfo(h.Method, segs, f, nil, "", nil)
if h.Method == "GET" {
hi.SetProcessTimeoutCallback(func(hdr *appsrv.SHandlerInfo, r *http.Request) time.Duration {
if len(r.URL.Query().Get("export_keys")) > 0 {
return time.Hour * 2
}
return -time.Second
})
}
app.AddHandler3(hi)
}
func NewMethodHandlerFactory(method string, mf appsrv.MiddlewareFunc, prefix string) func(handleFunc, ...string) SHandler {

4
pkg/appsrv/handlerinfo.go
View File

@@ -83,6 +83,10 @@ func (this *SHandlerInfo) GetTags() map[string]string {
return this.tags
}
func NewHandlerInfo(method string, path []string, handler func(context.Context, http.ResponseWriter, *http.Request), metadata map[string]interface{}, name string, tags map[string]string) *SHandlerInfo {
return newHandlerInfo(method, path, handler, metadata, name, tags)
}
func newHandlerInfo(method string, path []string, handler func(context.Context, http.ResponseWriter, *http.Request), metadata map[string]interface{}, name string, tags map[string]string) *SHandlerInfo {
hand := SHandlerInfo{method: method, path: path,
handler: handler,