middleware.go

package zapper

import (
	"context"
	"net/http"

	"go.uber.org/zap"
)

func LogMiddleware(h http.Handler) http.Handler {

	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {

		// TODO:
		// req body
		// resp body
		// latency 响应时长
		newLogger := wrapReqFields(defaultLogger, r)

		r.WithContext(context.WithValue(r.Context(), "logger", newLogger))

		h.ServeHTTP(w, r)

		newLogger.Info("finished handle a request")

	})

}

// wrapReqFields 包裹请求携带的字段
func wrapReqFields(logger *zap.Logger, r *http.Request) *zap.Logger {
	urlQuery := r.URL.Query()
	return logger.With(
		// TODO: 找到字段所在报文位置
		zap.String("logid", urlQuery.Get("")),
		zap.String("caller_ip", r.Header.Get("")),
		zap.String("host_ip", r.Host),
		zap.String("product", urlQuery.Get("")),
		zap.String("module", urlQuery.Get("")),
		zap.String("service_id", urlQuery.Get("")),
		zap.String("instance_id", urlQuery.Get("")),
		zap.String("uri_path", urlQuery.Get("")),
		zap.String("tag", urlQuery.Get("")),
		zap.String("env", urlQuery.Get("")),
		zap.String("sversion", urlQuery.Get("")),
		zap.String("stag", urlQuery.Get("")),
	)
}