medium / chi rate limiter

chi_3.go

func main() {
	router := chi.NewRouter()

	router.Use(httprate.LimitByIP(100, time.Minute))

	router.Get("/info", infoHandler)
	router.Group(func(router chi.Router) {
		router.Use(httprate.Limit(20, time.Minute, httprate.WithKeyFuncs(
			httprate.KeyByIP,
			httprate.KeyByEndpoint,
		)))
		router.Use(authMiddleware)
		router.Get("/profile", profileHandler)
		router.Route("/stat", func(router chi.Router) {
			router.Use(httprate.Limit(5, time.Minute,
				httprate.WithKeyFuncs(
					func(r *http.Request) (string, error) {
						token := r.Context().Value("userID").(string)
						return token, nil
					},
				),
			))
			router.Get("/", statHandler)
		})
	})

	_ = http.ListenAndServe(":8080", router)
}