2020-06-02 15:45:21 +00:00
|
|
|
package usecase
|
|
|
|
|
|
|
|
import (
|
|
|
|
"context"
|
2021-05-05 17:49:00 +00:00
|
|
|
"github.com/pkg/errors"
|
2021-05-05 17:32:45 +00:00
|
|
|
"github.com/tribalwarshelp/shared/tw/twmodel"
|
2020-06-02 15:45:21 +00:00
|
|
|
|
2020-08-09 12:32:46 +00:00
|
|
|
"github.com/tribalwarshelp/api/middleware"
|
2020-06-02 15:45:21 +00:00
|
|
|
"github.com/tribalwarshelp/api/server"
|
|
|
|
)
|
|
|
|
|
|
|
|
type usecase struct {
|
|
|
|
repo server.Repository
|
|
|
|
}
|
|
|
|
|
|
|
|
func New(repo server.Repository) server.Usecase {
|
|
|
|
return &usecase{repo}
|
|
|
|
}
|
|
|
|
|
2021-05-05 17:32:45 +00:00
|
|
|
func (ucase *usecase) Fetch(ctx context.Context, cfg server.FetchConfig) ([]*twmodel.Server, int, error) {
|
2020-11-21 09:50:52 +00:00
|
|
|
if cfg.Filter == nil {
|
2021-05-05 17:32:45 +00:00
|
|
|
cfg.Filter = &twmodel.ServerFilter{}
|
2020-06-02 15:45:21 +00:00
|
|
|
}
|
2021-03-21 06:42:12 +00:00
|
|
|
if !middleware.CanExceedLimit(ctx) && (cfg.Limit > server.FetchLimit || cfg.Limit <= 0) {
|
|
|
|
cfg.Limit = server.FetchLimit
|
2020-11-21 09:50:52 +00:00
|
|
|
}
|
2021-05-05 18:15:51 +00:00
|
|
|
if len(cfg.Sort) > server.MaxOrders {
|
|
|
|
cfg.Sort = cfg.Sort[0:server.MaxOrders]
|
|
|
|
}
|
2020-11-21 09:50:52 +00:00
|
|
|
return ucase.repo.Fetch(ctx, cfg)
|
2020-06-02 15:45:21 +00:00
|
|
|
}
|
|
|
|
|
2021-05-05 17:32:45 +00:00
|
|
|
func (ucase *usecase) GetByKey(ctx context.Context, key string) (*twmodel.Server, error) {
|
2020-06-24 13:05:31 +00:00
|
|
|
servers, _, err := ucase.repo.Fetch(ctx, server.FetchConfig{
|
2021-05-05 17:32:45 +00:00
|
|
|
Filter: &twmodel.ServerFilter{
|
2020-11-21 09:50:52 +00:00
|
|
|
Key: []string{key},
|
2020-06-24 13:05:31 +00:00
|
|
|
},
|
2021-03-21 12:03:39 +00:00
|
|
|
Limit: 1,
|
|
|
|
Count: false,
|
|
|
|
Select: true,
|
2020-06-02 15:45:21 +00:00
|
|
|
})
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
2020-06-24 13:05:31 +00:00
|
|
|
if len(servers) == 0 {
|
2021-05-05 17:49:00 +00:00
|
|
|
return nil, errors.Errorf("Server (key: %s) not found.", key)
|
2020-06-02 15:45:21 +00:00
|
|
|
}
|
|
|
|
return servers[0], nil
|
|
|
|
}
|