plan.go 8.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332
  1. package api
  2. import (
  3. "box-cost/db/model"
  4. "box-cost/db/repo"
  5. "box-cost/log"
  6. "errors"
  7. "fmt"
  8. "time"
  9. "github.com/gin-gonic/gin"
  10. "github.com/xuri/excelize/v2"
  11. "go.mongodb.org/mongo-driver/bson"
  12. "go.mongodb.org/mongo-driver/bson/primitive"
  13. )
  14. // 生产计划管理
  15. func ProductPlan(r *GinRouter) {
  16. // 创建生产计划
  17. r.POST("/plan/create", CreateProductPlan)
  18. // 获取生产计划详情
  19. r.GET("/plan/detail/:id", GetProductPlan)
  20. // 获取生产计划列表
  21. r.GET("/plan/list", GetProductPlans)
  22. // 更新生产计划
  23. r.POST("/plan/update", UpdateProductPlan)
  24. // 删除生产计划
  25. r.POST("/plan/delete/:id", DelProductPlan)
  26. // 下载部件打印单
  27. r.GET("/bill/plan/download", DownLoadPlan)
  28. // 生产成本表
  29. r.GET("/plan/cost/download", DownLoadPlanCost)
  30. }
  31. type SupplierPlanCost struct {
  32. *model.ProductPlan
  33. SupplierId string
  34. }
  35. func DownLoadPlanCost(c *gin.Context, apictx *ApiSession) (interface{}, error) {
  36. _planId := c.Query("id")
  37. supplierId := c.Query("supplierId")
  38. planId, err := primitive.ObjectIDFromHex(_planId)
  39. if err != nil {
  40. return nil, errors.New("planId错误")
  41. }
  42. supplierPlanCost := &SupplierPlanCost{}
  43. plan := model.ProductPlan{}
  44. found, err := repo.RepoSeachDoc(apictx.CreateRepoCtx(), &repo.DocSearchOptions{
  45. CollectName: repo.CollectionProductPlan,
  46. Query: repo.Map{"_id": planId},
  47. }, &plan)
  48. if !found || err != nil {
  49. return nil, errors.New("数据未找到")
  50. }
  51. supplierPlanCost.ProductPlan = &plan
  52. supplierPlanCost.SupplierId = supplierId
  53. f := excelize.NewFile()
  54. index := f.NewSheet("Sheet1")
  55. f.SetActiveSheet(index)
  56. f.SetDefaultFont("宋体")
  57. planCostExcel := NewPlanCostExcel(f)
  58. planCostExcel.Title = fmt.Sprintf("生产成本表(%s)%d盒", plan.Name, plan.Total)
  59. planCostExcel.Content = supplierPlanCost
  60. planCostExcel.Draws()
  61. c.Header("Content-Type", "application/octet-stream")
  62. c.Header("Content-Disposition", "attachment; filename="+"planCost.xlsx")
  63. c.Header("Content-Transfer-Encoding", "binary")
  64. err = f.Write(c.Writer)
  65. if err != nil {
  66. return nil, err
  67. }
  68. return nil, nil
  69. }
  70. func DownLoadPlan(c *gin.Context, apictx *ApiSession) (interface{}, error) {
  71. _planId := c.Query("id")
  72. compId := c.Query("compId")
  73. planId, err := primitive.ObjectIDFromHex(_planId)
  74. if err != nil {
  75. return nil, errors.New("planId错误")
  76. }
  77. plan := model.ProductPlan{}
  78. found, err := repo.RepoSeachDoc(apictx.CreateRepoCtx(), &repo.DocSearchOptions{
  79. CollectName: repo.CollectionProductPlan,
  80. Query: repo.Map{"_id": planId},
  81. }, &plan)
  82. if !found || err != nil {
  83. return nil, errors.New("数据未找到")
  84. }
  85. // 获取部件单据
  86. curComp := &model.PackComponent{}
  87. for _, comp := range plan.Pack.Components {
  88. if comp.Id == compId {
  89. curComp = comp
  90. }
  91. }
  92. if curComp.Id == "" {
  93. return nil, errors.New("该组件不存在")
  94. }
  95. // 获取bill
  96. if len(curComp.Mats) == 0 {
  97. return nil, errors.New("该组件数据不存在")
  98. }
  99. f := excelize.NewFile()
  100. // Create a new sheet.
  101. index := f.NewSheet("Sheet1")
  102. f.SetActiveSheet(index)
  103. f.SetDefaultFont("宋体")
  104. companyName := getCompanyName(apictx)
  105. offset := 0
  106. for _, mat := range curComp.Mats {
  107. // 采购单
  108. _purchaseId := mat.BillId
  109. purchaseId, err := primitive.ObjectIDFromHex(_purchaseId)
  110. if err == nil {
  111. purchase := model.PurchaseBill{}
  112. found, _ := repo.RepoSeachDoc(apictx.CreateRepoCtx(), &repo.DocSearchOptions{
  113. CollectName: repo.CollectionBillPurchase,
  114. Query: repo.Map{"_id": purchaseId},
  115. }, &purchase)
  116. if found {
  117. var billExcel IPurchBill
  118. if purchase.Process != nil {
  119. billExcel = NewProcessBill(f)
  120. }
  121. if len(purchase.Paper) > 0 {
  122. billExcel = NewPurchaseBill(f)
  123. }
  124. // purchaseExcel := NewPurchaseBill(f)
  125. billExcel.SetContent(&purchase)
  126. billExcel.SetTitle(fmt.Sprintf("%s原材料采购单", companyName))
  127. billExcel.SetOffset(offset)
  128. // billExcel.Title = fmt.Sprintf("%s原材料采购单", companyName)
  129. //设置对应的数据
  130. // purchaseExcel.Offset = offset
  131. billExcel.Draws()
  132. offset += 15
  133. }
  134. }
  135. if len(mat.Crafts) > 0 {
  136. for _, carft := range mat.Crafts {
  137. // 加工单
  138. _produceId := carft.BillId
  139. produceId, err := primitive.ObjectIDFromHex(_produceId)
  140. if err == nil {
  141. produce := model.ProduceBill{}
  142. found, _ := repo.RepoSeachDoc(apictx.CreateRepoCtx(), &repo.DocSearchOptions{
  143. CollectName: repo.CollectionBillProduce,
  144. Query: repo.Map{"_id": produceId},
  145. }, &produce)
  146. if found {
  147. produceExcel := NewProduceBill(f)
  148. produceExcel.Content = &produce
  149. produceExcel.Title = fmt.Sprintf("%s加工单", companyName)
  150. //设置对应的数据
  151. produceExcel.Offset = offset
  152. produceExcel.Draws()
  153. offset += 15
  154. }
  155. }
  156. }
  157. }
  158. }
  159. c.Header("Content-Type", "application/octet-stream")
  160. c.Header("Content-Disposition", "attachment; filename="+"bill.xlsx")
  161. c.Header("Content-Transfer-Encoding", "binary")
  162. err = f.Write(c.Writer)
  163. if err != nil {
  164. return nil, err
  165. }
  166. return nil, nil
  167. }
  168. // 创建生产计划
  169. func CreateProductPlan(c *gin.Context, apictx *ApiSession) (interface{}, error) {
  170. var plan model.ProductPlan
  171. err := c.ShouldBindJSON(&plan)
  172. if err != nil {
  173. fmt.Println(err)
  174. return nil, errors.New("参数错误!")
  175. }
  176. ctx := apictx.CreateRepoCtx()
  177. if plan.Name == "" {
  178. return nil, errors.New("生产计划名为空")
  179. }
  180. if plan.Total == 0 {
  181. return nil, errors.New("生产计划数应不为0")
  182. }
  183. plan.Status = "process" // 进行中
  184. plan.CreateTime = time.Now()
  185. plan.UpdateTime = time.Now()
  186. result, err := repo.RepoAddDoc(ctx, repo.CollectionProductPlan, &plan)
  187. return result, err
  188. }
  189. // 获取生产计划信息
  190. func GetProductPlan(c *gin.Context, apictx *ApiSession) (interface{}, error) {
  191. planId := c.Param("id")
  192. id, err := primitive.ObjectIDFromHex(planId)
  193. if err != nil {
  194. return nil, errors.New("非法id")
  195. }
  196. var plan model.ProductPlan
  197. option := &repo.DocSearchOptions{
  198. CollectName: repo.CollectionProductPlan,
  199. Query: repo.Map{"_id": id},
  200. }
  201. found, err := repo.RepoSeachDoc(apictx.CreateRepoCtx(), option, &plan)
  202. if !found || err != nil {
  203. log.Info(err)
  204. return nil, errors.New("数据未找到")
  205. }
  206. billStates := map[string]string{}
  207. if len(plan.Process) > 0 {
  208. for _, pp := range plan.Process {
  209. ok, state := repo.RepoSeachDocMap(apictx.CreateRepoCtx(), &repo.DocSearchOptions{CollectName: repo.CollectionBillPurchase, Query: repo.Map{"_id": pp.BillId}, Project: []string{"status"}})
  210. if ok {
  211. billStates[pp.BillId] = state["status"].(string)
  212. }
  213. }
  214. }
  215. if plan.Pack != nil && plan.Pack.Components != nil {
  216. for _, comp := range plan.Pack.Components {
  217. if comp.Mats != nil {
  218. for _, mat := range comp.Mats {
  219. if len(mat.BillId) > 0 {
  220. ok, state := repo.RepoSeachDocMap(apictx.CreateRepoCtx(), &repo.DocSearchOptions{CollectName: repo.CollectionBillPurchase, Query: repo.Map{"_id": mat.BillId}, Project: []string{"status"}})
  221. if ok {
  222. billStates[mat.BillId] = state["status"].(string)
  223. }
  224. }
  225. if mat.Crafts != nil {
  226. for _, craft := range mat.Crafts {
  227. if len(craft.BillId) > 0 {
  228. ok, state := repo.RepoSeachDocMap(apictx.CreateRepoCtx(), &repo.DocSearchOptions{CollectName: repo.CollectionBillProduce, Query: repo.Map{"_id": craft.BillId}, Project: []string{"status"}})
  229. if ok {
  230. billStates[craft.BillId] = state["status"].(string)
  231. }
  232. }
  233. }
  234. }
  235. }
  236. }
  237. }
  238. }
  239. return map[string]interface{}{
  240. "plan": plan,
  241. "billStates": billStates,
  242. }, nil
  243. }
  244. // 获取生产计划列表
  245. func GetProductPlans(c *gin.Context, apictx *ApiSession) (interface{}, error) {
  246. page, size, query := UtilQueryPageSize(c)
  247. if _packId, ok := query["packId"]; ok {
  248. packId, _ := primitive.ObjectIDFromHex(_packId.(string))
  249. query["pack._id"] = packId
  250. delete(query, "packId")
  251. }
  252. option := &repo.PageSearchOptions{
  253. CollectName: repo.CollectionProductPlan,
  254. Query: query,
  255. Page: page,
  256. Size: size,
  257. Sort: bson.M{"createTime": -1},
  258. Project: []string{"_id", "thumbnail", "name", "updateTime", "createTime", "createUser", "total", "totalPrice", "status"},
  259. }
  260. return repo.RepoPageSearch(apictx.CreateRepoCtx(), option)
  261. }
  262. // 更新生产计划
  263. func UpdateProductPlan(c *gin.Context, apictx *ApiSession) (interface{}, error) {
  264. var plan model.ProductPlan
  265. err := c.ShouldBindJSON(&plan)
  266. if err != nil {
  267. return nil, errors.New("参数错误")
  268. }
  269. if plan.Id.Hex() == "" {
  270. return nil, errors.New("id的为空")
  271. }
  272. plan.UpdateTime = time.Now()
  273. return repo.RepoUpdateSetDoc(apictx.CreateRepoCtx(), repo.CollectionProductPlan, plan.Id.Hex(), &plan)
  274. }
  275. // 删除生产计划
  276. func DelProductPlan(c *gin.Context, apictx *ApiSession) (interface{}, error) {
  277. planId := c.Param("id")
  278. if planId == "" {
  279. return nil, errors.New("id为空")
  280. }
  281. return repo.RepoDeleteDoc(apictx.CreateRepoCtx(), repo.CollectionProductPlan, planId)
  282. }