Как я могу настроить контекст регистрации из промежуточного программного обеспечения? - PullRequest
Новая
       42

Как я могу настроить контекст регистрации из промежуточного программного обеспечения?

0 голосов
/ 31 декабря 2018

Я хочу заполнить контекст ведения журнала элементами в запросе, например: r.Header.Get("X-Request-Id").Я предположил, что могу переопределить тип журнала в обработчике из промежуточного программного обеспечения.Хотя, похоже, это не работает, и я не уверен, почему! 

package main

import (
    "fmt"
    "net/http"
    "os"

    "github.com/apex/log"
    "github.com/gorilla/mux"
)

// Assumption: handler is the shared state between the functions
type handler struct{ Log *log.Entry }

// New creates a handler for this application to co-ordinate shared resources
func New() (h handler) { return handler{Log: log.WithFields(log.Fields{"test": "FAIL"})} }

func (h handler) index(w http.ResponseWriter, r *http.Request) {
    h.Log.Info("Hello from the logger")
    fmt.Fprint(w, "YO")
}

func main() {
    h := New()

    app := mux.NewRouter()
    app.HandleFunc("/", h.index)
    app.Use(h.loggingMiddleware)

    if err := http.ListenAndServe(":"+os.Getenv("PORT"), app); err != nil {
        log.WithError(err).Fatal("error listening")
    }

}

func (h handler) loggingMiddleware(next http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        h.Log = log.WithFields(log.Fields{"test": "PASS"})
        next.ServeHTTP(w, r)
    })
}

Можете ли вы понять, почему h.Log = log.WithFields(log.Fields{"test": "PASS"}) не оказывает никакого влияния на h.Log.Info("Hello from the logger"), который должен быть IIUC в том же самомпросить?

1 Ответ

0 голосов
/ 31 декабря 2018

Вам нужно, чтобы ваш регистратор находился в области запроса.Вы устанавливаете его глобально для всего обработчика, каждый раз, когда появляется новое соединение, что означает, что вы запрашиваете гонки данных и, как правило, нежелательное поведение.

Для контекста в области запроса, context.Context в заявке идеально.Вы можете получить к нему доступ через методы Context() и WithContext.

Пример: 

var loggerKey = "Some unique key"

func (h handler) loggingMiddleware(next http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        ctx := r.Context()
        ctx = context.WithValue(ctx, loggerKey, log.WithFields(log.Fields{"test": "PASS"}))
        next.ServeHTTP(w, r.WithContext(ctx)
    })
}

Затем для доступа к регистратору:

func doSomething(r *http.Request) error {
    log, ok := r.Context().Value(loggerKey).(*log.Logger) // Or whatever type is appropriate
    if !ok {
        return errors.New("logger not set on context!")
    }
    // Do stuff...
}
...