package service
|
|
import (
|
"fmt"
|
"github.com/shopspring/decimal"
|
"github.com/xuri/excelize/v2"
|
"strconv"
|
"time"
|
"wms/constvar"
|
"wms/models"
|
"wms/request"
|
"wms/response"
|
)
|
|
type Detail struct {
|
ProductId string `json:"productId"`
|
Amount decimal.Decimal `json:"amount"`
|
Status constvar.OperationStatus `json:"status"`
|
//ProductName string `json:"productName"`
|
}
|
|
type InventoryReportFormsService struct{}
|
|
func NewInventoryReportFormsService() *InventoryReportFormsService {
|
return &InventoryReportFormsService{}
|
}
|
|
func (slf *InventoryReportFormsService) Query(params request.GetInventoryForms) (result []*response.InventoryForms, err error) {
|
var (
|
materials = make([]*models.Material, 0)
|
)
|
search, productIds, productAmounts, locationIds, err := slf.BuildSearch(params)
|
if err != nil {
|
return nil, err
|
}
|
|
if params.Page*params.PageSize > 0 {
|
search.Orm = search.Orm.Offset((params.Page - 1) * params.PageSize).Limit(params.PageSize)
|
}
|
if err := search.Orm.Find(&materials).Error; err != nil {
|
return nil, err
|
}
|
//查询在库数量
|
if len(productIds) == 0 {
|
for _, material := range materials {
|
productIds = append(productIds, material.ID)
|
}
|
productAmounts, err = models.NewLocationProductAmountSearch().SetProductIds(productIds).SetLocationIds(locationIds).Find()
|
if err != nil {
|
return nil, err
|
}
|
}
|
|
//查询出入库就绪数量
|
var inHouse []Detail
|
var outHouse []Detail
|
dbIn := models.NewOperationDetailsSearch().Orm.Model(&models.OperationDetails{}).
|
Select("wms_operation_details.product_id,wms_operation_details.amount,wms_operation.status as status").
|
Joins("left join wms_operation ON wms_operation_details.operation_id=wms_operation.id").
|
Where("wms_operation.base_operation_type in (?)", []constvar.BaseOperationType{constvar.BaseOperationTypeIncoming, constvar.BaseOperationTypeInternal}).
|
Where("wms_operation.status in (?)", []constvar.OperationStatus{constvar.OperationStatus_Finish})
|
dbOut := models.NewOperationDetailsSearch().Orm.Model(&models.OperationDetails{}).
|
Select("wms_operation_details.product_id,wms_operation_details.amount,wms_operation.status as status").
|
Joins("left join wms_operation ON wms_operation_details.operation_id=wms_operation.id").
|
Where("wms_operation.base_operation_type in (?)", []constvar.BaseOperationType{constvar.BaseOperationTypeOutgoing, constvar.BaseOperationTypeInternal, constvar.BaseOperationTypeDisuse}).
|
Where("wms_operation.status in (?)", []constvar.OperationStatus{constvar.OperationStatus_Ready, constvar.OperationStatus_Finish})
|
if len(locationIds) > 0 {
|
dbIn.Where("wms_operation_details.to_location_id in (?)", locationIds)
|
dbOut.Where("wms_operation_details.from_location_id in (?)", locationIds)
|
}
|
if len(productIds) > 0 {
|
dbIn.Where("wms_operation_details.product_id in (?)", productIds)
|
dbOut.Where("wms_operation_details.product_id in (?)", productIds)
|
}
|
err = dbIn.Find(&inHouse).Error
|
if err != nil {
|
return
|
}
|
err = dbOut.Find(&outHouse).Error
|
if err != nil {
|
return
|
}
|
for _, material := range materials {
|
data := new(response.InventoryForms)
|
data.ProduceId = material.ID
|
data.ProductName = material.Name
|
data.Cost = material.Cost
|
data.Unit = material.Unit
|
data.Value = material.Amount.Mul(material.Cost)
|
data.ProductType = material.CategoryName
|
for _, amount := range productAmounts {
|
if material.ID == amount.ProductId {
|
data.Amount = data.Amount.Add(amount.Amount)
|
}
|
}
|
for _, details := range inHouse {
|
if material.ID == details.ProductId {
|
data.In = data.In.Add(details.Amount)
|
}
|
}
|
available := decimal.NewFromInt(0)
|
for _, details := range outHouse {
|
if material.ID == details.ProductId {
|
if details.Status == constvar.OperationStatus_Ready {
|
available = available.Add(details.Amount)
|
} else {
|
data.Out = data.Out.Add(details.Amount)
|
}
|
}
|
}
|
data.AvailableNumber = data.Amount.Sub(available)
|
result = append(result, data)
|
}
|
return
|
}
|
|
func (slf *InventoryReportFormsService) BuildSearch(params request.GetInventoryForms) (
|
search *models.MaterialSearch,
|
productIds []string,
|
productAmounts []*models.LocationProductAmount,
|
locationIds []int,
|
err error) {
|
productAmounts = make([]*models.LocationProductAmount, 0)
|
if params.WarehouseCode != "" {
|
locations, err := models.NewLocationSearch().SetJointName(params.WarehouseCode).FindNotTotal()
|
if err != nil {
|
return nil, nil, nil, nil, err
|
}
|
|
for _, location := range locations {
|
locationIds = append(locationIds, location.Id)
|
}
|
productAmounts, err = models.NewLocationProductAmountSearch().SetLocationIds(locationIds).SetQuery("amount > 0").Find()
|
if err != nil {
|
return nil, nil, nil, nil, err
|
}
|
for _, amount := range productAmounts {
|
productIds = append(productIds, amount.ProductId)
|
}
|
}
|
|
//查询产品
|
search = models.NewMaterialSearch()
|
search.Orm = search.Orm.Model(&models.Material{}).
|
Select(`material.id, material.name, material.cost, material.amount, material.unit, wms_product_category.name as category_name `).
|
Joins("left join wms_product_category on material.category_id = wms_product_category.id")
|
if len(params.CategoryIds) > 0 {
|
search.Orm.Where("material.category_id in (?)", params.CategoryIds)
|
}
|
if params.KeyWord != "" {
|
search.Orm.Where("material.name like ?", "%"+params.KeyWord+"%").Or("wms_product_category.name like ?", "%"+params.KeyWord+"%")
|
}
|
if len(productIds) > 0 {
|
search.Orm.Where("material.id in (?)", productIds)
|
}
|
|
return search, productIds, productAmounts, locationIds, nil
|
}
|
|
func (slf *InventoryReportFormsService) Count(params request.GetInventoryForms) (total int64, err error) {
|
search, _, _, _, err := slf.BuildSearch(params)
|
if err != nil {
|
return 0, err
|
}
|
if err := search.Orm.Count(&total).Error; err != nil {
|
return 0, err
|
}
|
return total, nil
|
}
|
|
func (slf *InventoryReportFormsService) FetchAll(params request.GetInventoryForms) (list []*response.InventoryForms, err error) {
|
total, err := slf.Count(params)
|
if err != nil {
|
return nil, err
|
}
|
list = make([]*response.InventoryForms, 0, total)
|
params.PageSize = 1000
|
page := 1
|
for {
|
params.Page = page
|
data, err := slf.Query(params)
|
if err != nil {
|
return nil, err
|
}
|
if len(data) == 0 {
|
break
|
}
|
list = append(list, data...)
|
page++
|
}
|
return
|
}
|
|
func (slf *InventoryReportFormsService) Export(dataList []*response.InventoryForms) (filename string, err error) {
|
var fileName string
|
f := excelize.NewFile()
|
|
// 自定义表头
|
headers := []string{"产品", "产品类别", "单位成本", "总价值", "在库", "可用库存", "入库", "出库", "单位"}
|
|
// 设置表头
|
for i, header := range headers {
|
cell := getColumnAlphabet(i+1) + "1"
|
f.SetCellValue("Sheet1", cell, header)
|
}
|
|
for i, v := range dataList {
|
f.SetCellValue("Sheet1", "A"+strconv.Itoa(i+2), v.ProductName)
|
f.SetCellValue("Sheet1", "B"+strconv.Itoa(i+2), v.ProductType)
|
f.SetCellValue("Sheet1", "C"+strconv.Itoa(i+2), v.Cost)
|
f.SetCellValue("Sheet1", "D"+strconv.Itoa(i+2), v.Value)
|
f.SetCellValue("Sheet1", "E"+strconv.Itoa(i+2), v.Amount)
|
f.SetCellValue("Sheet1", "F"+strconv.Itoa(i+2), v.AvailableNumber)
|
f.SetCellValue("Sheet1", "G"+strconv.Itoa(i+2), v.In)
|
f.SetCellValue("Sheet1", "H"+strconv.Itoa(i+2), v.Out)
|
f.SetCellValue("Sheet1", "I"+strconv.Itoa(i+2), v.Unit)
|
}
|
|
fileName = fmt.Sprintf("库存报表%s.xlsx", time.Now().Format("2006-01-02-1504"))
|
if err := f.SaveAs(fileName); err != nil {
|
return fileName, err
|
}
|
|
return fileName, nil
|
}
|
|
// 根据索引获取 Excel 列名（A、B、C...）
|
func getColumnAlphabet(index int) string {
|
const alphabet = "ABCDEFGHIJKLMNOPQRSTUVWXYZ"
|
if index <= 26 {
|
return string(alphabet[index-1])
|
}
|
return getColumnAlphabet((index-1)/26) + getColumnAlphabet((index-1)%26+1)
|
}
|
