package api import ( "box-cost/db/model" "box-cost/db/repo" "box-cost/log" "errors" "fmt" "strings" "time" "github.com/gin-gonic/gin" "github.com/xuri/excelize/v2" "go.mongodb.org/mongo-driver/bson" "go.mongodb.org/mongo-driver/bson/primitive" ) // 生产计划管理 func ProductPlan(r *GinRouter) { // 创建生产计划 r.POST("/plan/create", CreateProductPlan) // 获取生产计划详情 r.GET("/plan/detail/:id", GetProductPlan) // 获取生产计划列表 r.GET("/plan/list", GetProductPlans) // 更新生产计划 r.POST("/plan/update", UpdateProductPlan) // 删除生产计划 r.POST("/plan/delete/:id", DelProductPlan) // 下载部件单据 r.GET("/bill/plan/download", DownLoadCompBills) // 生产成本表 r.GET("/plan/cost/download", DownLoadPlanCost) } type SupplierPlanCost struct { *model.ProductPlan SupplierId string } func DownLoadPlanCost(c *gin.Context, apictx *ApiSession) (interface{}, error) { _planId := c.Query("id") supplierId := c.Query("supplierId") planId, _ := primitive.ObjectIDFromHex(_planId) if planId.IsZero() { return nil, errors.New("planId错误") } supplierPlanCost := &SupplierPlanCost{} plan := model.ProductPlan{} found, err := repo.RepoSeachDoc(apictx.CreateRepoCtx(), &repo.DocSearchOptions{ CollectName: repo.CollectionProductPlan, Query: repo.Map{"_id": planId}, }, &plan) if !found || err != nil { return nil, errors.New("数据未找到") } supplierPlanCost.ProductPlan = &plan supplierPlanCost.SupplierId = supplierId f := excelize.NewFile() index := f.NewSheet("Sheet1") f.SetActiveSheet(index) f.SetDefaultFont("宋体") planCostExcel := NewPlanCostExcel(f) planCostExcel.Title = fmt.Sprintf("生产成本表(%s)%d盒", plan.Name, plan.Total) planCostExcel.Content = supplierPlanCost planCostExcel.Draws() c.Header("Content-Type", "application/octet-stream") c.Header("Content-Disposition", "attachment; filename="+"planCost.xlsx") c.Header("Content-Transfer-Encoding", "binary") err = f.Write(c.Writer) if err != nil { return nil, err } return nil, nil } func DownLoadCompBills(c *gin.Context, apictx *ApiSession) (interface{}, error) { _planId := c.Query("id") compId := c.Query("compId") planId, err := primitive.ObjectIDFromHex(_planId) if err != nil { return nil, errors.New("planId错误") } plan := model.ProductPlan{} found, err := repo.RepoSeachDoc(apictx.CreateRepoCtx(), &repo.DocSearchOptions{ CollectName: repo.CollectionProductPlan, Query: repo.Map{"_id": planId}, }, &plan) if !found || err != nil { return nil, errors.New("数据未找到") } // 获取部件单据 curComp := &model.PackComponent{} for _, comp := range plan.Pack.Components { if comp.Id == compId { curComp = comp } } if curComp.Id == "" { return nil, errors.New("该组件不存在") } // 获取bill if len(curComp.Stages) == 0 { return nil, errors.New("该组件数据不存在") } // 获取不同类型的单据id // set := make(map[int]map[primitive.ObjectID]struct{}) billIds := make([]string, 0) for _, stage := range curComp.Stages { _billId := stage.BillId billId, _ := primitive.ObjectIDFromHex(_billId) if !billId.IsZero() { // set[stage.Type] = map[primitive.ObjectID]struct{}{billId: struct{}{}} billIds = append(billIds, fmt.Sprintf("%d_%s", stage.BillType, stage.BillId)) } } typeBillIds := removeDuplicationSort(billIds) if len(typeBillIds) < 1 { return nil, errors.New("未找到单据信息") } f := excelize.NewFile() index := f.NewSheet("Sheet1") f.SetActiveSheet(index) f.SetDefaultFont("宋体") companyName := getCompanyName(apictx) row := 0 for _, tId := range typeBillIds { tidArr := strings.Split(tId, "_") var billExcel IExcel // 采购 billId, _ := primitive.ObjectIDFromHex(tidArr[1]) if tidArr[0] == "1" { purchase := model.PurchaseBill{} found, _ := repo.RepoSeachDoc(apictx.CreateRepoCtx(), &repo.DocSearchOptions{ CollectName: repo.CollectionBillPurchase, Query: repo.Map{"_id": billId}, }, &purchase) if found { billExcel = NewPurchaseBill(f) if purchase.Reviewed == 1 { if len(purchase.SignUsers) > 0 { signs := []*model.Signature{} repo.RepoDocsSearch(apictx.CreateRepoCtx(), &repo.PageSearchOptions{ CollectName: repo.CollectionSignature, Query: repo.Map{"_id": bson.M{"$in": purchase.SignUsers}}, Sort: bson.M{"sort": 1}, // 升序 }, &signs) billExcel.SetSignatures(signs) } } billExcel.SetContent(&purchase) billExcel.SetTitle(fmt.Sprintf("%s原材料采购单", companyName)) } } // 工艺 if tidArr[0] == "2" { produce := model.ProduceBill{} found, _ := repo.RepoSeachDoc(apictx.CreateRepoCtx(), &repo.DocSearchOptions{ CollectName: repo.CollectionBillProduce, Query: repo.Map{"_id": billId}, }, &produce) if found { billExcel = NewProduceBill(f) if produce.Reviewed == 1 { if len(produce.SignUsers) > 0 { signs := []*model.Signature{} repo.RepoDocsSearch(apictx.CreateRepoCtx(), &repo.PageSearchOptions{ CollectName: repo.CollectionSignature, Query: repo.Map{"_id": bson.M{"$in": produce.SignUsers}}, Sort: bson.M{"sort": 1}, // 升序 }, &signs) billExcel.SetSignatures(signs) } } billExcel.SetContent(&produce) billExcel.SetTitle(fmt.Sprintf("%s加工单", companyName)) } } // 成品采购 if tidArr[0] == "3" { product := model.ProductBill{} found, _ := repo.RepoSeachDoc(apictx.CreateRepoCtx(), &repo.DocSearchOptions{ CollectName: repo.CollectionBillProduct, Query: repo.Map{"_id": billId}, }, &product) if found { billExcel = NewProduceBill(f) if product.Reviewed == 1 { if len(product.SignUsers) > 0 { signs := []*model.Signature{} repo.RepoDocsSearch(apictx.CreateRepoCtx(), &repo.PageSearchOptions{ CollectName: repo.CollectionSignature, Query: repo.Map{"_id": bson.M{"$in": product.SignUsers}}, Sort: bson.M{"sort": 1}, // 升序 }, &signs) billExcel.SetSignatures(signs) } } billExcel.SetContent(&product) billExcel.SetTitle(companyName) } } billExcel.SetRow(row) billExcel.Draws() fmt.Println(billExcel.GetRow()) row = billExcel.GetRow() + 5 } c.Header("Content-Type", "application/octet-stream") c.Header("Content-Disposition", "attachment; filename="+"bill.xlsx") c.Header("Content-Transfer-Encoding", "binary") err = f.Write(c.Writer) if err != nil { return nil, err } return nil, nil } // 创建生产计划 func CreateProductPlan(c *gin.Context, apictx *ApiSession) (interface{}, error) { var plan model.ProductPlan err := c.ShouldBindJSON(&plan) if err != nil { fmt.Println(err) return nil, errors.New("参数错误!") } ctx := apictx.CreateRepoCtx() if plan.Name == "" { return nil, errors.New("生产计划名为空") } if plan.Total == 0 { return nil, errors.New("生产计划数应不为0") } plan.Status = "process" // 进行中 plan.CreateTime = time.Now() plan.UpdateTime = time.Now() result, err := repo.RepoAddDoc(ctx, repo.CollectionProductPlan, &plan) return result, err } // 获取生产计划信息 func GetProductPlan(c *gin.Context, apictx *ApiSession) (interface{}, error) { planId := c.Param("id") id, err := primitive.ObjectIDFromHex(planId) if err != nil { return nil, errors.New("非法id") } var plan model.ProductPlan option := &repo.DocSearchOptions{ CollectName: repo.CollectionProductPlan, Query: repo.Map{"_id": id}, } found, err := repo.RepoSeachDoc(apictx.CreateRepoCtx(), option, &plan) if !found || err != nil { log.Info(err) return nil, errors.New("数据未找到") } billStates := map[string]string{} if plan.Pack != nil && plan.Pack.Components != nil { for _, comp := range plan.Pack.Components { if comp.Stages != nil { for _, stage := range comp.Stages { if len(stage.BillId) > 0 { collectName := "" // 材料 if stage.BillType == 1 { collectName = repo.CollectionBillPurchase } // 工艺 if stage.BillType == 2 { collectName = repo.CollectionBillProduce } // 成品 if stage.BillType == 3 { collectName = repo.CollectionBillProduct } ok, state := repo.RepoSeachDocMap(apictx.CreateRepoCtx(), &repo.DocSearchOptions{CollectName: collectName, Query: repo.Map{"_id": stage.BillId}, Project: []string{"status"}}) if ok { billStates[stage.BillId] = state["status"].(string) } } } } } } return map[string]interface{}{ "plan": plan, "billStates": billStates, }, nil } // 获取生产计划列表 func GetProductPlans(c *gin.Context, apictx *ApiSession) (interface{}, error) { page, size, query := UtilQueryPageSize(c) if _packId, ok := query["packId"]; ok { packId, _ := primitive.ObjectIDFromHex(_packId.(string)) query["pack._id"] = packId delete(query, "packId") } option := &repo.PageSearchOptions{ CollectName: repo.CollectionProductPlan, Query: query, Page: page, Size: size, Sort: bson.M{"createTime": -1}, Project: []string{"_id", "thumbnail", "name", "updateTime", "createTime", "createUser", "total", "totalPrice", "status"}, } return repo.RepoPageSearch(apictx.CreateRepoCtx(), option) } // 更新生产计划 func UpdateProductPlan(c *gin.Context, apictx *ApiSession) (interface{}, error) { var plan model.ProductPlan err := c.ShouldBindJSON(&plan) if err != nil { return nil, errors.New("参数错误") } if plan.Id.Hex() == "" { return nil, errors.New("id的为空") } plan.UpdateTime = time.Now() return repo.RepoUpdateSetDoc(apictx.CreateRepoCtx(), repo.CollectionProductPlan, plan.Id.Hex(), &plan) } // 删除生产计划 func DelProductPlan(c *gin.Context, apictx *ApiSession) (interface{}, error) { planId := c.Param("id") if planId == "" { return nil, errors.New("id为空") } return repo.RepoDeleteDoc(apictx.CreateRepoCtx(), repo.CollectionProductPlan, planId) }