Add a per-client rate limiter

#58 Go · Senior

Try another stack

internal/ratelimit/limiter.go hover a line, then click the + · drag to comment on a range

@@ -1,5 +1,31 @@
     package ratelimit
     
    + import "time"
    + 
    + // Limiter caps requests per client over a one-minute window.
    + type Limiter struct {
    + 	hits map[string]int
    + 	max  int
    + }
    + 
    + func New(max int) *Limiter {
    + 	l := &Limiter{hits: make(map[string]int), max: max}
    + 
    + 	go func() {
    + 		for range time.Tick(time.Minute) {
    + 			l.hits = make(map[string]int)
    + 		}
    + 	}()
    + 
    + 	return l
    + }
    + 
    + func (l *Limiter) Allow(client string) bool {
    + 	l.hits[client]++
    + 	return l.hits[client] <= l.max
    + }