package service
|
|
import (
|
"strings"
|
"wms/models"
|
)
|
|
func GetWarehouseByLocationIds(locationIDs []int) (locationHouseMap map[int]*models.Warehouse, houseLocationIds map[int][]int, houseMap map[int]*models.Warehouse, err error) {
|
//查询位置信息
|
locations, err := models.NewLocationSearch().SetIds(locationIDs).FindNotTotal()
|
if err != nil {
|
return
|
}
|
codes := make([]string, 0)
|
codeLocationMap := make(map[string]int, len(locations))
|
for _, loc := range locations {
|
//根据仓库缩写查询仓库
|
code := strings.Split(loc.JointName, "/")[0]
|
codes = append(codes, code)
|
codeLocationMap[code] = loc.Id
|
}
|
|
codeWareHouseMap := make(map[string]*models.Warehouse, len(codes))
|
warehouses, err := models.NewWarehouseSearch().SetCodes(codes).FindNotTotal()
|
if err != nil {
|
return
|
}
|
|
houseMap = make(map[int]*models.Warehouse, len(warehouses))
|
for _, w := range warehouses {
|
codeWareHouseMap[w.Code] = w
|
houseMap[w.Id] = w
|
}
|
|
locationHouseMap = make(map[int]*models.Warehouse)
|
|
for _, w := range warehouses {
|
locationHouseMap[codeLocationMap[w.Code]] = w
|
}
|
|
houseLocationIds = make(map[int][]int)
|
for locationID, w := range locationHouseMap {
|
if houseLocationIds[w.Id] == nil {
|
houseLocationIds[w.Id] = make([]int, 0)
|
}
|
houseLocationIds[w.Id] = append(houseLocationIds[w.Id], locationID)
|
}
|
|
return
|
}
|
