Timo Riegebauer
dd92f23e0b
Core framework: - Kite router with full HTTP method support (GET, POST, PUT, DELETE, HEAD, OPTIONS, CONNECT, TRACE) - Nestable route groups with prefix and middleware inheritance - Global, group, and route-level middleware with onion ordering - Centralized error, not-found, and method-not-allowed handlers - Graceful shutdown on SIGTERM/SIGINT with configurable timeout - Configurable server read, write, and idle timeouts - Context with typed request/response helpers, value store, cookie, form, and body binding support - Response writer wrapper for status code tracking Middleware: - Logger with configurable output, format, and skip function - Recovery with configurable panic handler - RequestID with configurable header and generator, forwards incoming IDs - CORS with configurable origins, methods, headers, credentials, and max age - MaxBodySize with configurable byte limit Docs: - README with quickstart, routing, middleware, context API reference, and TLS guide
29 lines
596 B
Go
29 lines
596 B
Go
package middleware
|
|
|
|
import (
|
|
"net/http"
|
|
|
|
"git.trcreatives.at/trcreatives/go-kite"
|
|
)
|
|
|
|
type MaxBodySizeConfig struct {
|
|
MaxBytes int64
|
|
}
|
|
|
|
func MaxBodySize(maxBytes int64) kite.Middleware {
|
|
return MaxBodySizeWithConfig(MaxBodySizeConfig{MaxBytes: maxBytes})
|
|
}
|
|
|
|
func MaxBodySizeWithConfig(cfg MaxBodySizeConfig) kite.Middleware {
|
|
if cfg.MaxBytes == 0 {
|
|
cfg.MaxBytes = 4 << 20
|
|
}
|
|
|
|
return func(h kite.Handler) kite.Handler {
|
|
return func(ctx *kite.Context) error {
|
|
ctx.GetRequest().Body = http.MaxBytesReader(ctx.GetResponse(), ctx.GetRequest().Body, cfg.MaxBytes)
|
|
return h(ctx)
|
|
}
|
|
}
|
|
}
|