库存报表下载

zhangqian

2024-06-04 c6c500d7044c5e4785fc856a5a1253b0604a8147

库存报表下载

 controllers/report_forms_controller.go

@@ -4,10 +4,13 @@
    "fmt"
    "github.com/gin-gonic/gin"
    "github.com/shopspring/decimal"
    "net/url"
    "os"
    "wms/constvar"
    "wms/extend/code"
    "wms/extend/util"
    "wms/models"
    "wms/pkg/logx"
    "wms/request"
    "wms/response"
    "wms/service"
@@ -28,6 +31,7 @@
// @Tags      报表
// @Summary   获取库存报表
// @Produce   application/json
// @Param     Authorization    header string true "token"
// @Param     object  body  request.GetInventoryForms true  "查询参数"
// @Success   200 {object} util.ResponseList{data=[]response.InventoryForms}    "成功"
// @Router    /api-wms/v1/forms/getInventoryForms [post]
@@ -38,137 +42,64 @@
        util.ResponseFormat(c, code.RequestParamError, "参数解析失败，数据类型错误")
        return
    }
    locationIds := make([]int, 0)
    productIds := make([]string, 0)
    productAmounts := make([]*models.LocationProductAmount, 0)
    if params.WarehouseCode != "" {
        locations, err := models.NewLocationSearch().SetJointName(params.WarehouseCode).FindNotTotal()
        if err != nil {
            util.ResponseFormat(c, code.RequestParamError, "查询仓库位置失败")
            return
        }

        for _, location := range locations {
            locationIds = append(locationIds, location.Id)
        }
        productAmounts, err = models.NewLocationProductAmountSearch().SetLocationIds(locationIds).SetQuery("amount > 0").Find()
        if err != nil {
            util.ResponseFormat(c, code.RequestParamError, "查询在库数量失败")
            return
        }
        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)
    }
    var (
        materials = make([]*models.Material, 0)
        total     int64
    )
    if err := search.Orm.Count(&total).Error; err != nil {
        util.ResponseFormat(c, code.RequestParamError, "查询total失败")
        return
    }
    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 {
        util.ResponseFormat(c, code.RequestParamError, fmt.Errorf("查询产品失败: %v", err))
        return
    }

    //查询在库数量
    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 {
            util.ResponseFormat(c, code.RequestParamError, "查询在库数量失败")
            return
        }
    }

    //查询出入库就绪数量
    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
    inventoryReportFormsService := service.NewInventoryReportFormsService()
    total, err := inventoryReportFormsService.Count(params)
    if err != nil {
        util.ResponseFormat(c, code.RequestParamError, "查询入库数量失败")
        logx.Errorf("GetInventoryForms count err:%v", err)
        util.ResponseFormat(c, code.InternalError, "查询总数失败")
        return
    }
    err = dbOut.Find(&outHouse).Error
    result, err := inventoryReportFormsService.Query(params)
    if err != nil {
        util.ResponseFormat(c, code.RequestParamError, "查询出库数量失败")
        logx.Errorf("GetInventoryForms query err:%v", err)
        util.ResponseFormat(c, code.InternalError, "查询失败")
        return
    }
    var result []response.InventoryForms
    for _, material := range materials {
        var resp response.InventoryForms
        resp.ProduceId = material.ID
        resp.ProductName = material.Name
        resp.Cost = material.Cost
        resp.Unit = material.Unit
        resp.Value = material.Amount.Mul(material.Cost)
        resp.ProductType = material.CategoryName
        for _, amount := range productAmounts {
            if material.ID == amount.ProductId {
                resp.Amount = resp.Amount.Add(amount.Amount)
            }
        }
        for _, details := range inHouse {
            if material.ID == details.ProductId {
                resp.In = resp.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 {
                    resp.Out = resp.Out.Add(details.Amount)
                }
            }
        }
        resp.AvailableNumber = resp.Amount.Sub(available)
        result = append(result, resp)
    }

    util.ResponseFormatList(c, code.Success, result, int(total))
}

// DownloadInventoryForms
// @Tags      报表
// @Summary   下载库存报表
// @Produce   application/json
// @Param     Authorization    header string true "token"
// @Param     object  body  request.GetInventoryForms true  "查询参数"
// @Success   200 {object} util.ResponseList{data=[]response.InventoryForms}    "成功"
// @Router    /api-wms/v1/forms/downloadInventoryForms [post]
func (slf ReportFormsController) DownloadInventoryForms(c *gin.Context) {
    var params request.GetInventoryForms
    err := c.BindJSON(&params)
    if err != nil {
        util.ResponseFormat(c, code.RequestParamError, "参数解析失败，数据类型错误")
        return
    }
    inventoryReportFormsService := service.NewInventoryReportFormsService()
    list, err := inventoryReportFormsService.FetchAll(params)
    if err != nil {
        logx.Errorf("DownloadInventoryForms FetchAll err:%v", err)
        util.ResponseFormat(c, code.InternalError, "查询失败")
        return
    }
    filename, err := inventoryReportFormsService.Export(list)
    if err != nil {
        logx.Errorf("DownloadInventoryForms Export err:%v", err)
        util.ResponseFormat(c, code.InternalError, "导出数据到文件失败")
        return
    }

    if err != nil {
        util.ResponseFormat(c, code.RequestParamError, "导出失败")
        return
    }

    fileContentDisposition := "attachment;filename=\"" + url.QueryEscape(filename) + "\""
    c.Header("Content-Type", "application/xlsx")
    c.Header("Content-Disposition", fileContentDisposition)
    c.File(filename)
    defer os.Remove(filename)
}

// GetHistory
// @Tags      报表
// @Summary   获取历史信息

 docs/docs.go

