adyxax/terraform-provider-forgejo
1
0
Fork 0
Code Issues1 Pull requests Releases3 Activity

feat(client): detect pagination limit at client instantiation

Browse source 
Closes
This commit is contained in:
Julien Dessaux 2025-05-25 18:12:46 +02:00
parent f4d69460af
commit 7e1e9eb26b
Signed by: adyxax
GPG key ID: F92E51B86E07177E
4 changed files with 25 additions and 15 deletions
internal/client
client.go

26
internal/client/client.go
View file

@ -13,16 +13,15 @@ import (
)
type Client struct {
baseURI *url.URL
headers *http.Header
httpClient *http.Client
baseURI *url.URL
headers *http.Header
httpClient *http.Client
maxItemsPerPage int
maxItemsPerPageStr string
}
const maxItemsPerPage = 50
const maxItemsPerPageStr = "50"
func NewClient(baseURL *url.URL, apiToken string) *Client {
return &Client{
func NewClient(baseURL *url.URL, apiToken string) (*Client, error) {
c := Client{
baseURI: baseURL,
headers: &http.Header{
"Accept": {"application/json"},
@ -33,6 +32,13 @@ func NewClient(baseURL *url.URL, apiToken string) *Client {
Timeout: time.Minute,
},
}
settings, err := c.settingsApiGet()
if err != nil {
return nil, fmt.Errorf("failed to get settings api: %w", err)
}
c.maxItemsPerPage = settings.MaxResponseItems
c.maxItemsPerPageStr = strconv.Itoa(c.maxItemsPerPage)
return &c, nil
}
func (c *Client) sendPaginated(ctx context.Context, method string, uriRef *url.URL, payload any, response any) error {
@ -40,7 +46,7 @@ func (c *Client) sendPaginated(ctx context.Context, method string, uriRef *url.U
if err != nil {
return fmt.Errorf("failed to parse query string: %w", err)
}
query.Set("limit", maxItemsPerPageStr)
query.Set("limit", c.maxItemsPerPageStr)
page := 1
var rawResponses []json.RawMessage
for {
@ -56,7 +62,7 @@ func (c *Client) sendPaginated(ctx context.Context, method string, uriRef *url.U
return fmt.Errorf("failed to unmarshal message: %w", err)
}
rawResponses = append(rawResponses, oneResponse...)
if count <= page*maxItemsPerPage {
if count <= page*c.maxItemsPerPage {
break
}
page++