feed_formula.go 12 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432
  1. package feed
  2. import (
  3. "fmt"
  4. "kpt-tmr-group/http/middleware"
  5. "kpt-tmr-group/pkg/apierr"
  6. "kpt-tmr-group/pkg/ginutil"
  7. "kpt-tmr-group/pkg/valid"
  8. "kpt-tmr-group/pkg/xerr"
  9. operationPb "kpt-tmr-group/proto/go/backend/operation"
  10. "mime/multipart"
  11. "net/http"
  12. "path"
  13. "strconv"
  14. "time"
  15. "github.com/gin-gonic/gin"
  16. )
  17. func AddFeedFormula(c *gin.Context) {
  18. var req operationPb.AddFeedFormulaRequest
  19. if err := c.BindJSON(&req); err != nil {
  20. apierr.AbortBadRequest(c, http.StatusBadRequest, err)
  21. return
  22. }
  23. if err := valid.ValidateStruct(&req,
  24. valid.Field(&req.Name, valid.Required),
  25. valid.Field(&req.EncodeNumber, valid.Required, valid.Length(1, 30)),
  26. valid.Field(&req.Colour, valid.Required),
  27. valid.Field(&req.CattleCategoryId, valid.Required, valid.Min(1)),
  28. valid.Field(&req.CattleCategoryName, valid.Required),
  29. valid.Field(&req.FormulaTypeId, valid.Required, valid.Min(1)),
  30. valid.Field(&req.FormulaTypeName, valid.Required),
  31. valid.Field(&req.DataSourceId, valid.Required, valid.Min(1)),
  32. valid.Field(&req.DataSourceName, valid.Required),
  33. valid.Field(&req.IsShow, valid.Required, valid.Min(1), valid.Max(2)),
  34. valid.Field(&req.IsModify, valid.Required, valid.Min(1), valid.Max(2)),
  35. ); err != nil {
  36. apierr.AbortBadRequest(c, http.StatusBadRequest, err)
  37. return
  38. }
  39. if err := middleware.BackendOperation(c).OpsService.CreateFeedFormula(c, &req); err != nil {
  40. apierr.ClassifiedAbort(c, err)
  41. return
  42. }
  43. ginutil.JSONResp(c, &operationPb.CommonOK{
  44. Code: http.StatusOK,
  45. Msg: "ok",
  46. Data: &operationPb.Success{Success: true},
  47. })
  48. }
  49. // EditFeedFormula 编辑配方
  50. func EditFeedFormula(c *gin.Context) {
  51. var req operationPb.AddFeedFormulaRequest
  52. if err := c.BindJSON(&req); err != nil {
  53. apierr.AbortBadRequest(c, http.StatusBadRequest, err)
  54. return
  55. }
  56. if err := valid.ValidateStruct(&req,
  57. valid.Field(&req.Id, valid.Required, valid.Min(1)),
  58. valid.Field(&req.Name, valid.Required),
  59. valid.Field(&req.EncodeNumber, valid.Required, valid.Length(1, 30)),
  60. valid.Field(&req.Colour, valid.Required),
  61. valid.Field(&req.CattleCategoryId, valid.Required, valid.Min(1)),
  62. valid.Field(&req.CattleCategoryName, valid.Required),
  63. valid.Field(&req.FormulaTypeId, valid.Required, valid.Min(1)),
  64. valid.Field(&req.FormulaTypeName, valid.Required),
  65. valid.Field(&req.DataSourceId, valid.Required, valid.Min(1)),
  66. valid.Field(&req.DataSourceName, valid.Required),
  67. valid.Field(&req.IsShow, valid.Required, valid.Min(1), valid.Max(2)),
  68. valid.Field(&req.IsModify, valid.Required, valid.Min(1), valid.Max(2)),
  69. ); err != nil {
  70. apierr.AbortBadRequest(c, http.StatusBadRequest, err)
  71. return
  72. }
  73. if err := middleware.BackendOperation(c).OpsService.EditFeedFormula(c, &req); err != nil {
  74. apierr.ClassifiedAbort(c, err)
  75. return
  76. }
  77. ginutil.JSONResp(c, &operationPb.CommonOK{
  78. Code: http.StatusOK,
  79. Msg: "ok",
  80. Data: &operationPb.Success{Success: true},
  81. })
  82. }
  83. // AddFeedByFeedFormula 配方添加饲料
  84. func AddFeedByFeedFormula(c *gin.Context) {
  85. var req operationPb.GroupAddFeedFormulaDetail
  86. if err := c.BindJSON(&req); err != nil {
  87. apierr.AbortBadRequest(c, http.StatusBadRequest, err)
  88. return
  89. }
  90. for _, v := range req.List {
  91. if err := valid.ValidateStruct(&req,
  92. valid.Field(&req.FeedFormulaId, valid.Required, valid.Min(1)),
  93. valid.Field(&v.ForageId, valid.Required, valid.Min(1)),
  94. valid.Field(&v.ForageName, valid.Required, valid.Length(1, 30)),
  95. valid.Field(&v.Weight, valid.Required, valid.Min(1)),
  96. valid.Field(&v.Sort, valid.Required, valid.Min(1)),
  97. ); err != nil {
  98. apierr.AbortBadRequest(c, http.StatusBadRequest, err)
  99. return
  100. }
  101. }
  102. if err := middleware.BackendOperation(c).OpsService.AddFeedByFeedFormula(c, &req); err != nil {
  103. apierr.ClassifiedAbort(c, err)
  104. return
  105. }
  106. ginutil.JSONResp(c, &operationPb.CommonOK{
  107. Code: http.StatusOK,
  108. Msg: "ok",
  109. Data: &operationPb.Success{Success: true},
  110. })
  111. }
  112. // DetailBySort 配方详情排序
  113. func DetailBySort(c *gin.Context) {
  114. var req operationPb.GroupAddFeedFormulaDetail
  115. if err := c.BindJSON(&req); err != nil {
  116. apierr.AbortBadRequest(c, http.StatusBadRequest, err)
  117. return
  118. }
  119. for _, v := range req.List {
  120. if err := valid.ValidateStruct(&req,
  121. valid.Field(&req.FeedFormulaId, valid.Required, valid.Min(1)),
  122. valid.Field(&v.ForageId, valid.Required, valid.Min(1)),
  123. valid.Field(&v.ForageName, valid.Required, valid.Length(1, 30)),
  124. valid.Field(&v.Weight, valid.Required, valid.Min(1)),
  125. valid.Field(&v.Sort, valid.Required, valid.Min(1)),
  126. ); err != nil {
  127. apierr.AbortBadRequest(c, http.StatusBadRequest, err)
  128. return
  129. }
  130. }
  131. if err := middleware.BackendOperation(c).OpsService.FeedFormulaDetailBySort(c, &req); err != nil {
  132. apierr.ClassifiedAbort(c, err)
  133. return
  134. }
  135. ginutil.JSONResp(c, &operationPb.CommonOK{
  136. Code: http.StatusOK,
  137. Msg: "ok",
  138. Data: &operationPb.Success{Success: true},
  139. })
  140. }
  141. func SearchFeedFormulaList(c *gin.Context) {
  142. req := &operationPb.SearchFeedFormulaRequest{}
  143. if err := ginutil.BindProto(c, req); err != nil {
  144. apierr.AbortBadRequest(c, http.StatusBadRequest, err)
  145. return
  146. }
  147. req.Pagination = &operationPb.PaginationModel{
  148. Page: int32(c.GetInt(middleware.Page)),
  149. PageSize: int32(c.GetInt(middleware.PageSize)),
  150. PageOffset: int32(c.GetInt(middleware.PageOffset)),
  151. }
  152. res, err := middleware.BackendOperation(c).OpsService.SearchFeedFormulaList(c, req)
  153. if err != nil {
  154. apierr.ClassifiedAbort(c, err)
  155. return
  156. }
  157. ginutil.JSONResp(c, res)
  158. }
  159. func SearchFeedFormulaByForageList(c *gin.Context) {
  160. req := &operationPb.SearchFeedFormulaRequest{}
  161. if err := ginutil.BindProto(c, req); err != nil {
  162. apierr.AbortBadRequest(c, http.StatusBadRequest, err)
  163. return
  164. }
  165. req.Pagination = &operationPb.PaginationModel{
  166. Page: int32(c.GetInt(middleware.Page)),
  167. PageSize: int32(c.GetInt(middleware.PageSize)),
  168. PageOffset: int32(c.GetInt(middleware.PageOffset)),
  169. }
  170. res, err := middleware.BackendOperation(c).OpsService.SearchFeedFormulaList(c, req)
  171. if err != nil {
  172. apierr.ClassifiedAbort(c, err)
  173. return
  174. }
  175. ginutil.JSONResp(c, res)
  176. }
  177. func DeleteFeedFormula(c *gin.Context) {
  178. feedFormulaIdStr := c.Param("feed_formula_id")
  179. feedFormulaId, _ := strconv.Atoi(feedFormulaIdStr)
  180. if err := valid.Validate(feedFormulaId, valid.Required, valid.Min(1)); err != nil {
  181. apierr.ClassifiedAbort(c, err)
  182. return
  183. }
  184. if err := middleware.BackendOperation(c).OpsService.DeleteFeedFormula(c, int64(feedFormulaId)); err != nil {
  185. apierr.ClassifiedAbort(c, err)
  186. return
  187. }
  188. ginutil.JSONResp(c, &operationPb.CommonOK{
  189. Code: http.StatusOK,
  190. Msg: "ok",
  191. Data: &operationPb.Success{Success: true},
  192. })
  193. }
  194. func IsShowModifyFeedFormula(c *gin.Context) {
  195. var req operationPb.IsShowModifyFeedFormula
  196. if err := c.BindJSON(&req); err != nil {
  197. apierr.AbortBadRequest(c, http.StatusBadRequest, err)
  198. return
  199. }
  200. if err := valid.ValidateStruct(&req,
  201. valid.Field(&req.FeedFormulaId, valid.Required, valid.Min(1)),
  202. valid.Field(&req.IsShow, valid.Required, valid.Min(1), valid.Max(2)),
  203. ); err != nil {
  204. apierr.AbortBadRequest(c, http.StatusBadRequest, err)
  205. return
  206. }
  207. if err := middleware.BackendOperation(c).OpsService.IsShowFeedFormula(c, &req); err != nil {
  208. apierr.ClassifiedAbort(c, err)
  209. return
  210. }
  211. ginutil.JSONResp(c, &operationPb.CommonOK{
  212. Code: http.StatusOK,
  213. Msg: "ok",
  214. Data: &operationPb.Success{Success: true},
  215. })
  216. }
  217. func ExcelImportFeedFormula(c *gin.Context) {
  218. file, err := func(c *gin.Context) (multipart.File, error) {
  219. if c.ContentType() != "multipart/form-data" {
  220. return nil, xerr.Custom("invalid Content-Type")
  221. }
  222. if c.Request.Body == nil || int(c.Request.ContentLength) <= 0 {
  223. return nil, xerr.Custom("invalid body")
  224. }
  225. file, fileHeader, err := c.Request.FormFile("file")
  226. if err != nil {
  227. return nil, xerr.WithStack(err)
  228. }
  229. fileSuffix := path.Ext(path.Base(fileHeader.Filename))
  230. if fileSuffix != ".xlsx" {
  231. return nil, xerr.Custom("invalid file suffix")
  232. }
  233. return file, nil
  234. }(c)
  235. if err != nil {
  236. apierr.AbortBadRequest(c, http.StatusBadRequest, err)
  237. return
  238. }
  239. if err = middleware.BackendOperation(c).OpsService.ExcelImportFeedFormula(c, file); err != nil {
  240. apierr.ClassifiedAbort(c, err)
  241. return
  242. }
  243. ginutil.JSONResp(c, &operationPb.CommonOK{
  244. Code: http.StatusOK,
  245. Msg: "ok",
  246. Data: &operationPb.Success{Success: true},
  247. })
  248. }
  249. func ExcelExportFeedFormula(c *gin.Context) {
  250. req := &operationPb.SearchFeedFormulaRequest{}
  251. if err := c.BindJSON(req); err != nil {
  252. apierr.AbortBadRequest(c, http.StatusBadRequest, err)
  253. return
  254. }
  255. req.Pagination = &operationPb.PaginationModel{
  256. Page: int32(c.GetInt(middleware.Page)),
  257. PageSize: int32(c.GetInt(middleware.PageSize)),
  258. PageOffset: int32(c.GetInt(middleware.PageOffset)),
  259. }
  260. buffer, err := middleware.BackendOperation(c).OpsService.ExcelExportFeedFormula(c, req)
  261. if err != nil {
  262. apierr.ClassifiedAbort(c, err)
  263. return
  264. }
  265. c.Header("Content-Type", "application/octet-stream")
  266. c.Header("Content-Disposition", "attachment; filename="+(fmt.Sprintf("饲料表 %s.xlsx", time.Now().Format("200601021504"))))
  267. if _, err = c.Writer.Write(buffer.Bytes()); err != nil {
  268. apierr.ClassifiedAbort(c, err)
  269. return
  270. }
  271. ginutil.JSONResp(c, &operationPb.CommonOK{
  272. Code: http.StatusOK,
  273. Msg: "ok",
  274. Data: &operationPb.Success{Success: true},
  275. })
  276. }
  277. func ExcelTemplateFeedFormula(c *gin.Context) {
  278. buffer, err := middleware.BackendOperation(c).OpsService.ExcelTemplateFeedFormula(c)
  279. if err != nil {
  280. apierr.ClassifiedAbort(c, err)
  281. return
  282. }
  283. c.Header("Content-Type", "application/octet-stream")
  284. c.Header("Content-Disposition", "attachment; filename="+(fmt.Sprintf("饲料表 %s.xlsx", time.Now().Format("200601021504"))))
  285. if _, err = c.Writer.Write(buffer.Bytes()); err != nil {
  286. apierr.ClassifiedAbort(c, err)
  287. return
  288. }
  289. ginutil.JSONResp(c, &operationPb.CommonOK{
  290. Code: http.StatusOK,
  291. Msg: "ok",
  292. Data: &operationPb.Success{Success: true},
  293. })
  294. }
  295. func EncodeNumber(c *gin.Context) {
  296. ginutil.JSONResp(c, &operationPb.UniqueID{
  297. Code: http.StatusOK,
  298. Msg: "ok",
  299. Data: &operationPb.UniqueID_UniqueData{EncodeNumber: middleware.BackendOperation(c).OpsService.EncodeNumber(c)},
  300. })
  301. }
  302. // DistributeFeedFormula 饲料配方下发
  303. func DistributeFeedFormula(c *gin.Context) {
  304. req := &operationPb.DistributeFeedFormulaRequest{}
  305. if err := ginutil.BindProto(c, req); err != nil {
  306. apierr.AbortBadRequest(c, http.StatusBadRequest, err)
  307. return
  308. }
  309. if err := middleware.BackendOperation(c).OpsService.DistributeFeedFormula(c, req); err != nil {
  310. apierr.ClassifiedAbort(c, err)
  311. return
  312. }
  313. ginutil.JSONResp(c, &operationPb.CommonOK{
  314. Code: http.StatusOK,
  315. Msg: "ok",
  316. Data: &operationPb.Success{Success: true},
  317. })
  318. }
  319. // CancelDistributeFeedFormula 取消配方下发
  320. func CancelDistributeFeedFormula(c *gin.Context) {
  321. req := &operationPb.DistributeFeedFormulaRequest{}
  322. if err := ginutil.BindProto(c, req); err != nil {
  323. apierr.AbortBadRequest(c, http.StatusBadRequest, err)
  324. return
  325. }
  326. if err := middleware.BackendOperation(c).OpsService.CancelDistributeFeedFormula(c, req); err != nil {
  327. apierr.ClassifiedAbort(c, err)
  328. return
  329. }
  330. ginutil.JSONResp(c, &operationPb.CommonOK{
  331. Code: http.StatusOK,
  332. Msg: "ok",
  333. Data: &operationPb.Success{Success: true},
  334. })
  335. }
  336. // EditRecordFeedFormula 配方修改记录
  337. func EditRecordFeedFormula(c *gin.Context) {
  338. var req operationPb.EditRecodeFeedFormulaRequest
  339. if err := ginutil.BindProto(c, &req); err != nil {
  340. apierr.AbortBadRequest(c, http.StatusBadRequest, err)
  341. return
  342. }
  343. if err := valid.ValidateStruct(&req,
  344. valid.Field(&req.PastureId, valid.Required, valid.Min(1)),
  345. valid.Field(&req.StartTime, valid.Required, valid.Min(1), valid.Max(time.Now().Unix())),
  346. valid.Field(&req.EndTime, valid.Required, valid.Min(1), valid.Max(time.Now().Unix())),
  347. ); err != nil {
  348. apierr.AbortBadRequest(c, http.StatusBadRequest, err)
  349. return
  350. }
  351. if list, err := middleware.BackendOperation(c).OpsService.EditRecodeFeedFormula(c, &req); err != nil {
  352. apierr.ClassifiedAbort(c, err)
  353. return
  354. } else {
  355. ginutil.JSONResp(c, list)
  356. }
  357. }
  358. // Usage 配方使用情况
  359. func Usage(c *gin.Context) {
  360. var req operationPb.FeedFormulaUsageRequest
  361. if err := ginutil.BindProto(c, &req); err != nil {
  362. apierr.AbortBadRequest(c, http.StatusBadRequest, err)
  363. return
  364. }
  365. if err := valid.ValidateStruct(&req,
  366. valid.Field(&req.FeedFormulaId, valid.Required, valid.Min(1)),
  367. valid.Field(&req.StartTime, valid.Required),
  368. valid.Field(&req.EndTime, valid.Required),
  369. ); err != nil {
  370. apierr.AbortBadRequest(c, http.StatusBadRequest, err)
  371. return
  372. }
  373. if res, err := middleware.BackendOperation(c).OpsService.FeedFormulaUsage(c, &req); err != nil {
  374. apierr.ClassifiedAbort(c, err)
  375. return
  376. } else {
  377. ginutil.JSONResp(c, res)
  378. }
  379. }