Dynamic retryafter

_example/main.go

@@ -7,9 +7,9 @@ import (
 	"net/http"
 	"time"
 
-	"github.com/go-chi/chi/v5"
-	"github.com/go-chi/chi/v5/middleware"
-	"github.com/go-chi/httprate"
+	"github.com/Alge/chi/v5"
+	"github.com/Alge/chi/v5/middleware"
+	"github.com/Alge/httprate"
 )
 
 func main() {

go.mod

@@ -1,4 +1,4 @@
-module github.com/go-chi/httprate
+module github.com/Alge/httprate
 
 go 1.17
 

limiter.go

@@ -97,7 +97,9 @@ func (l *RateLimiter) OnLimit(w http.ResponseWriter, r *http.Request, key string
 		setHeader(w, l.headers.Remaining, fmt.Sprintf("%d", limit-rate))
 
 		l.mu.Unlock()
-		setHeader(w, l.headers.RetryAfter, fmt.Sprintf("%d", int(l.windowLength.Seconds()))) // RFC 6585
+
+		retryAfter := int(math.Ceil(time.Now().UTC().Sub(currentWindow.Add(l.windowLength)).Seconds()))
+		setHeader(w, l.headers.RetryAfter, fmt.Sprintf("%d", retryAfter)) // RFC 6585
 		return true
 	}
 

limiter_test.go

@@ -12,7 +12,7 @@ import (
 	"testing"
 	"time"
 
-	"github.com/go-chi/httprate"
+	"github.com/Alge/httprate"
 )
 
 func TestLimit(t *testing.T) {

local_counter_test.go

@@ -7,7 +7,7 @@ import (
 	"testing"
 	"time"
 
-	"github.com/go-chi/httprate"
+	"github.com/Alge/httprate"
 	"golang.org/x/sync/errgroup"
 )