Method traceMiddleware

objectnode/api_middleware.go:78–158

TraceMiddleware returns a middleware handler to trace request. After receiving the request, the handler will assign a unique RequestID to the request and record the processing time of the request.

(next http.Handler)

// After receiving the request, the handler will assign a unique RequestID to
// the request and record the processing time of the request.
func (o *ObjectNode) traceMiddleware(next http.Handler) http.Handler {
	generateRequestID := func() (string, error) {
		var uUID uuid.UUID
		var err error
		if uUID, err = uuid.NewRandom(); err != nil {
			return "", err
		}
		return strings.ReplaceAll(uUID.String(), "-", ""), nil
	}
	var handlerFunc http.HandlerFunc = func(w http.ResponseWriter, r *http.Request) {
		// wrapper for w to record its stats
		w = NewResponseStater(w)
		defer func() {
			p := ParseRequestParam(r)
			extraHeader := auditlog.ExtraHeader(w)
			extraHeader.Set("Tbl", p.Bucket())
			extraHeader.Set("Api", p.API())
			extraHeader.Set("Owner", p.Owner())
			extraHeader.Set("Requester", p.Requester())
			if o.externalAudit != nil {
				o.externalAudit.Logger(w, r)
			}
		}()

		requestID, err := generateRequestID()
		if err != nil {
			log.LogErrorf("traceMiddleware: generate request ID fail, remote(%v) url(%v) err(%v)",
				r.RemoteAddr, r.URL.String(), err)
			InternalErrorCode(err).ServeResponse(w, r)
			// export ump warn info
			exporter.Warning(generateWarnDetail(r, err.Error()))
			return
		}

		// store request ID to context and write to header
		SetRequestID(r, requestID)
		w.Header().Set(XAmzRequestId, requestID)
		w.Header().Set(Server, ValueServer)

		if connHeader := r.Header.Get(Connection); strings.EqualFold(connHeader, "close") {
			w.Header().Set(Connection, "close")
		} else {
			w.Header().Set(Connection, "keep-alive")
		}

		action := ActionFromRouteName(mux.CurrentRoute(r).GetName())
		SetRequestAction(r, action)

		startTime := time.Now()
		metric := exporter.NewTPCnt(fmt.Sprintf("action_%v", action.Name()))
		defer func() {
			metric.Set(err)
		}()

		// Check action is whether enabled.
		if !action.IsNone() && !o.disabledActions.Contains(action) {
			log.LogInfof("traceMiddleware: start with "+
				"action(%v) requestID(%v) host(%v) method(%v) url(%v) header(%+v) remote(%v)",