in server/handler/track_handler.go [46:102]
func (h *TrackHandler) Handle(chain *handler.Chain, inv *invocation.Invocation, cb invocation.ResponseCallBack) {
req, ok := inv.Args.(*restful.Request)
if !ok {
chain.Next(inv, cb)
return
}
if req.Request.Method != http.MethodGet {
chain.Next(inv, cb)
return
}
if !strings.Contains(req.Request.URL.Path, "kie/kv") {
chain.Next(inv, cb)
return
}
sessionID := req.HeaderParameter(v1.HeaderSessionID)
if sessionID == "" {
chain.Next(inv, cb)
return
}
chain.Next(inv, func(ir *invocation.Response) {
if ir.Status != 200 {
cb(ir)
return
}
resp, _ := ir.Result.(*restful.Response)
revStr := req.QueryParameter(common.QueryParamRev)
wait := req.QueryParameter(common.QueryParamWait)
data := &model.PollingDetail{}
data.URLPath = req.Request.Method + " " + req.Request.URL.Path
data.SessionID = sessionID
data.SessionGroup = req.HeaderParameter(v1.HeaderSessionGroup)
data.UserAgent = req.HeaderParameter(v1.HeaderUserAgent)
data.Domain = v1.ReadDomain(req.Request.Context())
data.Project = req.PathParameter(common.PathParameterProject)
data.IP = iputil.ClientIP(req.Request)
data.ResponseBody = req.Attribute(common.RespBodyContextKey).([]*model.KVDoc)
data.ResponseCode = ir.Status
data.Timestamp = time.Now()
if resp != nil {
data.Revision = resp.Header().Get(common.HeaderRevision)
}
data.PollingData = map[string]interface{}{
"revision": revStr,
"wait": wait,
"labels": req.QueryParameter("label"),
}
_, err := datasource.GetBroker().GetTrackDao().CreateOrUpdate(inv.Ctx, data)
if err != nil {
openlog.Warn("record polling detail failed:" + err.Error())
cb(ir)
return
}
cb(ir)
})
}