package service
|
|
import (
|
"fmt"
|
"github.com/xuri/excelize/v2"
|
"strconv"
|
"time"
|
"wms/models"
|
"wms/request"
|
"wms/response"
|
)
|
|
type LocationFormsService struct{}
|
|
func NewLocationFormsService() *LocationFormsService {
|
return &LocationFormsService{}
|
}
|
|
func (slf *LocationFormsService) Query(params request.GetLocationForms) (result []*response.LocationForms, err error) {
|
search, err := slf.BuildSearch(params)
|
search = search.SetPage(params.Page, params.PageSize)
|
amounts, err := search.Find()
|
if err != nil {
|
return
|
}
|
for _, amount := range amounts {
|
resp := new(response.LocationForms)
|
resp.Amount = amount.Amount
|
resp.LocationId = amount.LocationId
|
resp.LocationName = amount.Location.JointName
|
resp.ProduceId = amount.Product.ID
|
resp.ProductName = amount.Product.Name
|
resp.ProductTypeName = amount.ProductCategory.Name
|
resp.Unit = amount.Product.Unit
|
resp.Value = resp.Amount.Mul(amount.Product.Cost)
|
|
moreUnit := amount.Product.MoreUnit
|
if moreUnit != nil && *moreUnit {
|
resp.AmountMoreUnits = amount.MoreUnitList
|
}
|
|
result = append(result, resp)
|
}
|
|
return
|
}
|
|
func (slf *LocationFormsService) BuildSearch(params request.GetLocationForms) (search *models.LocationProductAmountSearch, err error) {
|
ids := params.LocationIds
|
if params.LocationId != 0 {
|
ids = append(ids, params.LocationId)
|
}
|
if len(ids) == 0 {
|
//查询位置
|
locations, err := models.NewLocationSearch().SetJointName(params.WareHouseCode).SetType(3).FindAll()
|
if err != nil {
|
return nil, err
|
}
|
for _, location := range locations {
|
ids = append(ids, location.Id)
|
}
|
}
|
|
search = models.NewLocationProductAmountSearch().SetPreload(true).SetKeyword(params.KeyWord).SetProductId(params.ProductId).SetLocationIds(ids)
|
|
return search, nil
|
}
|
|
func (slf *LocationFormsService) Count(params request.GetLocationForms) (total int64, err error) {
|
search, err := slf.BuildSearch(params)
|
if err != nil {
|
return 0, err
|
}
|
return search.Count()
|
}
|
|
func (slf *LocationFormsService) FetchAll(params request.GetLocationForms) (list []*response.LocationForms, err error) {
|
total, err := slf.Count(params)
|
if err != nil {
|
return nil, err
|
}
|
list = make([]*response.LocationForms, 0, total)
|
params.PageSize = 500
|
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 *LocationFormsService) Export(dataList []*response.LocationForms) (filename string, err error) {
|
var fileName string
|
f := excelize.NewFile()
|
|
unitData, unitIndexMap := GetAllUnits()
|
// 自定义表头
|
headers := []interface{}{"位置", "产品", "产品类别", map[string][]string{"在库数量": unitData}, "价值"}
|
|
// 设置表头
|
i := 1
|
for _, h := range headers {
|
if v, ok := h.(string); ok {
|
f.SetCellValue("Sheet1", getColumnAlphabet(i)+"1", v)
|
i++
|
} else if childHeaders, ok := h.(map[string][]string); ok {
|
for title, list := range childHeaders {
|
f.SetCellValue("Sheet1", getColumnAlphabet(i)+"1", title)
|
err = f.MergeCell("Sheet1", getColumnAlphabet(i)+"1", getColumnAlphabet(i-1+len(list))+"1") // 合并单元格
|
for _, t := range list {
|
f.SetCellValue("Sheet1", getColumnAlphabet(i)+"2", t)
|
i++
|
}
|
}
|
}
|
}
|
|
// 设置表头样式
|
style := &excelize.Style{
|
Border: nil,
|
Fill: excelize.Fill{
|
Type: "pattern",
|
Pattern: 1,
|
Shading: 0,
|
},
|
Font: &excelize.Font{
|
Bold: true,
|
},
|
Alignment: &excelize.Alignment{
|
Horizontal: "center",
|
},
|
}
|
titleStyle, err := f.NewStyle(style)
|
if err != nil {
|
return
|
}
|
|
lastColumn := getColumnAlphabet(i)
|
f.SetCellStyle("Sheet1", "A1", lastColumn+"2", titleStyle)
|
// 设置列宽
|
f.SetColWidth("Sheet1", "A", "C", 30)
|
|
for i, v := range dataList {
|
row := strconv.Itoa(i + 3)
|
f.SetCellValue("Sheet1", "A"+row, v.LocationName)
|
f.SetCellValue("Sheet1", "B"+row, v.ProductName)
|
f.SetCellValue("Sheet1", "C"+row, v.ProductTypeName)
|
v.AmountMoreUnits = append(v.AmountMoreUnits, models.UnitItems{Amount: v.Amount, Unit: v.Unit})
|
FillMoreUnitToExcel(v.AmountMoreUnits, 4, i+3, unitIndexMap, f)
|
f.SetCellValue("Sheet1", getColumnAlphabet(4+len(unitData))+row, v.Value)
|
}
|
|
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
|
}
|
