func()

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)

	})

}