2020-06-02 15:45:21 +00:00
|
|
|
package dataloaders
|
|
|
|
|
|
|
|
import (
|
|
|
|
"context"
|
2020-06-24 13:05:31 +00:00
|
|
|
"github.com/tribalwarshelp/api/player"
|
|
|
|
"github.com/tribalwarshelp/api/tribe"
|
2020-06-22 15:59:56 +00:00
|
|
|
"github.com/tribalwarshelp/api/village"
|
2020-06-02 15:45:21 +00:00
|
|
|
"github.com/tribalwarshelp/shared/models"
|
|
|
|
)
|
|
|
|
|
2020-06-18 14:44:02 +00:00
|
|
|
type ServerDataLoaders struct {
|
2020-12-25 11:42:46 +00:00
|
|
|
PlayerByID *PlayerLoader
|
|
|
|
TribeByID *TribeLoader
|
|
|
|
VillageByID *VillageLoader
|
2020-06-02 15:45:21 +00:00
|
|
|
}
|
|
|
|
|
2020-06-18 14:44:02 +00:00
|
|
|
func NewServerDataLoaders(server string, cfg Config) *ServerDataLoaders {
|
|
|
|
return &ServerDataLoaders{
|
2020-12-25 11:42:46 +00:00
|
|
|
PlayerByID: &PlayerLoader{
|
2021-03-21 12:03:39 +00:00
|
|
|
wait: wait,
|
2020-06-02 15:45:21 +00:00
|
|
|
maxBatch: 0,
|
|
|
|
fetch: func(ids []int) ([]*models.Player, []error) {
|
2020-06-24 13:05:31 +00:00
|
|
|
players, _, err := cfg.PlayerRepo.Fetch(context.Background(), player.FetchConfig{
|
|
|
|
Filter: &models.PlayerFilter{
|
|
|
|
ID: ids,
|
|
|
|
},
|
2021-03-21 12:03:39 +00:00
|
|
|
Select: true,
|
2020-06-24 13:05:31 +00:00
|
|
|
Server: server,
|
2020-06-02 15:45:21 +00:00
|
|
|
})
|
|
|
|
if err != nil {
|
|
|
|
return nil, []error{err}
|
|
|
|
}
|
|
|
|
|
|
|
|
playerByID := make(map[int]*models.Player)
|
|
|
|
for _, player := range players {
|
|
|
|
playerByID[player.ID] = player
|
|
|
|
}
|
|
|
|
|
2020-06-07 14:53:32 +00:00
|
|
|
inOrder := make([]*models.Player, len(ids))
|
2020-06-02 15:45:21 +00:00
|
|
|
for i, id := range ids {
|
2020-06-07 14:53:32 +00:00
|
|
|
inOrder[i] = playerByID[id]
|
2020-06-02 15:45:21 +00:00
|
|
|
}
|
|
|
|
|
2020-06-07 14:53:32 +00:00
|
|
|
return inOrder, nil
|
2020-06-02 15:45:21 +00:00
|
|
|
},
|
|
|
|
},
|
2020-12-25 11:42:46 +00:00
|
|
|
TribeByID: &TribeLoader{
|
2021-03-21 12:03:39 +00:00
|
|
|
wait: wait,
|
2020-06-02 15:45:21 +00:00
|
|
|
maxBatch: 0,
|
|
|
|
fetch: func(ids []int) ([]*models.Tribe, []error) {
|
2020-06-24 13:05:31 +00:00
|
|
|
tribes, _, err := cfg.TribeRepo.Fetch(context.Background(), tribe.FetchConfig{
|
|
|
|
Server: server,
|
|
|
|
Filter: &models.TribeFilter{
|
|
|
|
ID: ids,
|
|
|
|
},
|
2021-03-21 12:03:39 +00:00
|
|
|
Select: true,
|
2020-06-02 15:45:21 +00:00
|
|
|
})
|
|
|
|
if err != nil {
|
|
|
|
return nil, []error{err}
|
|
|
|
}
|
|
|
|
|
|
|
|
tribeByID := make(map[int]*models.Tribe)
|
|
|
|
for _, tribe := range tribes {
|
|
|
|
tribeByID[tribe.ID] = tribe
|
|
|
|
}
|
|
|
|
|
2020-06-07 14:53:32 +00:00
|
|
|
inOrder := make([]*models.Tribe, len(ids))
|
2020-06-02 15:45:21 +00:00
|
|
|
for i, id := range ids {
|
2020-06-07 14:53:32 +00:00
|
|
|
inOrder[i] = tribeByID[id]
|
2020-06-02 15:45:21 +00:00
|
|
|
}
|
|
|
|
|
2020-06-07 14:53:32 +00:00
|
|
|
return inOrder, nil
|
2020-06-03 15:21:28 +00:00
|
|
|
},
|
|
|
|
},
|
2020-12-25 11:42:46 +00:00
|
|
|
VillageByID: &VillageLoader{
|
2021-03-21 12:03:39 +00:00
|
|
|
wait: wait,
|
2020-06-03 15:21:28 +00:00
|
|
|
maxBatch: 0,
|
|
|
|
fetch: func(ids []int) ([]*models.Village, []error) {
|
2020-06-22 15:59:56 +00:00
|
|
|
villages, _, err := cfg.VillageRepo.Fetch(context.Background(), village.FetchConfig{
|
|
|
|
Server: server,
|
|
|
|
Count: false,
|
|
|
|
Filter: &models.VillageFilter{
|
|
|
|
ID: ids,
|
|
|
|
},
|
2021-03-21 12:03:39 +00:00
|
|
|
Select: true,
|
2020-06-03 15:21:28 +00:00
|
|
|
})
|
|
|
|
if err != nil {
|
|
|
|
return nil, []error{err}
|
|
|
|
}
|
|
|
|
|
|
|
|
villageByID := make(map[int]*models.Village)
|
|
|
|
for _, village := range villages {
|
|
|
|
villageByID[village.ID] = village
|
|
|
|
}
|
|
|
|
|
2020-06-07 14:53:32 +00:00
|
|
|
inOrder := make([]*models.Village, len(ids))
|
2020-06-03 15:21:28 +00:00
|
|
|
for i, id := range ids {
|
2020-06-07 14:53:32 +00:00
|
|
|
inOrder[i] = villageByID[id]
|
2020-06-03 15:21:28 +00:00
|
|
|
}
|
|
|
|
|
2020-06-07 14:53:32 +00:00
|
|
|
return inOrder, nil
|
2020-06-02 15:45:21 +00:00
|
|
|
},
|
|
|
|
},
|
|
|
|
}
|
|
|
|
}
|