marc/scotty
1
0
Fork 0
mirror of https://git.sr.ht/~phw/scotty synced 2025-04-29 13:27:03 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Moved ratelimit to pkg

Browse source
This commit is contained in:
Philipp Wolfer 2023-12-11 23:21:29 +01:00
parent c4da3a40cc
commit 6ac2b4f142
No known key found for this signature in database
GPG key ID: 8FDF744D4919943B
4 changed files with 12 additions and 3 deletions
internal/backends
funkwhale
client.go
listenbrainz
client.go
spotify
client.go
pkg/ratelimit
httpheader.go

2
internal/backends/funkwhale/client.go
View file

@ -26,8 +26,8 @@ import (
"strconv"
"github.com/go-resty/resty/v2"
"go.uploadedlobster.com/scotty/internal/ratelimit"
"go.uploadedlobster.com/scotty/internal/version"
"go.uploadedlobster.com/scotty/pkg/ratelimit"
)
const MaxItemsPerGet = 50

2
internal/backends/listenbrainz/client.go
View file

@ -27,8 +27,8 @@ import (
"time"
"github.com/go-resty/resty/v2"
"go.uploadedlobster.com/scotty/internal/ratelimit"
"go.uploadedlobster.com/scotty/internal/version"
"go.uploadedlobster.com/scotty/pkg/ratelimit"
)
const (

2
internal/backends/spotify/client.go
View file

@ -29,8 +29,8 @@ import (
"time"
"github.com/go-resty/resty/v2"
"go.uploadedlobster.com/scotty/internal/ratelimit"
"go.uploadedlobster.com/scotty/internal/version"
"go.uploadedlobster.com/scotty/pkg/ratelimit"
"golang.org/x/oauth2"
)

9
internal/ratelimit/httpheader.go → pkg/ratelimit/httpheader.go
View file

@ -29,6 +29,15 @@ const (
MaxWaitTimeSeconds = 60
)
// Implements rate HTTP header based limiting for resty.
//
// This works with servers that return the status code 429 (Too Many Requests)
// and an HTTP header indicating the time in seconds until rate limit resets.
// Common headers used are "X-RateLimit-Reset-In" or "Retry-After".
//
// Usage:
//
// ratelimit.EnableHTTPHeaderRateLimit(client, "Retry-After")
func EnableHTTPHeaderRateLimit(client *resty.Client, resetInHeader string) {
client.SetRetryCount(RetryCount)
client.AddRetryCondition(