package service

import (
	"github.com/shopspring/decimal"
	"wms/models"
)

func GetLocationAmounts(productIds []string, locationIds []int) (locationAmountMap map[string]map[int]decimal.Decimal, err error) {
	locationAmounts, err := models.NewLocationProductAmountSearch().
		SetProductIds(productIds).SetLocationIds(locationIds).
		SetFields("product_id, location_id, amount").
		Find()
	if err != nil {
		return nil, err
	}
	locationAmountMap = make(map[string]map[int]decimal.Decimal)
	for _, locationAmount := range locationAmounts {
		if locationAmountMap[locationAmount.ProductId] == nil {
			locationAmountMap[locationAmount.ProductId] = make(map[int]decimal.Decimal)
		}
		locationAmountMap[locationAmount.ProductId][locationAmount.LocationId] = locationAmount.Amount
	}
	return
}