0762ddbbc2
Add per-context and per-action rate limiting using golang.org/x/time/rate. Configure globally via Options.ActionRateLimit or per-action with WithRateLimit(). Defaults to 10 req/s with burst of 20.
70 lines
2.2 KiB
Go
70 lines
2.2 KiB
Go
package via
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/alexedwards/scs/v2"
|
|
"github.com/rs/zerolog"
|
|
)
|
|
|
|
func ptr(l zerolog.Level) *zerolog.Level { return &l }
|
|
|
|
var (
|
|
LogLevelDebug = ptr(zerolog.DebugLevel)
|
|
LogLevelInfo = ptr(zerolog.InfoLevel)
|
|
LogLevelWarn = ptr(zerolog.WarnLevel)
|
|
LogLevelError = ptr(zerolog.ErrorLevel)
|
|
)
|
|
|
|
// Plugin is a func that can mutate the given *via.V app runtime. It is useful to integrate popular JS/CSS UI libraries or tools.
|
|
type Plugin func(v *V)
|
|
|
|
// Options defines configuration options for the via application
|
|
type Options struct {
|
|
// The development mode flag. If true, enables server and browser auto-reload on `.go` file changes.
|
|
DevMode bool
|
|
|
|
// The http server address. e.g. ':3000'
|
|
ServerAddress string
|
|
|
|
// LogLevel sets the minimum log level. nil keeps the default (Info).
|
|
LogLevel *zerolog.Level
|
|
|
|
// Logger overrides the default logger entirely. When set, LogLevel and
|
|
// DevMode have no effect on logging.
|
|
Logger *zerolog.Logger
|
|
|
|
// The title of the HTML document.
|
|
DocumentTitle string
|
|
|
|
// Plugins to extend the capabilities of the `Via` application.
|
|
Plugins []Plugin
|
|
|
|
// SessionManager enables cookie-based sessions. If set, Via wraps handlers
|
|
// with scs LoadAndSave middleware. Configure the session manager before
|
|
// passing it (lifetime, cookie settings, store, etc).
|
|
SessionManager *scs.SessionManager
|
|
|
|
// DatastarContent is the Datastar.js script content.
|
|
// If nil, the embedded default is used.
|
|
DatastarContent []byte
|
|
|
|
// DatastarPath is the URL path where the script is served.
|
|
// Defaults to "/_datastar.js" if empty.
|
|
DatastarPath string
|
|
|
|
// PubSub enables publish/subscribe messaging. Use vianats.New() for an
|
|
// embedded NATS backend, or supply any PubSub implementation.
|
|
PubSub PubSub
|
|
|
|
// ContextTTL is the maximum time a context may exist without an SSE
|
|
// connection before the background reaper disposes it.
|
|
// Default: 30s. Negative value disables the reaper.
|
|
ContextTTL time.Duration
|
|
|
|
// ActionRateLimit configures the default token-bucket rate limiter for
|
|
// action endpoints. Zero values use built-in defaults (10 req/s, burst 20).
|
|
// Set Rate to -1 to disable rate limiting entirely.
|
|
ActionRateLimit RateLimitConfig
|
|
}
|