diff --git a/db.go b/db.go index 93648fd..1b73622 100644 --- a/db.go +++ b/db.go @@ -70,6 +70,7 @@ type DB interface { // Non retarded APIs below GetItemTypes(itemIDs []int64) ([]models.InvType, error) + GetSolarSystems(systemIDs []int64) ([]models.MapSolarSystem, error) ExpandGroups(groups []int64) ([]int64, error) GetModuleSlots(moduleIDs []int64) (map[int64]ModuleSlot, error) } @@ -695,6 +696,14 @@ func (db *DBWrapper) GetItemTypes(itemIDs []int64) ([]models.InvType, error) { return itemTypes, res.Error } +func (db *DBWrapper) GetSolarSystems(systemIDs []int64) ([]models.MapSolarSystem, error) { + var systems []models.MapSolarSystem + res := db.gormDB.Model(&models.MapSolarSystem{}). + Where("solarSystemID IN ?", systemIDs). + Find(&systems) + return systems, res.Error +} + func deduplicateInt64(slice []int64) []int64 { seen := make(map[int64]bool) result := make([]int64, 0, len(slice))