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.Name
|
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)
|
result = append(result, resp)
|
}
|
|
return
|
}
|
|
func (slf *LocationFormsService) BuildSearch(params request.GetLocationForms) (search *models.LocationProductAmountSearch, err error) {
|
ids := make([]int, 0)
|
if params.LocationId != 0 {
|
ids = append(ids, params.LocationId)
|
} else {
|
//查询位置
|
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()
|
|
// 自定义表头
|
headers := []string{"位置", "产品", "产品类别", "在库数量", "价值"}
|
|
// 设置表头
|
for i, header := range headers {
|
cell := getColumnAlphabet(i+1) + "1"
|
f.SetCellValue("Sheet1", cell, header)
|
}
|
|
for i, v := range dataList {
|
column := strconv.Itoa(i + 2)
|
f.SetCellValue("Sheet1", "A"+column, v.LocationName)
|
f.SetCellValue("Sheet1", "B"+column, v.ProductName)
|
f.SetCellValue("Sheet1", "C"+column, v.ProductTypeName)
|
f.SetCellValue("Sheet1", "D"+column, v.Amount)
|
f.SetCellValue("Sheet1", "E"+column, 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
|
}
|
