123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518 |
- package api
- import (
- "archive/zip"
- "box-cost/db/model"
- "box-cost/db/repo"
- "box-cost/log"
- "bytes"
- "errors"
- "fmt"
- "io"
- "os"
- "path/filepath"
- "regexp"
- "strings"
- "sync"
- "time"
- "github.com/gin-gonic/gin"
- "github.com/xuri/excelize/v2"
- "go.mongodb.org/mongo-driver/bson"
- "go.mongodb.org/mongo-driver/bson/primitive"
- )
- // TODO 下载代码重复提取
- // 生产计划管理
- func ProductPlan(r *GinRouter) {
- // 创建生产计划
- r.POSTJWT("/plan/create", CreateProductPlan)
- // 获取生产计划详情
- r.GETJWT("/plan/detail/:id", GetProductPlan)
- // 获取生产计划列表
- r.GETJWT("/plan/list", GetProductPlans)
- // 更新生产计划
- r.POSTJWT("/plan/update", UpdateProductPlan)
- // 删除生产计划
- r.POSTJWT("/plan/delete/:id", DelProductPlan)
- // 下载部件单据
- // r.GET("/bill/plan/download", DownLoadCompBills)
- r.GETJWT("/bill/plan/download", DownLoadPlanBills)
- r.GETJWT("/bill/comp/download", DownLoadCompBills)
- r.GETJWT("/bill/comp/downloadPdf", DownLoadCompBillsPdf)
- r.GETJWT("/bill/plan/downloadPdf", DownLoadPlanBillsPdf)
- // 生产成本表
- r.GETJWT("/plan/cost/download", DownLoadPlanCost)
- // 单据批量分配给供应商
- r.GETJWT("/plan/alloc/batch", PlanAllocBatch)
- }
- // 把某个计划的订单批量分配给供应商
- // id 为计划id
- // /plan/alloc/batch?id=xxx
- func PlanAllocBatch(c *gin.Context, apictx *ApiSession) (interface{}, error) {
- // ?验证当前账户是否可发送订单
- userId, _ := primitive.ObjectIDFromHex(apictx.User.Parent)
- user, err1 := getUserById(apictx, userId)
- if err1 != nil {
- return nil, errors.New("用户错误")
- }
- if !isSender(user.Roles) {
- return nil, errors.New("没有发送权限")
- }
- _planId := c.Query("id")
- 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("数据未找到")
- }
- // 获取所有stages单据id
- billIds := make([]string, 0)
- for _, comp := range plan.Pack.Components {
- if comp.Id == "" || len(comp.Stages) == 0 {
- continue
- }
- for _, stage := range comp.Stages {
- billId, _ := primitive.ObjectIDFromHex(stage.BillId)
- if !billId.IsZero() {
- billIds = append(billIds, fmt.Sprintf("%d_%s", stage.BillType, stage.BillId))
- }
- }
- }
- // 去重单据号
- typeBillIds := removeDuplicationSort(billIds)
- if len(typeBillIds) < 1 {
- return nil, errors.New("未找到单据信息")
- }
- var wg sync.WaitGroup
- for _, tId := range typeBillIds {
- var err error
- billType := ""
- tidArr := strings.Split(tId, "_")
- // 采购
- billId, _ := primitive.ObjectIDFromHex(tidArr[1])
- if tidArr[0] == "1" {
- billType = PURCHASE_BILL_TYPE
- // _, err = repo.RepoUpdateSetDoc(apictx.CreateRepoCtx(), repo.CollectionBillPurchase, billId.Hex(), &model.PurchaseBill{IsSend: true, SendTime: time.Now()})
- _, err = repo.RepoUpdateSetDoc1(apictx.CreateRepoCtx(), repo.CollectionBillPurchase, billId.Hex(), &model.PurchaseBill{IsSend: true, SendTime: time.Now()}, &repo.RecordLogReq{
- Path: c.Request.URL.Path,
- UserInfo: user,
- TargetId: billId.Hex(),
- Type: "send",
- })
- }
- // 工艺
- if tidArr[0] == "2" {
- billType = PRODUCE_BILL_TYPE
- // _, err = repo.RepoUpdateSetDoc(apictx.CreateRepoCtx(), repo.CollectionBillProduce, billId.Hex(), &model.ProduceBill{IsSend: true, SendTime: time.Now()})
- _, err = repo.RepoUpdateSetDoc1(apictx.CreateRepoCtx(), repo.CollectionBillProduce, billId.Hex(), &model.ProduceBill{IsSend: true, SendTime: time.Now()}, &repo.RecordLogReq{
- Path: c.Request.URL.Path,
- UserInfo: user,
- TargetId: billId.Hex(),
- Type: "send",
- })
- }
- // 成品采购
- if tidArr[0] == "3" {
- billType = PRODUCT_BILL_TYPE
- // _, err = repo.RepoUpdateSetDoc(apictx.CreateRepoCtx(), repo.CollectionBillProduct, billId.Hex(), &model.ProductBill{IsSend: true, SendTime: time.Now()})
- _, err = repo.RepoUpdateSetDoc1(apictx.CreateRepoCtx(), repo.CollectionBillProduct, billId.Hex(), &model.ProductBill{IsSend: true, SendTime: time.Now()}, &repo.RecordLogReq{
- Path: c.Request.URL.Path,
- UserInfo: user,
- TargetId: billId.Hex(),
- Type: "send",
- })
- }
- if err == nil {
- // 给供应商发送通知短信
- smsInfo, err := genSupplierSmsTemp(billId, billType, apictx)
- fmt.Println(smsInfo)
- if err == nil {
- wg.Add(1)
- go SendSmsNotify(smsInfo.Phone, &SupplierSmsReq{smsInfo.Product, smsInfo.SerialNumber}, &wg)
- // err = SendSmsNotify1(smsInfo.Phone, &SupplierSmsReq{smsInfo.Product, smsInfo.SerialNumber})
- // fmt.Println(err)
- }
- }
- }
- wg.Wait()
- return true, nil
- }
- type UpdateBilltoStageReq struct {
- BillType string
- SupplierInfo *model.Supplier
- Norm string
- Width int
- Height int
- Price2 float64
- OrderCount int
- OrderPrice float64
- Remark string
- ConfirmCount int
- DeliveryTime time.Time
- PrintSize string
- Size string
- IsChangePrice2 bool
- }
- // 更新供应商确定数量与plan中stage项的同步
- func updateBilltoStage(c *gin.Context, planId primitive.ObjectID, idStatges map[string]*UpdateBilltoStageReq, apictx *ApiSession) (interface{}, error) {
- if len(idStatges) == 0 {
- return true, nil
- }
- 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("数据未找到")
- }
- for _, comp := range plan.Pack.Components {
- if comp.Id == "" || len(comp.Stages) == 0 {
- continue
- }
- for _, stage := range comp.Stages {
- if idStatge, ok := idStatges[stage.Id]; ok {
- if idStatge.SupplierInfo != nil {
- stage.SupplierInfo = idStatge.SupplierInfo
- }
- if idStatge.IsChangePrice2 {
- stage.Price = idStatge.Price2
- }
- stage.Norm = idStatge.Norm
- stage.BatchSizeWidth = idStatge.Width
- stage.BatchSizeHeight = idStatge.Height
- stage.OrderCount = idStatge.OrderCount
- stage.OrderPrice = idStatge.OrderPrice
- stage.Remark = idStatge.Remark
- stage.ConfirmCount = idStatge.ConfirmCount
- stage.DeliveryTime = idStatge.DeliveryTime
- stage.Size = idStatge.Size
- stage.UpdateTime = time.Now()
- fmt.Printf("stage信息:%#v\n", stage)
- }
- }
- }
- plan.UpdateTime = time.Now()
- userId, _ := primitive.ObjectIDFromHex(apictx.User.ID)
- user, _ := getUserById(apictx, userId)
- // return repo.RepoUpdateSetDoc(apictx.CreateRepoCtx(), repo.CollectionProductPlan, planId.Hex(), &plan)
- return repo.RepoUpdateSetDoc1(apictx.CreateRepoCtx(), repo.CollectionProductPlan, planId.Hex(), &plan, &repo.RecordLogReq{
- Path: c.Request.URL.Path,
- UserInfo: user,
- TargetId: planId.Hex(),
- Type: "sync",
- })
- }
- // 更新供应商确定数量与plan中stage项的同步
- // func updateStageCount(c *gin.Context, planId primitive.ObjectID, idCounts map[string]int, apictx *ApiSession) (interface{}, error) {
- // if len(idCounts) == 0 {
- // return true, nil
- // }
- // 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("数据未找到")
- // }
- // for _, comp := range plan.Pack.Components {
- // if comp.Id == "" || len(comp.Stages) == 0 {
- // continue
- // }
- // for _, stage := range comp.Stages {
- // if v, ok := idCounts[stage.Id]; ok {
- // stage.ConfirmCount = v
- // }
- // }
- // }
- // plan.UpdateTime = time.Now()
- // userId, _ := primitive.ObjectIDFromHex(apictx.User.ID)
- // user, _ := getUserById(apictx, userId)
- // // return repo.RepoUpdateSetDoc(apictx.CreateRepoCtx(), repo.CollectionProductPlan, planId.Hex(), &plan)
- // return repo.RepoUpdateSetDoc1(apictx.CreateRepoCtx(), repo.CollectionProductPlan, planId.Hex(), &plan, &repo.RecordLogReq{
- // Path: c.Request.URL.Path,
- // UserInfo: user,
- // TargetId: planId.Hex(),
- // Type: "sync",
- // })
- // }
- type SupplierPlanCost struct {
- *model.ProductPlan
- SupplierId string
- }
- func DownLoadPlanCost(c *gin.Context, apictx *ApiSession) (interface{}, error) {
- _planId := c.Query("id")
- _supplierId := c.Query("supplierId")
- supplierId, _ := primitive.ObjectIDFromHex(_supplierId)
- planId, _ := primitive.ObjectIDFromHex(_planId)
- if planId.IsZero() {
- 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("数据未找到")
- }
- summaryPlans := []*PlanSummary{}
- summaryPlan := GetPlanStatus(&plan, apictx)
- summaryPlans = append(summaryPlans, summaryPlan)
- if len(summaryPlans) < 1 {
- return nil, errors.New("数据不存在")
- }
- f := excelize.NewFile()
- index := f.NewSheet("Sheet1")
- f.SetActiveSheet(index)
- f.SetDefaultFont("宋体")
- planSummaryExcel := NewPlanCostExcel(f)
- planSummaryExcel.Content = &SupplierPlanSummary{
- Plans: summaryPlans,
- SupplierId: supplierId,
- }
- // 绘制表格
- planSummaryExcel.Title = fmt.Sprintf("生产成本表(%s)%d盒", plan.Name, plan.Total)
- planSummaryExcel.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, _ := primitive.ObjectIDFromHex(_planId)
- if planId.IsZero() {
- return nil, errors.New("planId错误")
- }
- if len(compId) < 1 {
- 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("数据未找到")
- }
- compBills := make([]string, 0)
- compNameId := ""
- for _, comp := range plan.Pack.Components {
- if comp.Id == "" || len(comp.Stages) == 0 {
- continue
- }
- if compId == comp.Id {
- compNameId = fmt.Sprintf("%s_%s", comp.Name, compId)
- for _, stage := range comp.Stages {
- billId, _ := primitive.ObjectIDFromHex(stage.BillId)
- if !billId.IsZero() {
- compBills = append(compBills, fmt.Sprintf("%d_%s", stage.BillType, stage.BillId))
- }
- }
- }
- }
- if len(compBills) < 1 {
- return nil, errors.New("数据未找到")
- }
- companyName := getCompanyName(apictx)
- typeBillIds := removeDuplicationSort(compBills)
- if len(typeBillIds) < 1 {
- return nil, errors.New("未找到单据信息")
- }
- f := excelize.NewFile()
- f.SetDefaultFont("宋体")
- flagIndex := -1
- // 采购 加工 加工-印刷 加工-覆膜 成品采购
- typeRows := []int{0, 0, 0, 0, 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 {
- continue
- }
- sheetName := "采购单"
- index := f.NewSheet(sheetName)
- if flagIndex < 0 {
- flagIndex = index
- }
- // index := f.NewSheet("采购单")
- // f.SetActiveSheet(index)
- billExcel = NewPurchaseBill(f)
- billExcel.SetSheetName(sheetName)
- 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))
- billExcel.SetRow(typeRows[0])
- billExcel.Draws()
- typeRows[0] = billExcel.GetRow() + 5
- }
- // 工艺
- 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 {
- continue
- }
- sheetName := "加工单"
- if produce.IsPrint {
- sheetName = "加工单-印刷"
- } else if produce.IsLam {
- sheetName = "加工单-覆膜"
- }
- index := f.NewSheet(sheetName)
- if flagIndex < 0 {
- flagIndex = index
- }
- billExcel = NewProduceBill(f)
- billExcel.SetSheetName(sheetName)
- 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 produce.IsPrint {
- billExcel.SetRow(typeRows[2])
- billExcel.Draws()
- typeRows[2] = billExcel.GetRow() + 5
- } else if produce.IsLam {
- billExcel.SetRow(typeRows[3])
- billExcel.Draws()
- typeRows[3] = billExcel.GetRow() + 5
- } else {
- billExcel.SetRow(typeRows[1])
- billExcel.Draws()
- typeRows[1] = billExcel.GetRow() + 5
- }
- }
- // 成品采购
- 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 {
- continue
- }
- sheetName := "成品采购单"
- index := f.NewSheet(sheetName)
- if flagIndex < 0 {
- flagIndex = index
- }
- billExcel = NewProductBill(f)
- billExcel.SetSheetName(sheetName)
- 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(typeRows[4])
- billExcel.Draws()
- typeRows[4] = billExcel.GetRow() + 5
- }
- }
- // 设置活跃sheet
- f.SetActiveSheet(flagIndex)
- // 删除默认Sheet1
- f.DeleteSheet("Sheet1")
- c.Header("Content-Type", "application/octet-stream")
- c.Header("Content-Disposition", "attachment; filename="+fmt.Sprintf("%s.xlsx", compNameId))
- c.Header("Content-Transfer-Encoding", "binary")
- f.Write(c.Writer)
- return nil, nil
- }
- func DownLoadPlanBills(c *gin.Context, apictx *ApiSession) (interface{}, error) {
- _planId := c.Query("id")
- 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("数据未找到")
- }
- // 获取组件中的单据并分工序排列
- // 获取所有stages单据id
- // billIds := make([]string, 0)
- type billIds []string
- compBillsMap := make(map[string]billIds, 0)
- for _, comp := range plan.Pack.Components {
- if comp.Id == "" || len(comp.Stages) == 0 {
- continue
- }
- // 唯一组价名,拼上id是因为防止可能有多个相同组件名
- compNameId := fmt.Sprintf("%s_%s", comp.Name, comp.Id)
- for _, stage := range comp.Stages {
- billId, _ := primitive.ObjectIDFromHex(stage.BillId)
- if !billId.IsZero() {
- compBillsMap[compNameId] = append(compBillsMap[compNameId], fmt.Sprintf("%d_%s", stage.BillType, stage.BillId))
- // billIds = append(billIds, fmt.Sprintf("%d_%s", stage.BillType, stage.BillId))
- }
- }
- }
- if len(compBillsMap) < 1 {
- return nil, errors.New("数据未找到")
- }
- companyName := getCompanyName(apictx)
- _planName := plan.Name
- r := regexp.MustCompile(`/`)
- planName := r.ReplaceAllString(_planName, `&`)
- // fmt.Println(planName)
- // 打包pdf的缓存目录
- saveTmpDir := fmt.Sprintf("tmp1/excel/%s", planName)
- if isExistDir(saveTmpDir) {
- os.RemoveAll(saveTmpDir)
- }
- for compNameId, billIds := range compBillsMap {
- // 去重单据号
- typeBillIds := removeDuplicationSort(billIds)
- if len(typeBillIds) < 1 {
- return nil, errors.New("未找到单据信息")
- }
- f := excelize.NewFile()
- f.SetDefaultFont("宋体")
- flagIndex := -1
- // 采购 加工 加工-印刷 加工-覆膜 成品采购
- typeRows := []int{0, 0, 0, 0, 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 {
- continue
- }
- sheetName := "采购单"
- index := f.NewSheet(sheetName)
- if flagIndex < 0 {
- flagIndex = index
- }
- // index := f.NewSheet("采购单")
- // f.SetActiveSheet(index)
- billExcel = NewPurchaseBill(f)
- billExcel.SetSheetName(sheetName)
- 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))
- billExcel.SetRow(typeRows[0])
- billExcel.Draws()
- typeRows[0] = billExcel.GetRow() + 5
- }
- // 工艺
- 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 {
- continue
- }
- sheetName := "加工单"
- if produce.IsPrint {
- sheetName = "加工单-印刷"
- } else if produce.IsLam {
- sheetName = "加工单-覆膜"
- }
- index := f.NewSheet(sheetName)
- if flagIndex < 0 {
- flagIndex = index
- }
- billExcel = NewProduceBill(f)
- billExcel.SetSheetName(sheetName)
- 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 produce.IsPrint {
- billExcel.SetRow(typeRows[2])
- billExcel.Draws()
- typeRows[2] = billExcel.GetRow() + 5
- } else if produce.IsLam {
- billExcel.SetRow(typeRows[3])
- billExcel.Draws()
- typeRows[3] = billExcel.GetRow() + 5
- } else {
- billExcel.SetRow(typeRows[1])
- billExcel.Draws()
- typeRows[1] = billExcel.GetRow() + 5
- }
- }
- // 成品采购
- 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 {
- continue
- }
- sheetName := "成品采购单"
- index := f.NewSheet(sheetName)
- if flagIndex < 0 {
- flagIndex = index
- }
- billExcel = NewProductBill(f)
- billExcel.SetSheetName(sheetName)
- 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(typeRows[4])
- billExcel.Draws()
- typeRows[4] = billExcel.GetRow() + 5
- }
- }
- // 设置活跃sheet
- f.SetActiveSheet(flagIndex)
- // 删除默认Sheet1
- f.DeleteSheet("Sheet1")
- buf, _ := f.WriteToBuffer()
- targeEXcelName := fmt.Sprintf("%s.xlsx", compNameId)
- err := savePdfToTmp(saveTmpDir, targeEXcelName, buf.Bytes())
- if err != nil {
- fmt.Println("保存文件失败!")
- return nil, err
- }
- }
- c.Header("Content-Type", "application/octet-stream")
- c.Header("Content-Disposition", "attachment; filename="+fmt.Sprintf("%s-execl.zip", planName))
- c.Header("Content-Transfer-Encoding", "binary")
- archive := zip.NewWriter(c.Writer)
- defer archive.Close()
- // 遍历路径信息
- filepath.Walk(saveTmpDir, func(path string, info os.FileInfo, _ error) error {
- // 如果是源路径,提前进行下一个遍历
- if path == saveTmpDir {
- return nil
- }
- // 获取:文件头信息
- header, _ := zip.FileInfoHeader(info)
- header.Name = strings.TrimPrefix(path, saveTmpDir+`/`)
- // 判断:文件是不是文件夹
- if info.IsDir() {
- header.Name += `/`
- } else {
- // 设置:zip的文件压缩算法
- header.Method = zip.Deflate
- }
- // 创建:压缩包头部信息
- writer, _ := archive.CreateHeader(header)
- if !info.IsDir() {
- file, _ := os.Open(path)
- defer file.Close()
- io.Copy(writer, file)
- }
- return nil
- })
- // 删除缓存目录
- os.RemoveAll(saveTmpDir)
- return nil, nil
- // http://127.0.0.1:8888/boxcost/bill/plan/download?id=652206412617b328da71655e
- // http://127.0.0.1:8888/boxcost/bill/comp/download?id=652206412617b328da71655e&compId=1677034398070
- // http://127.0.0.1:8888/boxcost/bill/comp/downloadPdf?id=652206412617b328da71655e&compId=1677034398070
- }
- func DownLoadPlanBillsPdf(c *gin.Context, apictx *ApiSession) (interface{}, error) {
- _planId := c.Query("id")
- 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("数据未找到")
- }
- // 获取所有stages单据id
- billIds := make([]string, 0)
- for _, comp := range plan.Pack.Components {
- if comp.Id == "" || len(comp.Stages) == 0 {
- continue
- }
- for _, stage := range comp.Stages {
- billId, _ := primitive.ObjectIDFromHex(stage.BillId)
- if !billId.IsZero() {
- billIds = append(billIds, fmt.Sprintf("%d_%s", stage.BillType, stage.BillId))
- }
- }
- }
- if len(billIds) < 1 {
- return nil, errors.New("数据未找到")
- }
- // 去重单据号
- typeBillIds := removeDuplicationSort(billIds)
- companyName := getCompanyName(apictx)
- _planName := plan.Name
- r := regexp.MustCompile(`/`)
- planName := r.ReplaceAllString(_planName, `&`)
- // fmt.Println(planName)
- // 打包pdf的缓存目录
- saveTmpDir := fmt.Sprintf("tmp1/%s", planName)
- if isExistDir(saveTmpDir) {
- os.RemoveAll(saveTmpDir)
- }
- // 记录文件数量
- var wg sync.WaitGroup
- c1 := make(chan int)
- for _, tId := range typeBillIds {
- productName := ""
- supplierName := ""
- serialNumber := ""
- f := excelize.NewFile()
- index := f.NewSheet("Sheet1")
- f.SetActiveSheet(index)
- f.SetDefaultFont("宋体")
- 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 {
- continue
- }
- 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)
- }
- }
- productName = purchase.ProductName
- supplierName = purchase.Supplier
- serialNumber = purchase.SerialNumber
- 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 {
- continue
- }
- 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)
- }
- }
- productName = produce.ProductName
- supplierName = produce.Supplier
- serialNumber = produce.SerialNumber
- 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 {
- continue
- }
- billExcel = NewProductBill(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)
- }
- }
- productName = product.ProductName
- supplierName = product.Supplier
- serialNumber = product.SerialNumber
- billExcel.SetContent(&product)
- billExcel.SetTitle(companyName)
- }
- billExcel.SetIsPdf("true")
- billExcel.Draws()
- buf, _ := f.WriteToBuffer()
- // r := regexp.MustCompile(`/`)
- _productName := r.ReplaceAllString(productName, `&`)
- _supplierName := r.ReplaceAllString(supplierName, `&`)
- targePdfName := fmt.Sprintf("%s-%s-%s.pdf", _supplierName, _productName, serialNumber)
- // fmt.Println(targePdfName)
- wg.Add(1)
- go toPdfAndSaveTask(buf, apictx.Svc.Conf.PdfApiAddr, saveTmpDir, targePdfName, c1, &wg)
- }
- go func() {
- // 等待所有goroutine
- wg.Wait()
- // 关闭channel
- close(c1)
- }()
- // channel关闭后结束后停止遍历接收channel中的值
- for n := range c1 {
- if n == -1 {
- return nil, errors.New("下载失败,请重试")
- }
- }
- c.Header("Content-Type", "application/octet-stream")
- c.Header("Content-Disposition", "attachment; filename="+planName+".zip")
- c.Header("Content-Transfer-Encoding", "binary")
- archive := zip.NewWriter(c.Writer)
- defer archive.Close()
- // 遍历路径信息
- filepath.Walk(saveTmpDir, func(path string, info os.FileInfo, _ error) error {
- // 如果是源路径,提前进行下一个遍历
- if path == saveTmpDir {
- return nil
- }
- // 获取:文件头信息
- header, _ := zip.FileInfoHeader(info)
- header.Name = strings.TrimPrefix(path, saveTmpDir+`/`)
- // 判断:文件是不是文件夹
- if info.IsDir() {
- header.Name += `/`
- } else {
- // 设置:zip的文件压缩算法
- header.Method = zip.Deflate
- }
- // 创建:压缩包头部信息
- writer, _ := archive.CreateHeader(header)
- if !info.IsDir() {
- file, _ := os.Open(path)
- defer file.Close()
- io.Copy(writer, file)
- }
- return nil
- })
- // 删除缓存目录
- os.RemoveAll(saveTmpDir)
- return nil, nil
- }
- type ToPdfResult struct {
- IsSucc bool
- Err error
- }
- func toPdfAndSaveTask(buf *bytes.Buffer, toPdfAddr, saveTmpDir, targetPdfName string, toPdfResult chan<- int, wg *sync.WaitGroup) {
- if buf.Len() < 1<<10 {
- fmt.Println("execl内容为空")
- log.Error("execl内容为空")
- toPdfResult <- -1
- wg.Done()
- return
- }
- res, err := excelToPdf(buf, toPdfAddr)
- if err != nil {
- fmt.Println(err)
- log.Error(err)
- // pdfRes := ToPdfResult{
- // IsSucc: false,
- // Err: err,
- // }
- // toPdfResult <- pdfRes
- toPdfResult <- -1
- wg.Done()
- return
- }
- byteData, err := io.ReadAll(res.Body)
- if err != nil {
- fmt.Println(err)
- // pdfRes := ToPdfResult{
- // IsSucc: false,
- // Err: err,
- // }
- // toPdfResult <- pdfRes
- toPdfResult <- -1
- wg.Done()
- return
- }
- if len(byteData) < 1 {
- fmt.Println("pdf内容为空")
- log.Error("pdf内容为空")
- toPdfResult <- -1
- wg.Done()
- return
- }
- defer res.Body.Close()
- err = savePdfToTmp(saveTmpDir, targetPdfName, byteData)
- if err != nil {
- // pdfRes := ToPdfResult{
- // IsSucc: false,
- // Err: err,
- // }
- // toPdfResult <- pdfRes
- toPdfResult <- -1
- wg.Done()
- return
- }
- // pdfRes := ToPdfResult{
- // IsSucc: true,
- // Err: err,
- // }
- // toPdfResult <- pdfRes
- toPdfResult <- 1
- wg.Done()
- }
- func DownLoadCompBillsPdf(c *gin.Context, apictx *ApiSession) (interface{}, error) {
- _planId := c.Query("id")
- compId := c.Query("compId")
- planId, _ := primitive.ObjectIDFromHex(_planId)
- if planId.IsZero() {
- return nil, errors.New("planId错误")
- }
- if len(compId) < 1 {
- return nil, errors.New("compId错误")
- }
- 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("数据未找到")
- }
- // 获取所有stages单据id
- billIds := make([]string, 0)
- compNameId := ""
- for _, comp := range plan.Pack.Components {
- if comp.Id == "" || len(comp.Stages) == 0 {
- continue
- }
- if comp.Id == compId {
- compNameId = fmt.Sprintf("%s_%s", comp.Name, comp.Id)
- for _, stage := range comp.Stages {
- billId, _ := primitive.ObjectIDFromHex(stage.BillId)
- if !billId.IsZero() {
- billIds = append(billIds, fmt.Sprintf("%d_%s", stage.BillType, stage.BillId))
- }
- }
- }
- }
- if len(billIds) < 1 {
- return nil, errors.New("数据未找到")
- }
- // 去重单据号
- typeBillIds := removeDuplicationSort(billIds)
- companyName := getCompanyName(apictx)
- _planName := plan.Name
- r := regexp.MustCompile(`/`)
- planName := r.ReplaceAllString(_planName, `&`)
- // fmt.Println(planName)
- // 打包pdf的缓存目录
- saveTmpDir := fmt.Sprintf("tmp1/%s/%s", planName, compNameId)
- if isExistDir(saveTmpDir) {
- os.RemoveAll(saveTmpDir)
- }
- // 记录文件数量
- var wg sync.WaitGroup
- c1 := make(chan int)
- for _, tId := range typeBillIds {
- productName := ""
- supplierName := ""
- serialNumber := ""
- f := excelize.NewFile()
- index := f.NewSheet("Sheet1")
- f.SetActiveSheet(index)
- f.SetDefaultFont("宋体")
- 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 {
- continue
- }
- 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)
- }
- }
- productName = purchase.ProductName
- supplierName = purchase.Supplier
- serialNumber = purchase.SerialNumber
- 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 {
- continue
- }
- 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)
- }
- }
- productName = produce.ProductName
- supplierName = produce.Supplier
- serialNumber = produce.SerialNumber
- 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 {
- continue
- }
- billExcel = NewProductBill(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)
- }
- }
- productName = product.ProductName
- supplierName = product.Supplier
- serialNumber = product.SerialNumber
- billExcel.SetContent(&product)
- billExcel.SetTitle(companyName)
- }
- billExcel.SetIsPdf("true")
- billExcel.Draws()
- buf, _ := f.WriteToBuffer()
- // r := regexp.MustCompile(`/`)
- _productName := r.ReplaceAllString(productName, `&`)
- _supplierName := r.ReplaceAllString(supplierName, `&`)
- targePdfName := fmt.Sprintf("%s-%s-%s.pdf", _supplierName, _productName, serialNumber)
- // fmt.Println(targePdfName)
- wg.Add(1)
- go toPdfAndSaveTask(buf, apictx.Svc.Conf.PdfApiAddr, saveTmpDir, targePdfName, c1, &wg)
- }
- go func() {
- // 等待所有goroutine
- wg.Wait()
- // 关闭channel
- close(c1)
- }()
- // channel关闭后结束后停止遍历接收channel中的值
- for n := range c1 {
- if n == -1 {
- return nil, errors.New("下载失败,请重试")
- }
- }
- c.Header("Content-Type", "application/octet-stream")
- c.Header("Content-Disposition", "attachment; filename="+fmt.Sprintf("%s.zip", compNameId))
- c.Header("Content-Transfer-Encoding", "binary")
- archive := zip.NewWriter(c.Writer)
- defer archive.Close()
- // 遍历路径信息
- filepath.Walk(saveTmpDir, func(path string, info os.FileInfo, _ error) error {
- // 如果是源路径,提前进行下一个遍历
- if path == saveTmpDir {
- return nil
- }
- // 获取:文件头信息
- header, _ := zip.FileInfoHeader(info)
- header.Name = strings.TrimPrefix(path, saveTmpDir+`/`)
- // 判断:文件是不是文件夹
- if info.IsDir() {
- header.Name += `/`
- } else {
- // 设置:zip的文件压缩算法
- header.Method = zip.Deflate
- }
- // 创建:压缩包头部信息
- writer, _ := archive.CreateHeader(header)
- if !info.IsDir() {
- file, _ := os.Open(path)
- defer file.Close()
- io.Copy(writer, file)
- }
- return nil
- })
- // 删除缓存目录
- os.RemoveAll(saveTmpDir)
- 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("参数错误!")
- }
- if plan.Name == "" {
- return nil, errors.New("生产计划名为空")
- }
- if plan.Total == 0 {
- return nil, errors.New("生产计划数应不为0")
- }
- if plan.TotalPrice == nil {
- var zero float64 = 0
- plan.TotalPrice = &zero
- }
- plan.Status = "process" // 进行中
- plan.CreateTime = time.Now()
- plan.UpdateTime = time.Now()
- userId, _ := primitive.ObjectIDFromHex(apictx.User.ID)
- userInfo, _ := getUserById(apictx, userId)
- result, err := repo.RepoAddDoc1(apictx.CreateRepoCtx(), repo.CollectionProductPlan, &plan, &repo.RecordLogReq{
- Path: c.Request.URL.Path,
- UserInfo: userInfo,
- TargetId: "",
- Type: "created",
- })
- 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("数据未找到")
- }
- billData := map[string]interface{}{}
- 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", "isSend", "reviewed", "isAck", "serialNumber", "sendTo", "remark"}
- })
- if ok {
- billData[stage.BillId] = state
- }
- }
- }
- }
- }
- }
- return map[string]interface{}{
- "plan": plan,
- "billData": billData,
- }, 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")
- }
- if _name, ok := query["name"]; ok {
- delete(query, "name")
- query["name"] = bson.M{"$regex": _name.(string)}
- }
- 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 {
- fmt.Println(err)
- log.Error(err)
- return nil, errors.New("参数错误")
- }
- if plan.Id.Hex() == "" {
- return nil, errors.New("id的为空")
- }
- plan.UpdateTime = time.Now()
- userId, _ := primitive.ObjectIDFromHex(apictx.User.ID)
- user, _ := getUserById(apictx, userId)
- // return repo.RepoUpdateSetDoc(apictx.CreateRepoCtx(), repo.CollectionProductPlan, plan.Id.Hex(), &plan)
- return repo.RepoUpdateSetDoc1(apictx.CreateRepoCtx(), repo.CollectionProductPlan, plan.Id.Hex(), &plan, &repo.RecordLogReq{
- Path: c.Request.URL.Path,
- UserInfo: user,
- TargetId: plan.Id.Hex(),
- Type: "update",
- })
- }
- // 删除生产计划
- func DelProductPlan(c *gin.Context, apictx *ApiSession) (interface{}, error) {
- userId, _ := primitive.ObjectIDFromHex(apictx.User.Parent)
- if userId.IsZero() {
- return nil, errors.New("用户错误,请重新登录")
- }
- _planId := c.Param("id")
- planId, _ := primitive.ObjectIDFromHex(_planId)
- if planId.IsZero() {
- return nil, errors.New("计划id错误")
- }
- 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("计划数据未找到")
- }
- res, err := repo.RepoDeleteDoc(apictx.CreateRepoCtx(), repo.CollectionProductPlan, _planId)
- // 删除计划对应订单
- if err == nil {
- // 获取所有stages单据id
- billIds := make([]string, 0)
- for _, comp := range plan.Pack.Components {
- if comp.Id == "" || len(comp.Stages) == 0 {
- continue
- }
- for _, stage := range comp.Stages {
- billId, _ := primitive.ObjectIDFromHex(stage.BillId)
- if !billId.IsZero() {
- billIds = append(billIds, fmt.Sprintf("%d_%s", stage.BillType, stage.BillId))
- }
- }
- }
- // 去重单据号
- typeBillIds := removeDuplicationSort(billIds)
- if len(typeBillIds) < 1 {
- return res, err
- }
- for _, tId := range typeBillIds {
- tidArr := strings.Split(tId, "_")
- // 采购
- if tidArr[0] == "1" {
- repo.RepoDeleteDoc(apictx.CreateRepoCtx(), repo.CollectionBillPurchase, tidArr[1])
- }
- // 工艺
- if tidArr[0] == "2" {
- repo.RepoDeleteDoc(apictx.CreateRepoCtx(), repo.CollectionBillProduce, tidArr[1])
- }
- // 成品采购
- if tidArr[0] == "3" {
- repo.RepoDeleteDoc(apictx.CreateRepoCtx(), repo.CollectionBillProduct, tidArr[1])
- }
- }
- }
- return res, err
- }
|