report.go 27 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947
  1. package api
  2. import (
  3. "archive/zip"
  4. "box-cost/db/model"
  5. "box-cost/db/repo"
  6. "errors"
  7. "fmt"
  8. "io"
  9. "os"
  10. "path/filepath"
  11. "regexp"
  12. "strings"
  13. "sync"
  14. "time"
  15. "github.com/gin-gonic/gin"
  16. "github.com/go-redis/redis/v8"
  17. "github.com/xuri/excelize/v2"
  18. "go.mongodb.org/mongo-driver/bson"
  19. "go.mongodb.org/mongo-driver/bson/primitive"
  20. )
  21. // 统计报表 按时间范围,供应商 包装-计划(多选) 维度进行过滤,形成报表。可以下载
  22. func Report(r *GinRouter) {
  23. // 加工列表
  24. r.GETJWT("/report/produce/list", ReportProduceList)
  25. // 采购列表
  26. r.GETJWT("/report/purchase/list", ReportPurchaseList)
  27. r.GETJWT("/report/product/list", ReportProductList)
  28. r.GETJWT("/report/list", ReportList)
  29. r.GETJWT("/report/download", ReportListDownload)
  30. //
  31. r.GETJWT("/report/bills/download", ReportBillsDownload1)
  32. }
  33. // 下载单据
  34. func ReportBillsDownload1(c *gin.Context, apictx *ApiSession) (interface{}, error) {
  35. _, _, query := UtilQueryPageSize(c)
  36. filtter := handleReportQuery(query)
  37. purchases := []*model.PurchaseBill{}
  38. produces := []*model.ProduceBill{}
  39. products := []*model.ProductBill{}
  40. repo.RepoDocsSearch(apictx.CreateRepoCtx(), &repo.PageSearchOptions{
  41. CollectName: repo.CollectionBillPurchase,
  42. Query: filtter,
  43. }, &purchases)
  44. repo.RepoDocsSearch(apictx.CreateRepoCtx(), &repo.PageSearchOptions{
  45. CollectName: repo.CollectionBillProduce,
  46. Query: filtter,
  47. }, &produces)
  48. repo.RepoDocsSearch(apictx.CreateRepoCtx(), &repo.PageSearchOptions{
  49. CollectName: repo.CollectionBillProduct,
  50. Query: filtter,
  51. }, &products)
  52. // 加入redis有序集合中进行排序
  53. // 把对应type_id:对象存入map中避免再次查询表
  54. typeBills := map[string]interface{}{}
  55. redisCli := apictx.Svc.Redis
  56. reportBillKey := "report-bill-list:" + apictx.User.Parent
  57. isExist := redisCli.Exists(apictx.CreateRepoCtx().Ctx, reportBillKey).Val()
  58. fmt.Println("是否存在:", isExist)
  59. // 不存在这个key时
  60. if isExist < 1 {
  61. if len(purchases) > 0 {
  62. for _, purchase := range purchases {
  63. member := "purchase_" + purchase.Id.Hex()
  64. score := purchase.CompleteTime.Unix()
  65. redisCli.ZAdd(apictx.CreateRepoCtx().Ctx, reportBillKey, &redis.Z{Score: float64(score), Member: member})
  66. typeBills[member] = purchase
  67. }
  68. }
  69. if len(produces) > 0 {
  70. for _, produce := range produces {
  71. member := "produce_" + produce.Id.Hex()
  72. score := produce.CompleteTime.Unix()
  73. redisCli.ZAdd(apictx.CreateRepoCtx().Ctx, reportBillKey, &redis.Z{Score: float64(score), Member: member})
  74. typeBills[member] = produce
  75. }
  76. }
  77. if len(products) > 0 {
  78. for _, product := range products {
  79. member := "product_" + product.Id.Hex()
  80. score := product.CompleteTime.Unix()
  81. redisCli.ZAdd(apictx.CreateRepoCtx().Ctx, reportBillKey, &redis.Z{Score: float64(score), Member: member})
  82. typeBills[member] = product
  83. }
  84. }
  85. // 设置过期时间
  86. redisCli.Expire(apictx.CreateRepoCtx().Ctx, reportBillKey, 1*time.Second)
  87. }
  88. total, err := redisCli.ZCard(apictx.CreateRepoCtx().Ctx, reportBillKey).Uint64()
  89. fmt.Println("单据总数量: ", total)
  90. if err != nil {
  91. fmt.Println(err)
  92. return nil, err
  93. }
  94. reports, err := redisCli.ZRevRange(apictx.CreateRepoCtx().Ctx, reportBillKey, 0, -1).Result()
  95. if err != nil {
  96. return nil, err
  97. }
  98. if len(reports) < 1 {
  99. return nil, errors.New("没有单据信息")
  100. }
  101. // 获取供应商名称
  102. // supplier := &model.Supplier{}
  103. // repo.RepoSeachDoc(apictx.CreateRepoCtx(), &repo.DocSearchOptions{
  104. // CollectName: repo.CollectionSupplier,
  105. // Query: repo.Map{"_id": supplierId},
  106. // Project: []string{"name"},
  107. // }, supplier)
  108. ct := time.Now().Format("20060102150405")
  109. // 下载文件的名字
  110. // eName := fmt.Sprintf("统计单据_%s", ct)
  111. companyName := getCompanyName(apictx)
  112. zipName := fmt.Sprintf("统计单据_%s", ct)
  113. // 获取订单信息下载
  114. saveTmpDir := fmt.Sprintf("tmp1/report/%s", zipName)
  115. if isExistDir(saveTmpDir) {
  116. os.RemoveAll(saveTmpDir)
  117. }
  118. f := excelize.NewFile()
  119. f.SetDefaultFont("宋体")
  120. flagIndex := -1
  121. // 采购 加工 加工-印刷 加工-覆膜 成品采购
  122. typeRows := []int{0, 0, 0, 0, 0}
  123. for _, tId := range reports {
  124. var billExcel IExcel
  125. // 采购
  126. if strings.Contains(tId, "purchase") {
  127. purchase := typeBills[tId].(*model.PurchaseBill)
  128. sheetName := "采购单"
  129. index := f.NewSheet(sheetName)
  130. if flagIndex < 0 {
  131. flagIndex = index
  132. }
  133. billExcel = NewPurchaseBill(f)
  134. billExcel.SetSheetName(sheetName)
  135. // 获取签名信息
  136. if purchase.Reviewed == 1 {
  137. if len(purchase.SignUsers) > 0 {
  138. signs := []*model.Signature{}
  139. repo.RepoDocsSearch(apictx.CreateRepoCtx(), &repo.PageSearchOptions{
  140. CollectName: repo.CollectionSignature,
  141. Query: repo.Map{"_id": bson.M{"$in": purchase.SignUsers}},
  142. Sort: bson.M{"sort": 1},
  143. }, &signs)
  144. billExcel.SetSignatures(signs)
  145. }
  146. }
  147. // 获取计划名
  148. // plan := &model.ProductPlan{}
  149. // repo.RepoSeachDoc(apictx.CreateRepoCtx(), &repo.DocSearchOptions{
  150. // CollectName: repo.CollectionProductPlan,
  151. // Query: repo.Map{"_id": purchase.PlanId},
  152. // Project: []string{"name"},
  153. // }, plan)
  154. // productName = purchase.ProductName
  155. // supplierName = purchase.Supplier
  156. // serialNumber = purchase.SerialNumber
  157. billExcel.SetContent(purchase)
  158. billExcel.SetTitle(fmt.Sprintf("%s原材料采购单", companyName))
  159. billExcel.SetRow(typeRows[0])
  160. billExcel.Draws()
  161. typeRows[0] = billExcel.GetRow() + 5
  162. }
  163. // 加工单
  164. if strings.Contains(tId, "produce") {
  165. produce := typeBills[tId].(*model.ProduceBill)
  166. sheetName := "加工单"
  167. if produce.IsPrint {
  168. sheetName = "加工单-印刷"
  169. } else if produce.IsLam {
  170. sheetName = "加工单-覆膜"
  171. }
  172. index := f.NewSheet(sheetName)
  173. if flagIndex < 0 {
  174. flagIndex = index
  175. }
  176. billExcel = NewProduceBill(f)
  177. billExcel.SetSheetName(sheetName)
  178. // 获取签名信息
  179. if produce.Reviewed == 1 {
  180. if len(produce.SignUsers) > 0 {
  181. signs := []*model.Signature{}
  182. repo.RepoDocsSearch(apictx.CreateRepoCtx(), &repo.PageSearchOptions{
  183. CollectName: repo.CollectionSignature,
  184. Query: repo.Map{"_id": bson.M{"$in": produce.SignUsers}},
  185. Sort: bson.M{"sort": 1},
  186. }, &signs)
  187. billExcel.SetSignatures(signs)
  188. }
  189. }
  190. // 获取计划名
  191. // plan := &model.ProductPlan{}
  192. // repo.RepoSeachDoc(apictx.CreateRepoCtx(), &repo.DocSearchOptions{
  193. // CollectName: repo.CollectionProductPlan,
  194. // Query: repo.Map{"_id": produce.PlanId},
  195. // Project: []string{"name"},
  196. // }, plan)
  197. billExcel.SetContent(produce)
  198. billExcel.SetTitle(fmt.Sprintf("%s加工单", companyName))
  199. if produce.IsPrint {
  200. billExcel.SetRow(typeRows[2])
  201. billExcel.Draws()
  202. typeRows[2] = billExcel.GetRow() + 5
  203. } else if produce.IsLam {
  204. billExcel.SetRow(typeRows[3])
  205. billExcel.Draws()
  206. typeRows[3] = billExcel.GetRow() + 5
  207. } else {
  208. billExcel.SetRow(typeRows[1])
  209. billExcel.Draws()
  210. typeRows[1] = billExcel.GetRow() + 5
  211. }
  212. }
  213. // 成品
  214. if strings.Contains(tId, "product") {
  215. product := typeBills[tId].(*model.ProductBill)
  216. sheetName := "成品采购单"
  217. index := f.NewSheet(sheetName)
  218. if flagIndex < 0 {
  219. flagIndex = index
  220. }
  221. billExcel = NewProductBill(f)
  222. billExcel.SetSheetName(sheetName)
  223. // 获取签名信息
  224. if product.Reviewed == 1 {
  225. if len(product.SignUsers) > 0 {
  226. signs := []*model.Signature{}
  227. repo.RepoDocsSearch(apictx.CreateRepoCtx(), &repo.PageSearchOptions{
  228. CollectName: repo.CollectionSignature,
  229. Query: repo.Map{"_id": bson.M{"$in": product.SignUsers}},
  230. Sort: bson.M{"sort": 1},
  231. }, &signs)
  232. billExcel.SetSignatures(signs)
  233. }
  234. }
  235. // 获取计划名
  236. // plan := &model.ProductPlan{}
  237. // repo.RepoSeachDoc(apictx.CreateRepoCtx(), &repo.DocSearchOptions{
  238. // CollectName: repo.CollectionProductPlan,
  239. // Query: repo.Map{"_id": product.PlanId},
  240. // Project: []string{"name"},
  241. // }, plan)
  242. billExcel.SetContent(product)
  243. billExcel.SetTitle(companyName)
  244. billExcel.SetRow(typeRows[4])
  245. billExcel.Draws()
  246. typeRows[4] = billExcel.GetRow() + 5
  247. }
  248. }
  249. // 设置活跃sheet
  250. f.SetActiveSheet(flagIndex)
  251. // 删除默认Sheet1
  252. f.DeleteSheet("Sheet1")
  253. buf, _ := f.WriteToBuffer()
  254. targeName := fmt.Sprintf("%s.xlsx", zipName)
  255. err = saveExcelToTmp1(saveTmpDir, targeName, buf.Bytes())
  256. if err != nil {
  257. return nil, err
  258. }
  259. c.Header("Content-Type", "application/octet-stream")
  260. c.Header("Content-Disposition", "attachment; filename="+fmt.Sprintf("%s.zip", zipName))
  261. c.Header("Content-Transfer-Encoding", "binary")
  262. archive := zip.NewWriter(c.Writer)
  263. defer archive.Close()
  264. // 遍历路径信息
  265. filepath.Walk(saveTmpDir, func(path string, info os.FileInfo, _ error) error {
  266. // 如果是源路径,提前进行下一个遍历
  267. if path == saveTmpDir {
  268. return nil
  269. }
  270. // 获取:文件头信息
  271. header, _ := zip.FileInfoHeader(info)
  272. header.Name = strings.TrimPrefix(path, saveTmpDir+`/`)
  273. // 判断:文件是不是文件夹
  274. if info.IsDir() {
  275. header.Name += `/`
  276. } else {
  277. // 设置:zip的文件压缩算法
  278. header.Method = zip.Deflate
  279. }
  280. // 创建:压缩包头部信息
  281. writer, _ := archive.CreateHeader(header)
  282. if !info.IsDir() {
  283. file, _ := os.Open(path)
  284. defer file.Close()
  285. io.Copy(writer, file)
  286. }
  287. return nil
  288. })
  289. // 删除缓存目录
  290. os.RemoveAll(saveTmpDir)
  291. return nil, nil
  292. }
  293. // 下载单据
  294. //
  295. // !更改为ReportBillsDownload1 确定不需要后删除
  296. func ReportBillsDownload(c *gin.Context, apictx *ApiSession) (interface{}, error) {
  297. _, _, query := UtilQueryPageSize(c)
  298. // supplierId := primitive.NilObjectID
  299. // if _supplierId, ok := query["supplierId"]; ok {
  300. // supplierId, _ = primitive.ObjectIDFromHex(_supplierId.(string))
  301. // }
  302. _isPdf := "false"
  303. if isPdf, ok := query["isPdf"]; ok {
  304. delete(query, "isPdf")
  305. if isPdf.(string) == "true" {
  306. _isPdf = "true"
  307. }
  308. }
  309. // timeRange := []interface{}{}
  310. // if _timeRange, ok := query["timeRange"]; ok {
  311. // timeRange, _ = _timeRange.([]interface{})
  312. // }
  313. filtter := handleReportQuery(query)
  314. purchases := []*model.PurchaseBill{}
  315. produces := []*model.ProduceBill{}
  316. products := []*model.ProductBill{}
  317. repo.RepoDocsSearch(apictx.CreateRepoCtx(), &repo.PageSearchOptions{
  318. CollectName: repo.CollectionBillPurchase,
  319. Query: filtter,
  320. }, &purchases)
  321. repo.RepoDocsSearch(apictx.CreateRepoCtx(), &repo.PageSearchOptions{
  322. CollectName: repo.CollectionBillProduce,
  323. Query: filtter,
  324. }, &produces)
  325. repo.RepoDocsSearch(apictx.CreateRepoCtx(), &repo.PageSearchOptions{
  326. CollectName: repo.CollectionBillProduct,
  327. Query: filtter,
  328. }, &products)
  329. // 加入redis有序集合中进行排序
  330. // 把对应type_id:对象存入map中避免再次查询表
  331. typeBills := map[string]interface{}{}
  332. redisCli := apictx.Svc.Redis
  333. reportBillKey := "report-bill-list:" + apictx.User.Parent
  334. isExist := redisCli.Exists(apictx.CreateRepoCtx().Ctx, reportBillKey).Val()
  335. // 不存在这个key时
  336. if isExist < 1 {
  337. if len(purchases) > 0 {
  338. for _, purchase := range purchases {
  339. member := "purchase_" + purchase.Id.Hex()
  340. score := purchase.CompleteTime.Unix()
  341. redisCli.ZAdd(apictx.CreateRepoCtx().Ctx, reportBillKey, &redis.Z{Score: float64(score), Member: member})
  342. typeBills[member] = purchase
  343. }
  344. }
  345. if len(produces) > 0 {
  346. for _, produce := range produces {
  347. member := "produce_" + produce.Id.Hex()
  348. score := produce.CompleteTime.Unix()
  349. redisCli.ZAdd(apictx.CreateRepoCtx().Ctx, reportBillKey, &redis.Z{Score: float64(score), Member: member})
  350. typeBills[member] = produce
  351. }
  352. }
  353. if len(products) > 0 {
  354. for _, product := range products {
  355. member := "product_" + product.Id.Hex()
  356. score := product.CompleteTime.Unix()
  357. redisCli.ZAdd(apictx.CreateRepoCtx().Ctx, reportBillKey, &redis.Z{Score: float64(score), Member: member})
  358. typeBills[member] = product
  359. }
  360. }
  361. // 设置过期时间
  362. redisCli.Expire(apictx.CreateRepoCtx().Ctx, reportBillKey, 1*time.Second)
  363. }
  364. total, err := redisCli.ZCard(apictx.CreateRepoCtx().Ctx, reportBillKey).Uint64()
  365. fmt.Println("单据总数量: ", total)
  366. if err != nil {
  367. fmt.Println(err)
  368. return nil, err
  369. }
  370. reports, err := redisCli.ZRevRange(apictx.CreateRepoCtx().Ctx, reportBillKey, 0, -1).Result()
  371. if err != nil {
  372. return nil, err
  373. }
  374. if len(reports) < 1 {
  375. return nil, errors.New("没有单据信息")
  376. }
  377. // 获取供应商名称
  378. // supplier := &model.Supplier{}
  379. // repo.RepoSeachDoc(apictx.CreateRepoCtx(), &repo.DocSearchOptions{
  380. // CollectName: repo.CollectionSupplier,
  381. // Query: repo.Map{"_id": supplierId},
  382. // Project: []string{"name"},
  383. // }, supplier)
  384. r := regexp.MustCompile(`/`)
  385. ct := time.Now().Format("20060102150405")
  386. zipName := fmt.Sprintf("统计单据_%s", ct)
  387. // 获取订单信息下载
  388. saveTmpDir := fmt.Sprintf("tmp1/report/%s", zipName)
  389. if isExistDir(saveTmpDir) {
  390. os.RemoveAll(saveTmpDir)
  391. }
  392. // 记录文件数量
  393. var wg sync.WaitGroup
  394. c1 := make(chan int, 10)
  395. for _, tId := range reports {
  396. productName := ""
  397. supplierName := ""
  398. serialNumber := ""
  399. f := excelize.NewFile()
  400. index := f.NewSheet("Sheet1")
  401. f.SetActiveSheet(index)
  402. f.SetDefaultFont("宋体")
  403. tidArr := strings.Split(tId, "_")
  404. var billExcel IExcel
  405. // 采购
  406. if tidArr[0] == "purchase" {
  407. purchase := typeBills[tId].(*model.PurchaseBill)
  408. billExcel = NewPurchaseBill(f)
  409. // 获取签名信息
  410. if purchase.Reviewed == 1 {
  411. if len(purchase.SignUsers) > 0 {
  412. signs := []*model.Signature{}
  413. repo.RepoDocsSearch(apictx.CreateRepoCtx(), &repo.PageSearchOptions{
  414. CollectName: repo.CollectionSignature,
  415. Query: repo.Map{"_id": bson.M{"$in": purchase.SignUsers}},
  416. Sort: bson.M{"sort": 1},
  417. }, &signs)
  418. billExcel.SetSignatures(signs)
  419. }
  420. }
  421. // 获取计划名
  422. plan := &model.ProductPlan{}
  423. repo.RepoSeachDoc(apictx.CreateRepoCtx(), &repo.DocSearchOptions{
  424. CollectName: repo.CollectionProductPlan,
  425. Query: repo.Map{"_id": purchase.PlanId},
  426. Project: []string{"name"},
  427. }, plan)
  428. productName = purchase.ProductName
  429. supplierName = purchase.Supplier
  430. serialNumber = purchase.SerialNumber
  431. billExcel.SetContent(purchase)
  432. billExcel.SetTitle(fmt.Sprintf("%s原材料采购单", plan.Name))
  433. }
  434. // 工艺
  435. if tidArr[0] == "produce" {
  436. produce := typeBills[tId].(*model.ProduceBill)
  437. billExcel = NewProduceBill(f)
  438. if produce.Reviewed == 1 {
  439. if len(produce.SignUsers) > 0 {
  440. signs := []*model.Signature{}
  441. repo.RepoDocsSearch(apictx.CreateRepoCtx(), &repo.PageSearchOptions{
  442. CollectName: repo.CollectionSignature,
  443. Query: repo.Map{"_id": bson.M{"$in": produce.SignUsers}},
  444. Sort: bson.M{"sort": 1},
  445. }, &signs)
  446. billExcel.SetSignatures(signs)
  447. }
  448. }
  449. // 获取计划名
  450. plan := &model.ProductPlan{}
  451. repo.RepoSeachDoc(apictx.CreateRepoCtx(), &repo.DocSearchOptions{
  452. CollectName: repo.CollectionProductPlan,
  453. Query: repo.Map{"_id": produce.PlanId},
  454. Project: []string{"name"},
  455. }, plan)
  456. productName = produce.ProductName
  457. supplierName = produce.Supplier
  458. serialNumber = produce.SerialNumber
  459. billExcel.SetContent(produce)
  460. billExcel.SetTitle(fmt.Sprintf("%s加工单", plan.Name))
  461. }
  462. // 成品采购
  463. if tidArr[0] == "product" {
  464. product := typeBills[tId].(*model.ProductBill)
  465. billExcel = NewProductBill(f)
  466. if product.Reviewed == 1 {
  467. if len(product.SignUsers) > 0 {
  468. signs := []*model.Signature{}
  469. repo.RepoDocsSearch(apictx.CreateRepoCtx(), &repo.PageSearchOptions{
  470. CollectName: repo.CollectionSignature,
  471. Query: repo.Map{"_id": bson.M{"$in": product.SignUsers}},
  472. Sort: bson.M{"sort": 1},
  473. }, &signs)
  474. billExcel.SetSignatures(signs)
  475. }
  476. }
  477. // 获取计划名
  478. plan := &model.ProductPlan{}
  479. repo.RepoSeachDoc(apictx.CreateRepoCtx(), &repo.DocSearchOptions{
  480. CollectName: repo.CollectionProductPlan,
  481. Query: repo.Map{"_id": product.PlanId},
  482. Project: []string{"name"},
  483. }, plan)
  484. productName = product.ProductName
  485. supplierName = product.Supplier
  486. serialNumber = product.SerialNumber
  487. billExcel.SetContent(product)
  488. billExcel.SetTitle(plan.Name)
  489. }
  490. billExcel.SetIsPdf(_isPdf)
  491. billExcel.Draws()
  492. buf, _ := f.WriteToBuffer()
  493. ft := "xlsx"
  494. if _isPdf == "true" {
  495. ft = "pdf"
  496. }
  497. _productName := r.ReplaceAllString(productName, `&`)
  498. _supplierName := r.ReplaceAllString(supplierName, `&`)
  499. targePdfName := fmt.Sprintf("%s-%s-%s.%s", _supplierName, _productName, serialNumber, ft)
  500. wg.Add(1)
  501. if _isPdf == "true" {
  502. go toPdfAndSaveTask(buf, apictx.Svc.Conf.PdfApiAddr, saveTmpDir, targePdfName, c1, &wg)
  503. } else {
  504. go saveExcelToTmp(saveTmpDir, targePdfName, buf.Bytes(), c1, &wg)
  505. }
  506. }
  507. go func() {
  508. // 等待所有goroutine
  509. wg.Wait()
  510. // 关闭channel
  511. close(c1)
  512. }()
  513. // channel关闭后结束后停止遍历接收channel中的值
  514. for n := range c1 {
  515. if n == -1 {
  516. return nil, errors.New("下载失败,请重试")
  517. }
  518. }
  519. c.Header("Content-Type", "application/octet-stream")
  520. c.Header("Content-Disposition", "attachment; filename="+zipName+".zip")
  521. c.Header("Content-Transfer-Encoding", "binary")
  522. archive := zip.NewWriter(c.Writer)
  523. defer archive.Close()
  524. // 遍历路径信息
  525. filepath.Walk(saveTmpDir, func(path string, info os.FileInfo, _ error) error {
  526. // 如果是源路径,提前进行下一个遍历
  527. if path == saveTmpDir {
  528. return nil
  529. }
  530. // 获取:文件头信息
  531. header, _ := zip.FileInfoHeader(info)
  532. header.Name = strings.TrimPrefix(path, saveTmpDir+`/`)
  533. // 判断:文件是不是文件夹
  534. if info.IsDir() {
  535. header.Name += `/`
  536. } else {
  537. // 设置:zip的文件压缩算法
  538. header.Method = zip.Deflate
  539. }
  540. // 创建:压缩包头部信息
  541. writer, _ := archive.CreateHeader(header)
  542. if !info.IsDir() {
  543. file, _ := os.Open(path)
  544. defer file.Close()
  545. io.Copy(writer, file)
  546. }
  547. return nil
  548. })
  549. // 删除缓存目录
  550. os.RemoveAll(saveTmpDir)
  551. return nil, nil
  552. }
  553. // 加工单
  554. func ReportListDownload(c *gin.Context, apictx *ApiSession) (interface{}, error) {
  555. _, _, query := UtilQueryPageSize(c)
  556. // start, stop := CreatePageRange(page, size)
  557. supplierId := primitive.NilObjectID
  558. if _supplierId, ok := query["supplierId"]; ok {
  559. supplierId, _ = primitive.ObjectIDFromHex(_supplierId.(string))
  560. }
  561. timeRange := []interface{}{}
  562. if _timeRange, ok := query["timeRange"]; ok {
  563. timeRange, _ = _timeRange.([]interface{})
  564. }
  565. filtter := handleReportQuery(query)
  566. purchases := []*model.PurchaseBill{}
  567. produces := []*model.ProduceBill{}
  568. products := []*model.ProductBill{}
  569. repo.RepoDocsSearch(apictx.CreateRepoCtx(), &repo.PageSearchOptions{
  570. CollectName: repo.CollectionBillPurchase,
  571. Query: filtter,
  572. Project: []string{"_id", "completeTime"},
  573. }, &purchases)
  574. repo.RepoDocsSearch(apictx.CreateRepoCtx(), &repo.PageSearchOptions{
  575. CollectName: repo.CollectionBillProduce,
  576. Query: filtter,
  577. Project: []string{"_id", "completeTime"},
  578. }, &produces)
  579. repo.RepoDocsSearch(apictx.CreateRepoCtx(), &repo.PageSearchOptions{
  580. CollectName: repo.CollectionBillProduct,
  581. Query: filtter,
  582. Project: []string{"_id", "completeTime"},
  583. }, &products)
  584. // 加入redis有序集合中
  585. redisCli := apictx.Svc.Redis
  586. reportBillKey := "report-bill-list:" + apictx.User.Parent
  587. isExist := redisCli.Exists(apictx.CreateRepoCtx().Ctx, reportBillKey).Val()
  588. // 不存在这个key时
  589. if isExist < 1 {
  590. if len(purchases) > 0 {
  591. for _, purchase := range purchases {
  592. member := "purchase_" + purchase.Id.Hex()
  593. score := purchase.CompleteTime.Unix()
  594. redisCli.ZAdd(apictx.CreateRepoCtx().Ctx, reportBillKey, &redis.Z{Score: float64(score), Member: member})
  595. }
  596. }
  597. if len(produces) > 0 {
  598. for _, produce := range produces {
  599. member := "produce_" + produce.Id.Hex()
  600. score := produce.CompleteTime.Unix()
  601. redisCli.ZAdd(apictx.CreateRepoCtx().Ctx, reportBillKey, &redis.Z{Score: float64(score), Member: member})
  602. }
  603. }
  604. if len(products) > 0 {
  605. for _, product := range products {
  606. member := "product_" + product.Id.Hex()
  607. score := product.CompleteTime.Unix()
  608. redisCli.ZAdd(apictx.CreateRepoCtx().Ctx, reportBillKey, &redis.Z{Score: float64(score), Member: member})
  609. }
  610. }
  611. // 设置过期时间
  612. redisCli.Expire(apictx.CreateRepoCtx().Ctx, reportBillKey, 1*time.Second)
  613. }
  614. total, err := redisCli.ZCard(apictx.CreateRepoCtx().Ctx, reportBillKey).Uint64()
  615. fmt.Println(total)
  616. if err != nil {
  617. fmt.Println(err)
  618. return nil, err
  619. }
  620. reports, err := redisCli.ZRevRange(apictx.CreateRepoCtx().Ctx, reportBillKey, 0, -1).Result()
  621. if err != nil {
  622. return nil, err
  623. }
  624. if len(reports) < 1 {
  625. return nil, errors.New("没有单据信息")
  626. }
  627. lists := []map[string]interface{}{}
  628. for _, report := range reports {
  629. billArray := strings.Split(report, "_")
  630. billType := billArray[0]
  631. _billId := billArray[1]
  632. billId, _ := primitive.ObjectIDFromHex(_billId)
  633. billData := map[string]interface{}{}
  634. found := false
  635. if billType == "purchase" {
  636. found, billData = repo.RepoSeachDocMap(apictx.CreateRepoCtx(), &repo.DocSearchOptions{
  637. CollectName: repo.CollectionBillPurchase,
  638. Query: repo.Map{"_id": billId},
  639. })
  640. }
  641. if billType == "produce" {
  642. found, billData = repo.RepoSeachDocMap(apictx.CreateRepoCtx(), &repo.DocSearchOptions{
  643. CollectName: repo.CollectionBillProduce,
  644. Query: repo.Map{"_id": billId},
  645. })
  646. }
  647. if billType == "product" {
  648. found, billData = repo.RepoSeachDocMap(apictx.CreateRepoCtx(), &repo.DocSearchOptions{
  649. CollectName: repo.CollectionBillProduct,
  650. Query: repo.Map{"_id": billId},
  651. })
  652. }
  653. if found {
  654. billData["billType"] = billType
  655. lists = append(lists, billData)
  656. }
  657. }
  658. f := excelize.NewFile()
  659. defer f.Close()
  660. index := f.NewSheet("Sheet1")
  661. f.SetActiveSheet(index)
  662. f.SetDefaultFont("宋体")
  663. report := NewReportExcel(f)
  664. supplier := model.Supplier{}
  665. supplierName := "【所有供应商】"
  666. if !supplierId.IsZero() {
  667. repo.RepoSeachDoc(apictx.CreateRepoCtx(), &repo.DocSearchOptions{
  668. CollectName: repo.CollectionSupplier,
  669. Query: repo.Map{"_id": supplierId},
  670. Project: []string{"name"},
  671. }, &supplier)
  672. supplierName = supplier.Name
  673. }
  674. report.SupplierName = supplierName
  675. if len(timeRange) == 2 {
  676. report.TimeRange = append(report.TimeRange, timeRange[0].(string), timeRange[1].(string))
  677. }
  678. report.Content = lists
  679. report.Draws()
  680. c.Header("Content-Type", "application/octet-stream")
  681. c.Header("Content-Disposition", "attachment; filename="+"report.xlsx")
  682. c.Header("Content-Transfer-Encoding", "binary")
  683. err = f.Write(c.Writer)
  684. if err != nil {
  685. return nil, err
  686. }
  687. return nil, nil
  688. }
  689. // 加工单
  690. func ReportList(c *gin.Context, apictx *ApiSession) (interface{}, error) {
  691. page, size, query := UtilQueryPageSize(c)
  692. start, stop := CreatePageRange(page, size)
  693. filtter := handleReportQuery(query)
  694. purchases := []*model.PurchaseBill{}
  695. produces := []*model.ProduceBill{}
  696. products := []*model.ProductBill{}
  697. repo.RepoDocsSearch(apictx.CreateRepoCtx(), &repo.PageSearchOptions{
  698. CollectName: repo.CollectionBillPurchase,
  699. Query: filtter,
  700. Project: []string{"_id", "completeTime"},
  701. }, &purchases)
  702. repo.RepoDocsSearch(apictx.CreateRepoCtx(), &repo.PageSearchOptions{
  703. CollectName: repo.CollectionBillProduce,
  704. Query: filtter,
  705. Project: []string{"_id", "completeTime"},
  706. }, &produces)
  707. repo.RepoDocsSearch(apictx.CreateRepoCtx(), &repo.PageSearchOptions{
  708. CollectName: repo.CollectionBillProduct,
  709. Query: filtter,
  710. Project: []string{"_id", "completeTime"},
  711. }, &products)
  712. // 加入redis有序集合中
  713. redisCli := apictx.Svc.Redis
  714. reportBillKey := "report-bill-list:" + apictx.User.Parent
  715. isExist := redisCli.Exists(apictx.CreateRepoCtx().Ctx, reportBillKey).Val()
  716. // 不存在这个key时
  717. if isExist < 1 {
  718. if len(purchases) > 0 {
  719. for _, purchase := range purchases {
  720. member := "purchase_" + purchase.Id.Hex()
  721. score := purchase.CompleteTime.Unix()
  722. redisCli.ZAdd(apictx.CreateRepoCtx().Ctx, reportBillKey, &redis.Z{Score: float64(score), Member: member})
  723. }
  724. }
  725. if len(produces) > 0 {
  726. for _, produce := range produces {
  727. member := "produce_" + produce.Id.Hex()
  728. score := produce.CompleteTime.Unix()
  729. redisCli.ZAdd(apictx.CreateRepoCtx().Ctx, reportBillKey, &redis.Z{Score: float64(score), Member: member})
  730. }
  731. }
  732. if len(products) > 0 {
  733. for _, product := range products {
  734. member := "product_" + product.Id.Hex()
  735. score := product.CompleteTime.Unix()
  736. redisCli.ZAdd(apictx.CreateRepoCtx().Ctx, reportBillKey, &redis.Z{Score: float64(score), Member: member})
  737. }
  738. }
  739. // 设置过期时间
  740. redisCli.Expire(apictx.CreateRepoCtx().Ctx, reportBillKey, 1*time.Second)
  741. }
  742. total, err := redisCli.ZCard(apictx.CreateRepoCtx().Ctx, reportBillKey).Uint64()
  743. if err != nil {
  744. fmt.Println(err)
  745. return nil, err
  746. }
  747. reports, err := redisCli.ZRevRange(apictx.CreateRepoCtx().Ctx, reportBillKey, start, stop).Result()
  748. if err != nil {
  749. return nil, err
  750. }
  751. if len(reports) < 1 {
  752. return repo.PageResult{
  753. List: []map[string]interface{}{},
  754. Page: page,
  755. Size: size,
  756. Total: 0,
  757. }, nil
  758. }
  759. lists := []map[string]interface{}{}
  760. for _, report := range reports {
  761. billArray := strings.Split(report, "_")
  762. billType := billArray[0]
  763. _billId := billArray[1]
  764. billId, _ := primitive.ObjectIDFromHex(_billId)
  765. billData := map[string]interface{}{}
  766. found := false
  767. if billType == "purchase" {
  768. found, billData = repo.RepoSeachDocMap(apictx.CreateRepoCtx(), &repo.DocSearchOptions{
  769. CollectName: repo.CollectionBillPurchase,
  770. Query: repo.Map{"_id": billId},
  771. })
  772. }
  773. if billType == "produce" {
  774. found, billData = repo.RepoSeachDocMap(apictx.CreateRepoCtx(), &repo.DocSearchOptions{
  775. CollectName: repo.CollectionBillProduce,
  776. Query: repo.Map{"_id": billId},
  777. })
  778. }
  779. if billType == "product" {
  780. found, billData = repo.RepoSeachDocMap(apictx.CreateRepoCtx(), &repo.DocSearchOptions{
  781. CollectName: repo.CollectionBillProduct,
  782. Query: repo.Map{"_id": billId},
  783. })
  784. }
  785. if found {
  786. billData["billType"] = billType
  787. lists = append(lists, billData)
  788. }
  789. }
  790. return repo.PageResult{
  791. List: lists,
  792. Total: int64(total),
  793. Page: page,
  794. Size: size,
  795. }, nil
  796. }
  797. func CreatePageRange(page, size int64) (int64, int64) {
  798. if page < 1 {
  799. page = 1
  800. }
  801. if size < 1 {
  802. size = 10
  803. }
  804. start := (page - 1) * size
  805. stop := page*size - 1
  806. return start, stop
  807. }
  808. func ReportProduceList(c *gin.Context, apictx *ApiSession) (interface{}, error) {
  809. page, size, query := UtilQueryPageSize(c)
  810. // 条件处理
  811. // 获取采购单符合条件的信息
  812. return repo.RepoPageSearch(apictx.CreateRepoCtx(), &repo.PageSearchOptions{
  813. CollectName: repo.CollectionBillProduce,
  814. Query: handleReportQuery(query),
  815. Page: page,
  816. Size: size,
  817. })
  818. }
  819. // 采购单
  820. func ReportPurchaseList(c *gin.Context, apictx *ApiSession) (interface{}, error) {
  821. page, size, query := UtilQueryPageSize(c)
  822. return repo.RepoPageSearch(apictx.CreateRepoCtx(), &repo.PageSearchOptions{
  823. CollectName: repo.CollectionBillPurchase,
  824. Query: handleReportQuery(query),
  825. Page: page,
  826. Size: size,
  827. })
  828. }
  829. func ReportProductList(c *gin.Context, apictx *ApiSession) (interface{}, error) {
  830. page, size, query := UtilQueryPageSize(c)
  831. return repo.RepoPageSearch(apictx.CreateRepoCtx(), &repo.PageSearchOptions{
  832. CollectName: repo.CollectionBillProduct,
  833. Query: handleReportQuery(query),
  834. Page: page,
  835. Size: size,
  836. })
  837. }