package service
|
|
import (
|
"fmt"
|
"github.com/xuri/excelize/v2"
|
"strconv"
|
"time"
|
"wms/constvar"
|
"wms/models"
|
"wms/request"
|
"wms/response"
|
)
|
|
type HistoryFormsService struct{}
|
|
func NewHistoryFormsService() *HistoryFormsService {
|
return &HistoryFormsService{}
|
}
|
|
func (slf *HistoryFormsService) Query(params request.GetInventoryHistory) (result []*response.InventoryHistory, err error) {
|
search, err := slf.BuildSearch(params)
|
if err != nil {
|
return nil, err
|
}
|
|
if params.Page > 0 && params.PageSize > 0 {
|
search = search.SetPage(params.Page, params.PageSize)
|
}
|
result = make([]*response.InventoryHistory, 0, params.PageSize)
|
err = search.FindAs(&result)
|
|
return result, nil
|
}
|
|
func (slf *HistoryFormsService) BuildSearch(params request.GetInventoryHistory) (search *models.MoveHistorySearch, err error) {
|
search = models.NewMoveHistorySearch()
|
var (
|
ids []int
|
)
|
if params.KeyWord != "" {
|
ids, _, err = SearchHistoryReport(params.KeyWord, params.BaseOperationType, params.Page, params.PageSize)
|
if err != nil {
|
return
|
}
|
if len(ids) == 0 {
|
return nil, nil
|
}
|
}
|
|
search.Orm = search.Orm.Model(&models.MoveHistory{}).
|
Select("number, updated_at as date, product_name as product_name, from_location_id, operation_id,to_location_id, amount, " +
|
"unit, operator as contacted_name, base_operation_type, weight, product_id, from_location, to_location, operation_type_name, weight").Order("id desc")
|
if len(ids) > 0 {
|
search.Orm = search.Orm.Where("id in ?", ids)
|
}
|
if params.BaseOperationType != 0 {
|
search.Orm = search.Orm.Where("base_operation_type = ?", params.BaseOperationType)
|
}
|
return search, err
|
}
|
|
func (slf *HistoryFormsService) Count(params request.GetInventoryHistory) (total int64, err error) {
|
search, err := slf.BuildSearch(params)
|
if err != nil {
|
return 0, err
|
}
|
total, err = search.Count()
|
return
|
}
|
|
func (slf *HistoryFormsService) FetchAll(params request.GetInventoryHistory) (list []*response.InventoryHistory, err error) {
|
total, err := slf.Count(params)
|
if err != nil {
|
return nil, err
|
}
|
list = make([]*response.InventoryHistory, 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 *HistoryFormsService) Export(dataList []*response.InventoryHistory, params request.GetInventoryHistory) (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.Date.Format("2006-01-02"))
|
f.SetCellValue("Sheet1", "B"+column, v.Number)
|
f.SetCellValue("Sheet1", "C"+column, v.ProductName)
|
f.SetCellValue("Sheet1", "D"+column, v.ProductId)
|
f.SetCellValue("Sheet1", "E"+column, v.OperationTypeName)
|
f.SetCellValue("Sheet1", "F"+column, v.FromLocation)
|
f.SetCellValue("Sheet1", "G"+column, v.ToLocation)
|
f.SetCellValue("Sheet1", "H"+column, v.Amount)
|
f.SetCellValue("Sheet1", "I"+column, v.Unit)
|
f.SetCellValue("Sheet1", "J"+column, v.Weight)
|
}
|
|
if params.BaseOperationType == constvar.BaseOperationTypeIncoming {
|
fileName = fmt.Sprintf("入库明细报表%s.xlsx", time.Now().Format("2006-01-02-1504"))
|
} else if params.BaseOperationType == constvar.BaseOperationTypeOutgoing {
|
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
|
}
|
