package controllers import ( "errors" "github.com/gin-gonic/gin" "google.golang.org/grpc" "google.golang.org/grpc/credentials/insecure" "gorm.io/gorm" "sort" "strconv" "time" "wms/conf" "wms/constvar" "wms/extend/code" "wms/extend/util" "wms/models" "wms/pkg/logx" "wms/pkg/structx" "wms/proto/product_inventory" "wms/request" ) type OperationController struct { } // Add // @Tags 入库/出库 // @Summary 添加入库/出库 // @Produce application/json // @Param object body request.AddOperation true "入库/出库信息" // @Success 200 {object} util.Response "成功" // @Router /api-wms/v1/operation/operation [post] func (slf OperationController) Add(c *gin.Context) { var reqParams request.AddOperation var params models.Operation if err := c.BindJSON(&reqParams); err != nil { util.ResponseFormat(c, code.RequestParamError, "参数解析失败,数据类型错误") return } if err := structx.AssignTo(reqParams, ¶ms); err != nil { util.ResponseFormat(c, code.RequestParamError, "数据转换错误"+err.Error()) return } if err := slf.CheckParams(params); err != nil { util.ResponseFormat(c, code.RequestParamError, err.Error()) return } if err := slf.FormatLocation(¶ms); err != nil { util.ResponseFormat(c, code.RequestParamError, err.Error()) return } if CheckDetailsRepeat(params.Details) { util.ResponseFormat(c, code.RequestParamError, "明细中不能存在重复的产品") return } operationType, err := models.NewOperationTypeSearch().SetID(uint(params.OperationTypeId)).First() if err != nil { util.ResponseFormat(c, code.RequestParamError, err.Error()) return } params.Status = constvar.OperationStatus_Ready params.Number = strconv.FormatInt(time.Now().Unix(), 10) params.BaseOperationType = operationType.BaseOperationType if err := models.NewOperationSearch().Create(¶ms); err != nil { logx.Errorf("Operation create err: %v", err) util.ResponseFormat(c, code.SaveFail, "添加失败:"+err.Error()) return } util.ResponseFormat(c, code.Success, "添加成功") } func CheckDetailsRepeat(details []*models.OperationDetails) bool { var detailIDs []string var tempID string for _, v := range details { detailIDs = append(detailIDs, v.ProductId) } sort.Strings(detailIDs) for _, v := range detailIDs { if v != tempID { tempID = v } else { return true } } return false } func (slf OperationController) FormatLocation(params *models.Operation) error { operationType, err := models.NewOperationTypeSearch().SetID(uint(params.OperationTypeId)).First() if err != nil { return err } if operationType.BaseOperationType == constvar.BaseOperationTypeIncoming { if location, err := models.NewLocationSearch().SetType(int(constvar.LocationTypeVendor)).First(); err != nil { return err } else { params.FromLocationID = location.Id } if params.ToLocationID == 0 { return errors.New("请选择目标位置") } } if operationType.BaseOperationType == constvar.BaseOperationTypeOutgoing { if location, err := models.NewLocationSearch().SetType(int(constvar.LocationTypeCustomer)).First(); err != nil { return err } else { params.ToLocationID = location.Id } if params.FromLocationID == 0 { return errors.New("请选择源位置") } } if operationType.BaseOperationType == constvar.BaseOperationTypeInternal { if params.ToLocationID == 0 { return errors.New("请选择目标位置") } if params.FromLocationID == 0 { return errors.New("请选择源位置") } } return nil } func (slf OperationController) CheckParams(params models.Operation) error { if params.SourceNumber == "" { return errors.New("请填入源单号") } if params.OperationTypeId == 0 && int(params.BaseOperationType) == 0 { return errors.New("未识别该记录类型") } if params.OperationDate == "" { return errors.New("请选择安排日期") } if len(params.Details) <= 0 { return errors.New("请添加明细信息") } //检查明细部分 for _, v := range params.Details { if v.ProductId == "" { return errors.New("productID为空") } //if v.ProductName == "" { // return errors.New("产品名称异常") //} if v.Amount.IsNegative() { return errors.New("产品数量出错") } } return nil } // List // @Tags 入库/出库 // @Summary 入库/出库列表 // @Produce application/json // @Param object body request.OperationList true "查询参数" // @Success 200 {object} util.Response "成功" // @Router /api-wms/v1/operation/list [post] func (slf OperationController) List(c *gin.Context) { var params request.OperationList if err := c.BindJSON(¶ms); err != nil { util.ResponseFormat(c, code.RequestParamError, "参数解析失败,数据类型错误:"+err.Error()) return } if err := slf.CheckListParams(¶ms); err != nil { util.ResponseFormat(c, code.RequestParamError, err.Error()) return } search := models.NewOperationSearch() 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 { util.ResponseFormat(c, code.RequestError, "查找失败:"+err.Error()) return } util.ResponseFormatListWithPage(c, code.Success, list, int(total), params.Page, params.PageSize) } func (slf OperationController) CheckListParams(params *request.OperationList) error { if !params.PageInfo.Check() { return errors.New("数据分页信息错误") } if params.OperationTypeId == 0 { return errors.New("operationTypeId为0") } return nil } // Update // @Tags 入库/出库 // @Summary 修改入库/出库信息 // @Produce application/json // @Param object body request.UpdateOperation true "入库信息" // @Success 200 {object} util.Response "成功" // @Router /api-wms/v1/operation/update [post] func (slf OperationController) Update(c *gin.Context) { var reqParams request.UpdateOperation var params models.Operation if err := c.BindJSON(&reqParams); err != nil { util.ResponseFormat(c, code.RequestParamError, "参数解析失败,数据类型错误:"+err.Error()) return } if reqParams.Status != constvar.OperationStatus_Ready { util.ResponseFormat(c, code.RequestParamError, "该入库信息已完成,无法进行修改") return } if err := structx.AssignTo(reqParams, ¶ms); err != nil { util.ResponseFormat(c, code.RequestParamError, "数据转换错误"+err.Error()) return } if err := slf.CheckParams(params); err != nil { util.ResponseFormat(c, code.RequestParamError, err.Error()) return } if CheckDetailsRepeat(params.Details) { util.ResponseFormat(c, code.RequestParamError, "明细中不能存在重复的产品") return } if err := slf.FormatLocation(¶ms); err != nil { util.ResponseFormat(c, code.RequestParamError, err.Error()) return } if err := models.WithTransaction(func(tx *gorm.DB) error { if err := models.NewOperationDetailsSearch().SetOrm(tx).SetOperationId(params.Id).Delete(); err != nil { return err } operationSearch := models.NewOperationSearch().SetOrm(tx) if err := operationSearch.Orm.Model(¶ms).Association("Details").Replace(params.Details); err != nil { return err } if err := models.NewOperationSearch().SetOrm(tx).SetID(params.Id).Save(¶ms); err != nil { return err } return nil }); err != nil { util.ResponseFormat(c, code.RequestParamError, "修改失败:"+err.Error()) return } util.ResponseFormat(c, code.Success, "修改成功") } // Delete // // @Tags 入库/出库 // @Summary 删除入库/出库信息 // @Produce application/json // @Param id path int true "id" // @Success 200 {object} util.Response "成功" // @Router /api-wms/v1/operation/operation/{id} [delete] func (slf OperationController) Delete(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().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 } if err := models.WithTransaction(func(tx *gorm.DB) error { if err := models.NewOperationDetailsSearch().SetOrm(tx).SetOperationId(id).Delete(); err != nil { return err } if err := models.NewOperationSearch().SetOrm(tx).SetID(id).Delete(); err != nil { return err } return nil }); err != nil { util.ResponseFormat(c, code.RequestParamError, "修改失败:"+err.Error()) return } util.ResponseFormat(c, code.Success, "删除成功") } // Finish // // @Tags 入库/出库 // @Summary 更改记录状态 // @Produce application/json // @Param id path int true "id" // @Success 200 {object} util.Response "成功" // @Router /api-wms/v1/operation/finish/{id} [put] func (slf OperationController) Finish(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 //} //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 // } // // 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) // 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 // //} // // //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) // 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) // 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.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.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) // 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) // 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 //} //if operation.SourceNumber != "" { // go UpdateSalesDetailStatus(c, operation.SourceNumber) //} UpdateSalesDetailStatus(c, c.Param("id")) util.ResponseFormat(c, code.Success, "操作成功") } var ProductInventoryServiceConn *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 } } func CloseProductInventoryServiceConn() { if ProductInventoryServiceConn != nil { ProductInventoryServiceConn.Close() } } func UpdateSalesDetailStatus(ctx *gin.Context, number string) { client := product_inventory.NewProductInventoryServiceClient(ProductInventoryServiceConn) _, err := client.UpdateSalesDetailStatus(ctx, &product_inventory.UpdateSalesDetailStatusRequest{ Number: number, SalesDetailStatus: "已出库", }) if err != nil { logx.Errorf("grpc dial UpdateSalesDetailStatus service error: %v", err) } } // ListTransfer // @Tags 入库/出库 // @Summary 库存调拨列表 // @Produce application/json // @Param object body request.OperationAllList true "参数" // @Success 200 {object} util.Response "成功" // @Router /api-wms/v1/operation/listTransfer [post] func (slf OperationController) ListTransfer(c *gin.Context) { var params request.OperationAllList if err := c.BindJSON(¶ms); err != nil { util.ResponseFormat(c, code.RequestParamError, "参数解析失败,数据类型错误:"+err.Error()) return } if !params.PageInfo.Check() { util.ResponseFormat(c, code.RequestParamError, "数据分页信息错误") return } search := models.NewOperationSearch() search.SetPage(params.Page, params.PageSize) search.SetPage(params.Page, params.PageSize) if params.Number != "" { search.SetKeyword(params.Number) } 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) }