42 lines
919 B
Go
42 lines
919 B
Go
|
package usecase
|
||
|
|
||
|
import (
|
||
|
"context"
|
||
|
"fmt"
|
||
|
|
||
|
"github.com/tribalwarshelp/api/server"
|
||
|
"github.com/tribalwarshelp/shared/models"
|
||
|
)
|
||
|
|
||
|
type usecase struct {
|
||
|
repo server.Repository
|
||
|
}
|
||
|
|
||
|
func New(repo server.Repository) server.Usecase {
|
||
|
return &usecase{repo}
|
||
|
}
|
||
|
|
||
|
func (ucase *usecase) Fetch(ctx context.Context, filter *models.ServerFilter) ([]*models.Server, int, error) {
|
||
|
if filter == nil {
|
||
|
filter = &models.ServerFilter{}
|
||
|
}
|
||
|
if filter.Limit > server.PaginationLimit || filter.Limit <= 0 {
|
||
|
filter.Limit = server.PaginationLimit
|
||
|
}
|
||
|
return ucase.repo.Fetch(ctx, filter)
|
||
|
}
|
||
|
|
||
|
func (ucase *usecase) GetByKey(ctx context.Context, key string) (*models.Server, error) {
|
||
|
servers, total, err := ucase.repo.Fetch(ctx, &models.ServerFilter{
|
||
|
Key: []string{key},
|
||
|
Limit: 1,
|
||
|
})
|
||
|
if err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
if total == 0 {
|
||
|
return nil, fmt.Errorf("Server with key: %s not found.", key)
|
||
|
}
|
||
|
return servers[0], nil
|
||
|
}
|