|
@@ -7,6 +7,7 @@ import (
|
|
"github.com/gogf/gf/v2/frame/g"
|
|
"github.com/gogf/gf/v2/frame/g"
|
|
"github.com/gogf/gf/v2/net/ghttp"
|
|
"github.com/gogf/gf/v2/net/ghttp"
|
|
"github.com/gogf/gf/v2/os/gcmd"
|
|
"github.com/gogf/gf/v2/os/gcmd"
|
|
|
|
+ "time"
|
|
)
|
|
)
|
|
|
|
|
|
var (
|
|
var (
|
|
@@ -18,8 +19,9 @@ var (
|
|
s := g.Server()
|
|
s := g.Server()
|
|
s.Use(service.Middleware)
|
|
s.Use(service.Middleware)
|
|
s.BindHandler("/*", func(r *ghttp.Request) {
|
|
s.BindHandler("/*", func(r *ghttp.Request) {
|
|
- queryLevel := r.GetCtxVar(consts.QueryLevelKey).Int()
|
|
|
|
|
|
+ queryLevel, now := r.GetCtxVar(consts.QueryLevelKey).Int(), time.Now()
|
|
rp, err := service.EsProxyManager.GetProxy(r.Context(), queryLevel)
|
|
rp, err := service.EsProxyManager.GetProxy(r.Context(), queryLevel)
|
|
|
|
+ r.SetCtxVar(consts.QueryWaitPoolTime, time.Now().Sub(now).Seconds()) //记录等待时长
|
|
if err != nil {
|
|
if err != nil {
|
|
r.Response.Status = 500
|
|
r.Response.Status = 500
|
|
g.Log().Errorf(r.Context(), "GetProxy Error:%v", err)
|
|
g.Log().Errorf(r.Context(), "GetProxy Error:%v", err)
|