Method ServeHTTP

erpc/http_timeout.go:36–143

(w http.ResponseWriter, r *http.Request)

}

func (h *timeoutHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
	ctx, cancelCtx := context.WithTimeoutCause(r.Context(), h.dt, ErrHandlerTimeout)
	defer func() {
		cancelCtx()
	}()
	r = r.WithContext(ctx)
	done := make(chan struct{})
	tw := &timeoutWriter{
		logger: h.logger,
		w:      w,
		h:      make(http.Header),
		req:    r,
		wbuf:   util.BorrowBuf(),
	}
	panicChan := make(chan any, 1)
	go func() {
		defer func() {
			if p := recover(); p != nil {
				telemetry.MetricUnexpectedPanicTotal.WithLabelValues(
					"timeout-handler",
					"",
					common.ErrorFingerprint(p),
				).Inc()
				h.logger.Error().
					Interface("panic", p).
					Str("stack", string(debug.Stack())).
					Msgf("unexpected panic on timeout handler")
				panicChan <- p
			}
		}()
		h.handler.ServeHTTP(tw, r)
		close(done)
	}()
	select {
	case p := <-panicChan:
		panic(p)
	case <-done:
		tw.mu.Lock()
		defer tw.mu.Unlock()
		if err := ctx.Err(); err != nil {
			h.logger.Debug().Err(err).Msg("context canceled before writing response")
			util.ReturnBuf(tw.wbuf)
			tw.wbuf = nil
			return
		}
		dst := w.Header()
		for k, vv := range tw.h {
			dst[k] = vv
		}
		if !tw.wroteHeader {
			tw.code = http.StatusOK
		}
		w.WriteHeader(tw.code)
		_, err := w.Write(tw.wbuf.Bytes())
		util.ReturnBuf(tw.wbuf)
		tw.wbuf = nil
		if err != nil {
			if common.IsClientDisconnect(err) {