middleware

Documentation

Overview

From go-chi https://github.com/go-chi/chi/blob/d32a83448b5f43e42bc96487c6b0b3667a92a2e4/middleware/request_id.go

WrapWriter from go-chi https://github.com/go-chi/chi/blob/d32a83448b5f43e42bc96487c6b0b3667a92a2e4/middleware/wrap_writer.go

Index

• Constants
• Variables
• func BasicAuth(next http.Handler, username, password, realm string) http.Handler
• func GetReqID(ctx context.Context) string
• func JsonToFormMiddleware(next http.Handler) http.Handler
• func Logger(next http.Handler) http.Handler
• func NextRequestID() uint64
• func RequestID(next http.Handler) http.Handler
• func RequestLogger(f LogFormatter) func(next http.Handler) http.Handler
• func WithLogEntry(r *http.Request, entry LogEntry) *http.Request
• type AuthLevel
• type DefaultLogFormatter
  • func (l *DefaultLogFormatter) NewLogEntry(r *http.Request) LogEntry
• type LogEntry
  • func GetLogEntry(r *http.Request) LogEntry
• type LogFormatter
• type Middleware
  • func (m *Middleware) AbuseProtectorMiddleware() func(next http.Handler) http.Handler
  • func (m *Middleware) AuthMiddleware(levels ...AuthLevel) func(next http.Handler) http.Handler
  • func (m *Middleware) GeoIPMiddleware() func(next http.Handler) http.Handler
  • func (m *Middleware) RateLimit(requestLimit int, customKey string, windowLength time.Duration) func(next http.Handler) http.Handler
• type RuleFunc
• type ShouldBanRule
• type WrapResponseWriter
  • func NewWrapResponseWriter(w http.ResponseWriter, protoMajor int) WrapResponseWriter

Constants

const RequestIDKey ctxKeyRequestID = 0

RequestIDKey is the key that holds the unique request ID in a request context.

Variables

var (
	// LogEntryCtxKey is the context.Context key to store the request log entry.
	LogEntryCtxKey = &contextKey{"LogEntry"}

	// DefaultLogger is called by the Logger middleware handler to log each request.
	// Its made a package-level variable so that it can be reconfigured for custom
	// logging configurations.
	DefaultLogger func(next http.Handler) http.Handler
)

var IsTTY bool

var RequestIDHeader = "X-Request-Id"

RequestIDHeader is the name of the HTTP Header which contains the request id. Exported so that it can be changed by developers

Functions

func BasicAuth

func BasicAuth(next http.Handler, username, password, realm string) http.Handler

BasicAuth is a wrapper for Handler that requires username and password

func GetReqID

func GetReqID(ctx context.Context) string

GetReqID returns a request ID from the given context if one is present. Returns the empty string if a request ID cannot be found.

func JsonToFormMiddleware

func JsonToFormMiddleware(next http.Handler) http.Handler

func Logger

func Logger(next http.Handler) http.Handler

Logger is a middleware that logs the start and end of each request, along with some useful data about what was requested, what the response status was, and how long it took to return. When standard output is a TTY, Logger will print in color, otherwise it will print in black and white. Logger prints a request ID if one is provided.

Alternatively, look at https://github.com/goware/httplog for a more in-depth http logger with structured logging support.

IMPORTANT NOTE: Logger should go before any other middleware that may change the response, such as middleware.Recoverer. Example:

r := chi.NewRouter()
r.Use(middleware.Logger)        // <--<< Logger should come before Recoverer
r.Use(middleware.Recoverer)
r.Get("/", handler)

func NextRequestID

func NextRequestID() uint64

NextRequestID generates the next request ID in the sequence.

func RequestID

func RequestID(next http.Handler) http.Handler

RequestID is a middleware that injects a request ID into the context of each request. A request ID is a string of the form "host.example.com/random-0001", where "random" is a base62 random string that uniquely identifies this go process, and where the last number is an atomically incremented request counter.

func RequestLogger

func RequestLogger(f LogFormatter) func(next http.Handler) http.Handler

RequestLogger returns a logger handler using a custom LogFormatter.

func WithLogEntry

func WithLogEntry(r *http.Request, entry LogEntry) *http.Request

WithLogEntry sets the in-context LogEntry for a request.

Types

type AuthLevel

type AuthLevel int

const (
	AuthLevelAny AuthLevel = iota
	AuthLevelGalleryAdmin
	AuthLevelSuperAdmin
	AuthLevelAPIToken
	AuthLevelOptional
)

type DefaultLogFormatter

type DefaultLogFormatter struct {
	NoColor bool
}

DefaultLogFormatter is a simple logger that implements a LogFormatter.

func (*DefaultLogFormatter) NewLogEntry

func (l *DefaultLogFormatter) NewLogEntry(r *http.Request) LogEntry

NewLogEntry creates a new LogEntry for the request.

type LogEntry

type LogEntry interface {
	Write(status, bytes int, header http.Header, elapsed time.Duration, extra interface{})
}

LogEntry records the final log when a request completes. See defaultLogEntry for an example implementation.

func GetLogEntry

func GetLogEntry(r *http.Request) LogEntry

GetLogEntry returns the in-context LogEntry for a request.

type LogFormatter

type LogFormatter interface {
	NewLogEntry(r *http.Request) LogEntry
}

LogFormatter initiates the beginning of a new LogEntry per request. See DefaultLogFormatter for an example implementation.

type Middleware

type Middleware struct {
	SupabaseAuth *database.SupabaseAuth
	Repo         *repository.Repository
	Redis        *database.RedisWrapper
	Track        *analytics.AnalyticsService
}

func (*Middleware) AbuseProtectorMiddleware

func (m *Middleware) AbuseProtectorMiddleware() func(next http.Handler) http.Handler

func (*Middleware) AuthMiddleware

func (m *Middleware) AuthMiddleware(levels ...AuthLevel) func(next http.Handler) http.Handler

Enforces authorization at specific level

func (*Middleware) GeoIPMiddleware

func (m *Middleware) GeoIPMiddleware() func(next http.Handler) http.Handler

func (*Middleware) RateLimit

func (m *Middleware) RateLimit(requestLimit int, customKey string, windowLength time.Duration) func(next http.Handler) http.Handler

Rate limit middleware @requestLimit: The number of requests they can make @windowLength: In this time window

type RuleFunc

type RuleFunc func(*http.Request) bool

type ShouldBanRule

type ShouldBanRule struct {
	Reason string
	Func   RuleFunc
}

type WrapResponseWriter

type WrapResponseWriter interface {
	http.ResponseWriter
	// Status returns the HTTP status of the request, or 0 if one has not
	// yet been sent.
	Status() int
	// BytesWritten returns the total number of bytes sent to the client.
	BytesWritten() int
	// Tee causes the response body to be written to the given io.Writer in
	// addition to proxying the writes through. Only one io.Writer can be
	// tee'd to at once: setting a second one will overwrite the first.
	// Writes will be sent to the proxy before being written to this
	// io.Writer. It is illegal for the tee'd writer to be modified
	// concurrently with writes.
	Tee(io.Writer)
	// Unwrap returns the original proxied target.
	Unwrap() http.ResponseWriter
}

WrapResponseWriter is a proxy around an http.ResponseWriter that allows you to hook into various parts of the response process.

func NewWrapResponseWriter

func NewWrapResponseWriter(w http.ResponseWriter, protoMajor int) WrapResponseWriter

NewWrapResponseWriter wraps an http.ResponseWriter, returning a proxy that allows you to hook into various parts of the response process.