~cuijin/WMS.git

			@@ -1,20 +1,29 @@
			package controllers

			import (
			"context"
			"encoding/json"
			"errors"
			"fmt"
			"github.com/gin-gonic/gin"
			"github.com/shopspring/decimal"
			"google.golang.org/grpc"
			"google.golang.org/grpc/credentials/insecure"
			"gorm.io/gorm"
			"os"
			"sort"
			"strconv"
			"time"
			"wms/conf"
			"wms/constvar"
			"wms/extend/code"
			"wms/extend/util"
			"wms/models"
			"wms/opa"
			"wms/pkg/logx"
			"wms/pkg/structx"
			"wms/proto/product_inventory"
			"wms/proto/purchase_wms"
			"wms/request"
			)

			@@ -71,7 +80,7 @@
			}

			func CheckDetailsRepeat(details []*models.OperationDetails) bool {
			detailIDs := []string{}
			var detailIDs []string
			var tempID string
			for _, v := range details {
			detailIDs = append(detailIDs, v.ProductId)
			@@ -128,8 +137,8 @@
			return errors.New("请填入源单号")
			}

			if params.OperationTypeId == 0 {
			return errors.New("operationTypeId为0")
			if params.OperationTypeId == 0 && int(params.BaseOperationType) == 0 {
			return errors.New("未识别该记录类型")
			}

			if params.OperationDate == "" {
			@@ -177,6 +186,9 @@
			search.SetPage(params.Page, params.PageSize)
			if params.Number != "" {
			search.SetKeyword(params.Number)
			}
			if int(params.Status) != 0 {
			search.SetStatus(params.Status)
			}
			list, total, err := search.SetOperationTypeId(params.OperationTypeId).SetPreload(true).SetOrder("created_at desc").Find()
			if err != nil {
			@@ -251,7 +263,7 @@
			util.ResponseFormat(c, code.Success, "修改成功")
			}

			// DeleteDevice
			// Delete
			//
			// @Tags 入库/出库
			// @Summary 删除入库/出库信息
			@@ -293,7 +305,7 @@
			util.ResponseFormat(c, code.Success, "删除成功")
			}

			// DeleteDevice
			// Finish
			//
			// @Tags 入库/出库
			// @Summary 更改记录状态
			@@ -311,7 +323,7 @@
			util.ResponseFormat(c, code.RequestParamError, "id为0")
			return
			}
			operation, err := models.NewOperationSearch().SetID(id).First()
			operation, err := models.NewOperationSearch().SetPreload(true).SetID(id).First()
			if err != nil {
			util.ResponseFormat(c, code.RequestParamError, "未找到相关出入库信息:"+err.Error())
			return
			@@ -320,82 +332,429 @@
			util.ResponseFormat(c, code.RequestError, "该出入库信息无法完成")
			return
			}
			operationType, err := models.NewOperationTypeSearch().SetID(uint(operation.OperationTypeId)).First()
			if err != nil {
			util.ResponseFormat(c, code.RequestError, err.Error())
			return
			}
			if err := models.WithTransaction(func(tx *gorm.DB) error {

			if err := models.NewOperationSearch().SetOrm(tx).SetID(id).Update(&models.Operation{Status: constvar.OperationStatus_Finish}); err != nil {
			return err
			}
			if operationType.BaseOperationType != constvar.BaseOperationTypeInternal {
			var listProdtId []string
			var listProdt []*models.Material
			mapProdt := make(map[string]decimal.Decimal)
			listDetails, err := models.NewOperationDetailsSearch().SetOperationId(operation.Id).FindAll()
			if err != nil {
			return err
			}
			for _, v := range listDetails {
			listProdtId = append(listProdtId, v.ProductId)
			mapProdt[v.ProductId] = v.Amount
			}
			if err := models.NewMaterialSearch().Orm.Where("id IN ?", listProdtId).Find(&listProdt).Error; err != nil {
			return err
			}
			if operationType.BaseOperationType == constvar.BaseOperationTypeIncoming {
			for k, v := range listProdt {
			if value, ok := mapProdt[v.ID]; !ok {
			return errors.New("产品种类异常")
			} else {
			listProdt[k].Amount = listProdt[k].Amount.Add(value)
			if err := tx.Save(listProdt[k]).Error; err != nil {
			return err

			var listProdtId []string
			var listProdt []*models.Material
			mapProdt := make(map[string]decimal.Decimal)
			listDetails, err := models.NewOperationDetailsSearch().SetOperationId(operation.Id).FindAll()
			if err != nil {
			return err
			}
			for _, v := range listDetails {
			listProdtId = append(listProdtId, v.ProductId)
			mapProdt[v.ProductId] = v.Amount
			}

			if err := models.NewMaterialSearch().Orm.Where("id IN ?", listProdtId).Find(&listProdt).Error; err != nil {
			return err
			}

			if operation.BaseOperationType == constvar.BaseOperationTypeIncoming {
			var operationInputs []*models.Operation
			for k, v := range listProdt {
			value, ok := mapProdt[v.ID]
			if !ok {
			return errors.New("产品种类异常")
			}

			listProdt[k].Amount = listProdt[k].Amount.Add(value)
			if err := tx.Save(listProdt[k]).Error; err != nil {
			return err
			}

			locationRule, err := models.NewLocationProductSearch().SetProductId(v.ID).SetAreaId(operation.ToLocationID).First()
			if err != nil && !errors.Is(err, gorm.ErrRecordNotFound) {
			return err
			}

			if err == nil {
			operationTransfer := &models.Operation{
			Number: operation.Number,
			SourceNumber: operation.SourceNumber,
			OperationTypeId: 0,
			OperationTypeName: operation.OperationTypeName,
			Status: constvar.OperationStatus_Finish,
			FromLocationID: locationRule.AreaId,
			ToLocationID: locationRule.LocationId,
			OperationDate: operation.OperationDate,
			ContacterID: operation.ContacterID,
			ContacterName: operation.ContacterName,
			CompanyID: operation.CompanyID,
			CompanyName: operation.CompanyName,
			Comment: operation.Comment,
			BaseOperationType: constvar.BaseOperationTypeInternal,
			Details: []*models.OperationDetails{
			{
			ProductId: v.ID,
			Amount: value,
			},
			},
			}
			operationInputs = append(operationInputs, operationTransfer)

			//if err := tx.Create(&operationTransfer).Error; err != nil {
			// return err
			//}

			//TODO:出入库的finish和报废的finish都要增加对location_product_amount表数量的更新,因为此表有ProductCategory字段，所以operation_details表中要增加ProductCategoryId字段
			locAmount, err := models.NewLocationProductAmountSearch().
			SetProductId(v.ID).
			SetLocationId(locationRule.LocationId).
			First()
			if err != nil && !errors.Is(err, gorm.ErrRecordNotFound) {
			return err
			}
			locAmount.LocationId = locationRule.LocationId
			locAmount.ProductId = v.ID
			locAmount.CreateDate = time.Now().Format("2006-01-02 15:04:05")
			locAmount.Amount = locAmount.Amount.Add(value)
			locAmount.ProductCategoryID = v.CategoryId
			if res := models.NewLocationProductAmountSearch().Orm.Where("id=?", locAmount.ID).Save(locAmount); res.Error != nil {
			return res.Error
			}
			} else {
			locationRule, err = models.NewLocationProductSearch().SetProductCategoryId(v.CategoryId).SetAreaId(operation.ToLocationID).First()
			if err != nil && !errors.Is(err, gorm.ErrRecordNotFound) {
			return err
			}
			if err == nil {
			operationTransfer := &models.Operation{
			Number: operation.Number,
			SourceNumber: operation.SourceNumber,
			OperationTypeId: 0,
			OperationTypeName: operation.OperationTypeName,
			Status: constvar.OperationStatus_Finish,
			FromLocationID: locationRule.AreaId,
			ToLocationID: locationRule.LocationId,
			OperationDate: operation.OperationDate,
			ContacterID: operation.ContacterID,
			ContacterName: operation.ContacterName,
			CompanyID: operation.CompanyID,
			CompanyName: operation.CompanyName,
			Comment: operation.Comment,
			BaseOperationType: constvar.BaseOperationTypeInternal,
			Details: []*models.OperationDetails{
			{
			ProductId: v.ID,
			Amount: value,
			},
			},
			}
			//TODO:出入库的finish和报废的finish都要增加对location_product_amount表数量的更新,因为此表有ProductCategory字段，所以operation_details表中要增加ProductCategoryId字段
			//var locAmount models.LocationProductAmount
			//if err := models.NewLocationProductAmountSearch().Orm.Table("wms_location_produt_amount").Joins("inner join wms_location_product on wms_location_produt.id=wms_location_produt_amount.location_product_id").Where("wms_location_produt.product_id=? and wms_location_produt.location_id=?",operation.ToLocationID,v.ID).First(&locAmount).Error;err!=nil {
			operationInputs = append(operationInputs, operationTransfer)

			//if err := tx.Create(&operationTransfer).Error; err != nil {
			// return err
			//}
			}
			}

			}
			if operationType.BaseOperationType == constvar.BaseOperationTypeOutgoing {
			for k, v := range listProdt {
			if value, ok := mapProdt[v.ID]; !ok {
			return errors.New("产品种类异常")
			} else {
			if v.Amount.LessThan(value) {
			return errors.New(fmt.Sprintf("产品：%v,库存：%v,出库：%v,数量不够，无法完成出库操作", v.Name, v.Amount.String(), value.String()))
			}
			listProdt[k].Amount = listProdt[k].Amount.Sub(value)
			if err := tx.Save(listProdt[k]).Error; err != nil {
			//TODO:出入库的finish和报废的finish都要增加对location_product_amount表数量的更新,因为此表有ProductCategory字段，所以operation_details表中要增加ProductCategoryId字段
			locAmount, err := models.NewLocationProductAmountSearch().
			SetProductId(v.ID).
			SetLocationId(locationRule.LocationId).
			First()
			if err != nil && !errors.Is(err, gorm.ErrRecordNotFound) {
			return err
			}
			locAmount.LocationId = locationRule.LocationId
			locAmount.ProductId = v.ID
			locAmount.CreateDate = time.Now().Format("2006-01-02 15:04:05")
			locAmount.Amount = locAmount.Amount.Add(value)
			locAmount.ProductCategoryID = v.CategoryId
			if res := models.NewLocationProductAmountSearch().Orm.Where("id=?", locAmount.ID).Save(locAmount); res.Error != nil {
			return res.Error
			}
			} else {
			//TODO:出入库的finish和报废的finish都要增加对location_product_amount表数量的更新,因为此表有ProductCategory字段，所以operation_details表中要增加ProductCategoryId字段
			locAmount, err := models.NewLocationProductAmountSearch().
			SetProductId(v.ID).
			SetLocationId(operation.ToLocationID).
			First()
			if err != nil && !errors.Is(err, gorm.ErrRecordNotFound) {
			return err
			}
			locAmount.LocationId = operation.ToLocationID
			locAmount.ProductId = v.ID
			locAmount.CreateDate = time.Now().Format("2006-01-02 15:04:05")
			locAmount.Amount = locAmount.Amount.Add(value)
			locAmount.ProductCategoryID = v.CategoryId
			if res := models.NewLocationProductAmountSearch().Orm.Where("id=?", locAmount.ID).Save(locAmount); res.Error != nil {
			return res.Error
			}
			}
			}
			}
			if len(operationInputs) > 0 {
			if err := tx.Create(&operationInputs).Error; err != nil {
			return err
			}
			}
			if operation.SourceNumber != "" {
			go UpdatePurchaseStatus(operation.SourceNumber)
			}

			}

			if operation.BaseOperationType == constvar.BaseOperationTypeOutgoing {
			for k, v := range listProdt {
			value, ok := mapProdt[v.ID]
			if !ok {
			return errors.New("产品种类异常")
			}
			//todo 演示测试数据
			data, err := os.ReadFile("conf/input.json")
			if err != nil {
			return errors.New("文件读取失败")
			}
			m := make(map[string]interface{})
			err = json.Unmarshal(data, &m)
			if err != nil {
			return errors.New("格式转换失败")
			}
			if opa.OpaCheck(c, m, "operation") {
			if v.Amount.LessThan(value) {
			return errors.New(fmt.Sprintf("产品：%v,库存：%v,出库：%v,数量不够，无法完成出库操作", v.Name, v.Amount.String(), value.String()))
			}
			}
			listProdt[k].Amount = listProdt[k].Amount.Sub(value)
			if err := tx.Save(listProdt[k]).Error; err != nil {
			return err
			}
			locAmount, res := models.NewLocationProductAmountSearch().
			SetProductId(v.ID).
			SetLocationId(operation.ToLocationID).
			FirstRes()
			if res.Error != nil {
			return err
			}
			if locAmount.Amount.LessThan(value) {
			return errors.New(fmt.Sprintf("产品：%v,库存：%v,出库：%v,数量不够，无法完成出库操作", v.Name, v.Amount.String(), value.String()))
			}
			locAmount.Amount = locAmount.Amount.Sub(value)
			if err := models.NewLocationProductAmountSearch().SetID(locAmount.Id).Update(locAmount); err != nil {
			return err
			}
			}
			if operation.SourceNumber != "" {
			go UpdateSalesDetailStatus(operation.SourceNumber)
			}
			}

			if operation.BaseOperationType == constvar.BaseOperationTypeInternal {
			var operationInputs []*models.Operation
			for _, v := range listProdt {
			value, ok := mapProdt[v.ID]
			if !ok {
			return errors.New("产品种类异常")
			}
			//listProdt[k].Amount = listProdt[k].Amount.Add(value)
			//if err := tx.Save(listProdt[k]).Error; err != nil {
			// return err
			//}
			fromLocAmount, res := models.NewLocationProductAmountSearch().
			SetProductId(v.ID).
			SetLocationId(operation.FromLocationID).
			FirstRes()
			if res.Error != nil {
			return err
			}
			if fromLocAmount.Amount.LessThan(value) {
			return errors.New(fmt.Sprintf("产品：%v,库存：%v,调拨：%v,数量不够，无法完成调拨操作", v.Name, v.Amount.String(), value.String()))
			}
			fromLocAmount.Amount = fromLocAmount.Amount.Sub(value)
			if err := models.NewLocationProductAmountSearch().SetID(fromLocAmount.Id).Update(fromLocAmount); err != nil {
			return err
			}

			locationRule, err := models.NewLocationProductSearch().SetProductId(v.ID).SetAreaId(operation.ToLocationID).First()
			if err != nil && !errors.Is(err, gorm.ErrRecordNotFound) {
			return err
			}

			if err == nil {
			operationTransfer := &models.Operation{
			Number: operation.Number,
			SourceNumber: operation.SourceNumber,
			OperationTypeId: 0,
			OperationTypeName: operation.OperationTypeName,
			Status: constvar.OperationStatus_Finish,
			FromLocationID: locationRule.AreaId,
			ToLocationID: locationRule.LocationId,
			OperationDate: operation.OperationDate,
			ContacterID: operation.ContacterID,
			ContacterName: operation.ContacterName,
			CompanyID: operation.CompanyID,
			CompanyName: operation.CompanyName,
			Comment: operation.Comment,
			BaseOperationType: constvar.BaseOperationTypeInternal,
			Details: []*models.OperationDetails{
			{
			ProductId: v.ID,
			Amount: value,
			},
			},
			}
			operationInputs = append(operationInputs, operationTransfer)

			//if err := tx.Create(&operationTransfer).Error; err != nil {
			// return err
			//}

			locAmount, err := models.NewLocationProductAmountSearch().
			SetProductId(v.ID).
			SetLocationId(locationRule.LocationId).
			First()
			if err != nil && !errors.Is(err, gorm.ErrRecordNotFound) {
			return err
			}
			locAmount.LocationId = locationRule.LocationId
			locAmount.ProductId = v.ID
			locAmount.CreateDate = time.Now().Format("2006-01-02 15:04:05")
			locAmount.Amount = locAmount.Amount.Add(value)
			locAmount.ProductCategoryID = v.CategoryId
			if res := models.NewLocationProductAmountSearch().Orm.Where("id=?", locAmount.ID).Save(locAmount); res.Error != nil {
			return res.Error
			}
			} else {
			locationRule, err = models.NewLocationProductSearch().SetProductCategoryId(v.CategoryId).SetAreaId(operation.ToLocationID).First()
			if err != nil && !errors.Is(err, gorm.ErrRecordNotFound) {
			return err
			}
			if err == nil {
			operationTransfer := &models.Operation{
			Number: operation.Number,
			SourceNumber: operation.SourceNumber,
			OperationTypeId: 0,
			OperationTypeName: operation.OperationTypeName,
			Status: constvar.OperationStatus_Finish,
			FromLocationID: locationRule.AreaId,
			ToLocationID: locationRule.LocationId,
			OperationDate: operation.OperationDate,
			ContacterID: operation.ContacterID,
			ContacterName: operation.ContacterName,
			CompanyID: operation.CompanyID,
			CompanyName: operation.CompanyName,
			Comment: operation.Comment,
			BaseOperationType: constvar.BaseOperationTypeInternal,
			Details: []*models.OperationDetails{
			{
			ProductId: v.ID,
			Amount: value,
			},
			},
			}
			operationInputs = append(operationInputs, operationTransfer)

			//if err := tx.Create(&operationTransfer).Error; err != nil {
			// return err
			//}

			locAmount, err := models.NewLocationProductAmountSearch().
			SetProductId(v.ID).
			SetLocationId(locationRule.LocationId).
			First()
			if err != nil && !errors.Is(err, gorm.ErrRecordNotFound) {
			return err
			}
			locAmount.LocationId = locationRule.LocationId
			locAmount.ProductId = v.ID
			locAmount.CreateDate = time.Now().Format("2006-01-02 15:04:05")
			locAmount.Amount = locAmount.Amount.Add(value)
			locAmount.ProductCategoryID = v.CategoryId
			if res := models.NewLocationProductAmountSearch().Orm.Where("id=?", locAmount.ID).Save(locAmount); res.Error != nil {
			return res.Error
			}
			} else {
			locAmount, err := models.NewLocationProductAmountSearch().
			SetProductId(v.ID).
			SetLocationId(operation.ToLocationID).
			First()
			if err != nil && !errors.Is(err, gorm.ErrRecordNotFound) {
			return err
			}
			locAmount.LocationId = operation.ToLocationID
			locAmount.ProductId = v.ID
			locAmount.CreateDate = time.Now().Format("2006-01-02 15:04:05")
			locAmount.Amount = locAmount.Amount.Add(value)
			if res := models.NewLocationProductAmountSearch().Orm.Where("id=?", locAmount.ID).Save(locAmount); res.Error != nil {
			return res.Error
			}
			}
			}
			}
			if len(operationInputs) > 0 {
			if err := tx.Create(&operationInputs).Error; err != nil {
			return err
			}
			}
			}
			return nil
			}); err != nil {
			util.ResponseFormat(c, code.RequestError, err.Error())
			return
			}

			util.ResponseFormat(c, code.Success, "操作成功")
			}

			// ListAll
			var (
			ProductInventoryServiceConn *grpc.ClientConn
			PurchaseServiceConn *grpc.ClientConn
			)

			func InitProductInventoryServiceConn() {
			var err error
			ProductInventoryServiceConn, err = grpc.Dial(conf.GrpcServerConf.CrmAddr, grpc.WithTransportCredentials(insecure.NewCredentials()))
			if err != nil {
			logx.Errorf("grpc dial product service error: %v", err.Error())
			return
			}
			PurchaseServiceConn, err = grpc.Dial(conf.GrpcServerConf.SrmAddr, grpc.WithTransportCredentials(insecure.NewCredentials()))
			if err != nil {
			logx.Errorf("grpc dial product service error: %v", err.Error())
			return
			}
			}

			func CloseProductInventoryServiceConn() {
			if ProductInventoryServiceConn != nil {
			ProductInventoryServiceConn.Close()
			}
			if PurchaseServiceConn != nil {
			PurchaseServiceConn.Close()
			}
			}

			func UpdateSalesDetailStatus(number string) {
			client := product_inventory.NewProductInventoryServiceClient(ProductInventoryServiceConn)
			_, err := client.UpdateSalesDetailStatus(context.Background(), &product_inventory.UpdateSalesDetailStatusRequest{
			Number: number,
			SalesDetailStatus: "已出库",
			})
			if err != nil {
			logx.Errorf("grpc dial UpdateSalesDetailStatus service error: %v", err)
			}
			}

			func UpdatePurchaseStatus(number string) {
			client := purchase_wms.NewPurchaseServiceClient(PurchaseServiceConn)
			_, err := client.UpdatePurchaseStatus(context.Background(), &purchase_wms.UpdatePurchaseStatusRequest{Number: number})
			if err != nil {
			logx.Errorf("grpc dial UpdatePurchaseStatus service error: %v", err)
			}
			}

			// ListTransfer
			// @Tags 入库/出库
			// @Summary 调拨
			// @Summary 库存调拨列表
			// @Produce application/json
			// @Param object body request.OperationAllList true "参数"
			// @Success 200 {object} util.Response "成功"
			// @Router /api-wms/v1/operation/listAll [post]
			func (slf OperationController) ListAll(c *gin.Context) {
			// @Router /api-wms/v1/operation/listTransfer [post]
			func (slf OperationController) ListTransfer(c *gin.Context) {
			var params request.OperationAllList
			if err := c.BindJSON(&params); err != nil {
			util.ResponseFormat(c, code.RequestParamError, "参数解析失败，数据类型错误:"+err.Error())
			@@ -411,10 +770,60 @@
			if params.Number != "" {
			search.SetKeyword(params.Number)
			}
			list, total, err := search.SetPreload(true).SetOrder("created_at desc").Find()
			list, total, err := search.SetPreload(true).SetBaseOperationType(constvar.BaseOperationTypeInternal).SetOrder("created_at desc").Find()
			if err != nil {
			util.ResponseFormat(c, code.RequestError, "查找失败:"+err.Error())
			return
			}
			util.ResponseFormatListWithPage(c, code.Success, list, int(total), params.Page, params.PageSize)
			}

			// GetLogisticCompanyList
			// @Tags 入库/出库
			// @Summary 获取物流公司列表
			// @Produce application/json
			// @Success 200 {object} util.ResponseList{data=[]models.LogisticCompany} "成功"
			// @Router /api-wms/v1/operation/getLogisticCompanyList [get]
			func (slf OperationController) GetLogisticCompanyList(c *gin.Context) {
			companies, err := models.NewLogisticCompanySearch().FindNotTotal()
			if err != nil {
			util.ResponseFormat(c, code.RequestError, "查找失败:"+err.Error())
			return
			}
			util.ResponseFormat(c, code.Success, companies)
			}

			// Cancel
			//
			// @Tags 入库/出库
			// @Summary 取消
			// @Produce application/json
			// @Param id path int true "id"
			// @Success 200 {object} util.Response "成功"
			// @Router /api-wms/v1/operation/cancel/{id} [put]
			func (slf OperationController) Cancel(c *gin.Context) {
			id, err := strconv.Atoi(c.Param("id"))
			if err != nil {
			util.ResponseFormat(c, code.RequestParamError, "错误的id值")
			return
			}
			if id == 0 {
			util.ResponseFormat(c, code.RequestParamError, "id为0")
			return
			}
			operation, err := models.NewOperationSearch().SetPreload(true).SetID(id).First()
			if err != nil {
			util.ResponseFormat(c, code.RequestParamError, "未找到相关出入库信息:"+err.Error())
			return
			}
			if operation.Status != constvar.OperationStatus_Ready {
			util.ResponseFormat(c, code.RequestError, "该出入库信息无法取消")
			return
			}
			operation.Status = constvar.OperationStatus_Cancel
			if err := models.NewOperationSearch().SetID(operation.Id).Save(operation); err != nil {
			util.ResponseFormat(c, code.SaveFail, err.Error())
			return
			}
			util.ResponseFormat(c, code.Success, "操作成功")
			}