u

endpoint/base.go

@@ -1,38 +1,70 @@
 package endpoint
 
 import (
+	"fmt"
+	"strconv"
+	"strings"
+
 	"github.com/go-resty/resty/v2"
 )
 
-type BaseEndpoint struct {
+type BaseEndpoint[T any] struct {
 	request      func(URL string, dataBody any) (*resty.Response, error)
 	endpointName EndpointName
+	queryFunc    func(string) ([]*T, error)
 }
 
-func NewBaseEndpoint(request func(URL string, dataBody any) (*resty.Response, error), endpointName EndpointName) *BaseEndpoint {
-	return &BaseEndpoint{
-		request:      request,
-		endpointName: endpointName,
-	}
-}
-
-func (b *BaseEndpoint) GetEndpointName() EndpointName {
+func (b *BaseEndpoint[T]) GetEndpointName() EndpointName {
 	return b.endpointName
 }
 
-func (b *BaseEndpoint) Query(query string) (any, error) {
-	return nil, nil
+func (b *BaseEndpoint[T]) Query(query string) ([]*T, error) {
+	if b.queryFunc == nil {
+		return nil, fmt.Errorf("Query method must be implemented by specific endpoint")
+	}
+	return b.queryFunc(query)
 }
 
-func (b *BaseEndpoint) QueryAny(query string) (any, error) {
-	return b.Query(query)
+func (b *BaseEndpoint[T]) GetByID(id uint64) (*T, error) {
+	res, err := b.Query(fmt.Sprintf("where id = %d; fields *;", id))
+	if err != nil {
+		return nil, err
+	}
+	if len(res) == 0 {
+		return nil, fmt.Errorf("no results")
+	}
+	return res[0], nil
 }
 
-type Endpoint interface {
-	GetEndpointName() EndpointName
+func (b *BaseEndpoint[T]) GetByIDs(ids []uint64) ([]*T, error) {
+	builder := strings.Builder{}
+	for i, v := range ids {
+		if i > 0 {
+			builder.WriteByte(',')
+		}
+		builder.WriteString(strconv.FormatUint(v, 10))
+	}
+	return b.Query(fmt.Sprintf("where id = (%s); fields *;", builder.String()))
 }
 
-type EntityEndpoint interface {
-	QueryAny(query string) (any, error)
-	GetEndpointName() EndpointName
+func (b *BaseEndpoint[T]) GetLastOneId() (uint64, error) {
+	res, err := b.Query("fields *; sort id desc; limit 1;")
+	if err != nil {
+		return 0, err
+	}
+	if len(res) == 0 {
+		return 0, fmt.Errorf("no results")
+	}
+	type IdGetter interface {
+		GetId() uint64
+	}
+	item, ok := any(res[0]).(IdGetter)
+	if !ok {
+		return 0, fmt.Errorf("invalid type")
+	}
+	return item.GetId(), nil
+}
+
+func (b *BaseEndpoint[T]) Paginated(offset, limit uint64) ([]*T, error) {
+	return b.Query(fmt.Sprintf("offset %d; limit %d; fields *; sort id asc;", offset, limit))
 }