Compare commits
1 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
aaf697a005
|
@@ -113,7 +113,7 @@ func NewWithFlaresolverr(clientID, clientSecret string, f *flaresolverr.Flaresol
|
|||||||
return c
|
return c
|
||||||
}
|
}
|
||||||
|
|
||||||
func (g *Client) Request(ctx context.Context, method string, URL string, dataBody any) (*resty.Response, error) {
|
func (g *Client) Request(ctx context.Context, method string, requestURL string, dataBody any) (*resty.Response, error) {
|
||||||
err := g.limiter.Wait(ctx)
|
err := g.limiter.Wait(ctx)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, fmt.Errorf("failed to get rate limiter token: %w", err)
|
return nil, fmt.Errorf("failed to get rate limiter token: %w", err)
|
||||||
@@ -129,14 +129,14 @@ func (g *Client) Request(ctx context.Context, method string, URL string, dataBod
|
|||||||
"Authorization": "Bearer " + t,
|
"Authorization": "Bearer " + t,
|
||||||
"User-Agent": "",
|
"User-Agent": "",
|
||||||
"Content-Type": "text/plain",
|
"Content-Type": "text/plain",
|
||||||
}).Execute(strings.ToUpper(method), URL)
|
}).Execute(strings.ToUpper(method), requestURL)
|
||||||
|
|
||||||
if resp.StatusCode() != 200 {
|
if resp.StatusCode() != 200 {
|
||||||
return nil, fmt.Errorf("failed to request, expected 200 but got: %v", resp.StatusCode())
|
return nil, fmt.Errorf("failed to request, expected 200 but got: %v", resp.StatusCode())
|
||||||
}
|
}
|
||||||
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, fmt.Errorf("failed to request: %s: %w", URL, err)
|
return nil, fmt.Errorf("failed to request: %s: %w", requestURL, err)
|
||||||
}
|
}
|
||||||
return resp, nil
|
return resp, nil
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -2,6 +2,7 @@ package endpoint
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
"strconv"
|
"strconv"
|
||||||
"strings"
|
"strings"
|
||||||
@@ -32,6 +33,9 @@ func (b *BaseEndpoint[T]) Query(ctx context.Context, query string) ([]*T, error)
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (b *BaseEndpoint[T]) GetByID(ctx context.Context, id uint64) (*T, error) {
|
func (b *BaseEndpoint[T]) GetByID(ctx context.Context, id uint64) (*T, error) {
|
||||||
|
if id == 0 {
|
||||||
|
return nil, errors.New("id cant be 0")
|
||||||
|
}
|
||||||
res, err := b.Query(ctx, fmt.Sprintf("where id = %d; fields *;", id))
|
res, err := b.Query(ctx, fmt.Sprintf("where id = %d; fields *;", id))
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
@@ -44,7 +48,7 @@ func (b *BaseEndpoint[T]) GetByID(ctx context.Context, id uint64) (*T, error) {
|
|||||||
|
|
||||||
func (b *BaseEndpoint[T]) GetByIDs(ctx context.Context, ids []uint64) ([]*T, error) {
|
func (b *BaseEndpoint[T]) GetByIDs(ctx context.Context, ids []uint64) ([]*T, error) {
|
||||||
if len(ids) == 0 {
|
if len(ids) == 0 {
|
||||||
return nil, fmt.Errorf("ids cant be empty")
|
return []*T{}, nil
|
||||||
}
|
}
|
||||||
batches := make([][]uint64, 0)
|
batches := make([][]uint64, 0)
|
||||||
for i := 0; i < len(ids); i += 500 {
|
for i := 0; i < len(ids); i += 500 {
|
||||||
@@ -80,7 +84,11 @@ func (b *BaseEndpoint[T]) Count(ctx context.Context) (uint64, error) {
|
|||||||
return 0, fmt.Errorf("failed to unmarshal: %w", err)
|
return 0, fmt.Errorf("failed to unmarshal: %w", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if res.Count > 0 {
|
||||||
return uint64(res.Count), nil
|
return uint64(res.Count), nil
|
||||||
|
} else {
|
||||||
|
return 0, fmt.Errorf("failed to count, count should larger than 0, but got %v", res.Count)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (b *BaseEndpoint[T]) Paginated(ctx context.Context, offset, limit uint64) ([]*T, error) {
|
func (b *BaseEndpoint[T]) Paginated(ctx context.Context, offset, limit uint64) ([]*T, error) {
|
||||||
|
|||||||
Reference in New Issue
Block a user