@@ -468,6 +468,58 @@
                }
            }
        },
        "/api-wms/v1/forms/downloadInventoryForms": {
            "post": {
                "produces": [
                    "application/json"
                ],
                "tags": [
                    "报表"
                ],
                "summary": "下载库存报表",
                "parameters": [
                    {
                        "type": "string",
                        "description": "token",
                        "name": "Authorization",
                        "in": "header",
                        "required": true
                    },
                    {
                        "description": "查询参数",
                        "name": "object",
                        "in": "body",
                        "required": true,
                        "schema": {
                            "$ref": "#/definitions/request.GetInventoryForms"
                        }
                    }
                ],
                "responses": {
                    "200": {
                        "description": "成功",
                        "schema": {
                            "allOf": [
                                {
                                    "$ref": "#/definitions/util.ResponseList"
                                },
                                {
                                    "type": "object",
                                    "properties": {
                                        "data": {
                                            "type": "array",
                                            "items": {
                                                "$ref": "#/definitions/response.InventoryForms"
                                            }
                                        }
                                    }
                                }
                            ]
                        }
                    }
                }
            }
        },
        "/api-wms/v1/forms/getHistory": {
            "post": {
                "produces": [
@@ -524,6 +576,13 @@
                "summary": "获取库存报表",
                "parameters": [
                    {
                        "type": "string",
                        "description": "token",
                        "name": "Authorization",
                        "in": "header",
                        "required": true
                    },
                    {
                        "description": "查询参数",
                        "name": "object",
                        "in": "body",

 docs/swagger.json

@@ -456,6 +456,58 @@
                }
            }
        },
        "/api-wms/v1/forms/downloadInventoryForms": {
            "post": {
                "produces": [
                    "application/json"
                ],
                "tags": [
                    "报表"
                ],
                "summary": "下载库存报表",
                "parameters": [
                    {
                        "type": "string",
                        "description": "token",
                        "name": "Authorization",
                        "in": "header",
                        "required": true
                    },
                    {
                        "description": "查询参数",
                        "name": "object",
                        "in": "body",
                        "required": true,
                        "schema": {
                            "$ref": "#/definitions/request.GetInventoryForms"
                        }
                    }
                ],
                "responses": {
                    "200": {
                        "description": "成功",
                        "schema": {
                            "allOf": [
                                {
                                    "$ref": "#/definitions/util.ResponseList"
                                },
                                {
                                    "type": "object",
                                    "properties": {
                                        "data": {
                                            "type": "array",
                                            "items": {
                                                "$ref": "#/definitions/response.InventoryForms"
                                            }
                                        }
                                    }
                                }
                            ]
                        }
                    }
                }
            }
        },
        "/api-wms/v1/forms/getHistory": {
            "post": {
                "produces": [
@@ -512,6 +564,13 @@
                "summary": "获取库存报表",
                "parameters": [
                    {
                        "type": "string",
                        "description": "token",
                        "name": "Authorization",
                        "in": "header",
                        "required": true
                    },
                    {
                        "description": "查询参数",
                        "name": "object",
                        "in": "body",

 docs/swagger.yaml

@@ -2220,6 +2220,37 @@
      summary: 手动跑月度统计库存报表
      tags:
      - 报表
  /api-wms/v1/forms/downloadInventoryForms:
    post:
      parameters:
      - description: token
        in: header
        name: Authorization
        required: true
        type: string
      - description: 查询参数
        in: body
        name: object
        required: true
        schema:
          $ref: '#/definitions/request.GetInventoryForms'
      produces:
      - application/json
      responses:
        "200":
          description: 成功
          schema:
            allOf:
            - $ref: '#/definitions/util.ResponseList'
            - properties:
                data:
                  items:
                    $ref: '#/definitions/response.InventoryForms'
                  type: array
              type: object
      summary: 下载库存报表
      tags:
      - 报表
  /api-wms/v1/forms/getHistory:
    post:
      parameters:
@@ -2249,6 +2280,11 @@
  /api-wms/v1/forms/getInventoryForms:
    post:
      parameters:
      - description: token
        in: header
        name: Authorization
        required: true
        type: string
      - description: 查询参数
        in: body
        name: object

 router/router.go

@@ -153,11 +153,12 @@
    reportFormsController := new(controllers.ReportFormsController)
    reportFormsAPI := r.Group(urlPrefix + "/forms")
    {
        reportFormsAPI.POST("getInventoryForms", reportFormsController.GetInventoryForms) //获取库存报表
        reportFormsAPI.POST("getHistory", reportFormsController.GetHistory)               //获取库存历史
        reportFormsAPI.POST("getLocationForms", reportFormsController.GetLocationForms)   //获取位置报表
        reportFormsAPI.POST("monthStats", reportFormsController.MonthStats)               //获取月度统计报表
        reportFormsAPI.POST("doMonthStats", reportFormsController.DoMonthStats)           //获取月度统计报表
        reportFormsAPI.POST("getInventoryForms", reportFormsController.GetInventoryForms)           //获取库存报表
        reportFormsAPI.POST("downloadInventoryForms", reportFormsController.DownloadInventoryForms) //下载库存报表
        reportFormsAPI.POST("getHistory", reportFormsController.GetHistory)                         //获取库存历史
        reportFormsAPI.POST("getLocationForms", reportFormsController.GetLocationForms)             //获取位置报表
        reportFormsAPI.POST("monthStats", reportFormsController.MonthStats)                         //获取月度统计报表
        reportFormsAPI.POST("doMonthStats", reportFormsController.DoMonthStats)                     //获取月度统计报表
    }

    //重订货规则

 service/inventory_report_forms.go

New file
@@ -0,0 +1,234 @@
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)
}