ntfy/server/visitor.go

66 lines
1.3 KiB
Go
Raw Normal View History

2021-11-01 19:21:38 +00:00
package server
import (
"golang.org/x/time/rate"
"heckel.io/ntfy/config"
2021-11-01 20:39:40 +00:00
"heckel.io/ntfy/util"
2021-11-01 19:21:38 +00:00
"sync"
"time"
)
const (
visitorExpungeAfter = 30 * time.Minute
)
// visitor represents an API user, and its associated rate.Limiter used for rate limiting
type visitor struct {
config *config.Config
limiter *rate.Limiter
2021-11-01 20:39:40 +00:00
subscriptions *util.Limiter
2021-11-01 19:21:38 +00:00
seen time.Time
mu sync.Mutex
}
func newVisitor(conf *config.Config) *visitor {
return &visitor{
2021-11-01 20:39:40 +00:00
config: conf,
limiter: rate.NewLimiter(rate.Every(conf.VisitorRequestLimitReplenish), conf.VisitorRequestLimitBurst),
2021-11-01 20:39:40 +00:00
subscriptions: util.NewLimiter(int64(conf.VisitorSubscriptionLimit)),
seen: time.Now(),
2021-11-01 19:21:38 +00:00
}
}
func (v *visitor) RequestAllowed() error {
if !v.limiter.Allow() {
return errHTTPTooManyRequests
}
return nil
}
func (v *visitor) AddSubscription() error {
v.mu.Lock()
defer v.mu.Unlock()
2021-11-01 20:39:40 +00:00
if err := v.subscriptions.Add(1); err != nil {
2021-11-01 19:21:38 +00:00
return errHTTPTooManyRequests
}
return nil
}
func (v *visitor) RemoveSubscription() {
v.mu.Lock()
defer v.mu.Unlock()
2021-11-01 20:39:40 +00:00
v.subscriptions.Sub(1)
2021-11-01 19:21:38 +00:00
}
func (v *visitor) Keepalive() {
v.mu.Lock()
defer v.mu.Unlock()
v.seen = time.Now()
}
func (v *visitor) Stale() bool {
v.mu.Lock()
defer v.mu.Unlock()
return time.Since(v.seen) > visitorExpungeAfter
}