marc/scotty
1
0
Fork 0
mirror of https://git.sr.ht/~phw/scotty synced 2025-04-10 23:49:28 +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
Show stats Download patch file Download diff file Expand all files Collapse all files

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

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

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

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

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

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

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

@ -29,6 +29,15 @@ const (
MaxWaitTimeSeconds = 60 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) { func EnableHTTPHeaderRateLimit(client *resty.Client, resetInHeader string) {
client.SetRetryCount(RetryCount) client.SetRetryCount(RetryCount)
client.AddRetryCondition( client.AddRetryCondition(