Function Middleware

agent/agentchat/log.go:56–72

Middleware tags agent logs for requests that originate from chatd. Agent log lines emitted while serving a request with Coder-Chat-Id, or by background work started by such a request, should include chat_id. Install after loggermw.Logger so access-log enrichment can run.

(next http.Handler)

// or by background work started by such a request, should include chat_id.
// Install after loggermw.Logger so access-log enrichment can run.
func Middleware(next http.Handler) http.Handler {
	return http.HandlerFunc(func(rw http.ResponseWriter, r *http.Request) {
		chatID, ancestorIDs, ok := extractContext(r)
		if !ok {
			next.ServeHTTP(rw, r)
			return
		}

		fields := chatFields(chatID, ancestorIDs)
		if requestLogger := loggermw.RequestLoggerFromContext(r.Context()); requestLogger != nil {
			requestLogger.WithFields(fields...)
		}

		ctx := WithContext(r.Context(), chatID, ancestorIDs)
		next.ServeHTTP(rw, r.WithContext(ctx))
	})
}

func chatFields(chatID uuid.UUID, ancestorIDs []uuid.UUID) []slog.Field {
	fields := []slog.Field{slog.F("chat_id", chatID.String())}