statistic_service.go 20 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642
  1. package backend
  2. import (
  3. "bytes"
  4. "context"
  5. "encoding/json"
  6. "fmt"
  7. "kpt-tmr-group/model"
  8. "kpt-tmr-group/pkg/logger/zaplog"
  9. "kpt-tmr-group/pkg/xerr"
  10. operationPb "kpt-tmr-group/proto/go/backend/operation"
  11. "net/http"
  12. "sort"
  13. "sync"
  14. "github.com/xuri/excelize/v2"
  15. "go.uber.org/zap"
  16. )
  17. type PastureClientHandler func(ctx context.Context, pastureId int64, body interface{}) error
  18. // PastureDetailById 获取指定牧场详情
  19. func (s *StoreEntry) PastureDetailById(ctx context.Context, pastureId int64) (*model.GroupPasture, error) {
  20. result := &model.GroupPasture{Id: pastureId}
  21. if err := s.DB.Where("is_delete = ? and is_show = ?", operationPb.IsShow_OK, operationPb.IsShow_OK).First(result).Error; err != nil {
  22. return nil, xerr.WithStack(err)
  23. }
  24. return result, nil
  25. }
  26. func (s *StoreEntry) PastureHttpClient(ctx context.Context, apiUrl string, pastureId int64, body, response interface{}) error {
  27. pastureDetail, err := s.PastureDetailById(ctx, pastureId)
  28. if err != nil {
  29. zaplog.Error("PastureHttpClient", zap.Any("Err", err), zap.Int64("pastureId", pastureId))
  30. return xerr.Customf("该牧场数据错误,Err:%s", err)
  31. }
  32. pastureClient := model.NewPastureClient(pastureDetail)
  33. url := fmt.Sprintf("%s/%s", pastureDetail.Domain, apiUrl)
  34. result, err := pastureClient.DoPost(url, body)
  35. if err != nil {
  36. return xerr.WithStack(err)
  37. }
  38. zaplog.Info("PastureHttpClient", zap.String("url", url), zap.Any("request", body), zap.String("response", string(result)))
  39. if err = json.Unmarshal(result, response); err != nil {
  40. return xerr.WithStack(err)
  41. }
  42. return nil
  43. }
  44. // SearchFormulaEstimateList 配方评估
  45. func (s *StoreEntry) SearchFormulaEstimateList(ctx context.Context, req *operationPb.SearchFormulaEstimateRequest) (*model.PastureCommonResponse, error) {
  46. body := &model.PastureCommonRequest{
  47. Name: req.ApiName,
  48. Page: req.Pagination.Page,
  49. Offset: req.Pagination.PageOffset,
  50. PageCount: req.Pagination.PageSize,
  51. ReturnType: "Map",
  52. ParamMaps: &model.FormulaEstimateParams{
  53. PastureId: fmt.Sprintf("%d", req.PastureId),
  54. StartTime: req.StartTime,
  55. StopTime: req.EndTime,
  56. Search: fmt.Sprintf("%d", req.SearchType),
  57. TempletId: fmt.Sprintf("%d", req.TemplateId),
  58. Barid: fmt.Sprintf("%d", req.BarnId),
  59. },
  60. }
  61. response := &model.PastureCommonResponse{Data: &model.PastureCommonData{}}
  62. if err := s.PastureHttpClient(ctx, model.UrlDataByName, int64(req.PastureId), body, response); err != nil {
  63. return nil, xerr.WithStack(err)
  64. }
  65. return response, nil
  66. }
  67. // SearchInventoryStatistics 库存管理-库存统计
  68. func (s *StoreEntry) SearchInventoryStatistics(ctx context.Context, req *operationPb.SearchInventoryStatisticsRequest) (*model.PastureCommonResponse, error) {
  69. body := &model.PastureCommonRequest{
  70. Name: req.ApiName,
  71. Page: req.Pagination.Page,
  72. Offset: req.Pagination.PageOffset,
  73. PageCount: req.Pagination.PageSize,
  74. ReturnType: "Map",
  75. ParamMaps: &model.InventoryStatisticsParams{
  76. PastureId: fmt.Sprintf("%d", req.PastureId),
  77. StartTime: req.StartTime,
  78. StopTime: req.EndTime,
  79. FeedName: req.FeedName,
  80. },
  81. }
  82. response := &model.PastureCommonResponse{
  83. Data: &model.PastureCommonData{
  84. List: make([]*model.InventoryStatisticsList, 0),
  85. },
  86. }
  87. if err := s.PastureHttpClient(ctx, model.UrlDataByName, int64(req.PastureId), body, response); err != nil {
  88. return nil, xerr.WithStack(err)
  89. }
  90. return response, nil
  91. }
  92. // InventoryStatisticsExcelExport 库存管理-库存统计-报表导出
  93. func (s *StoreEntry) InventoryStatisticsExcelExport(ctx context.Context, req *operationPb.SearchInventoryStatisticsRequest) (*bytes.Buffer, error) {
  94. result, err := s.SearchInventoryStatistics(ctx, req)
  95. if err != nil {
  96. return nil, xerr.WithStack(err)
  97. }
  98. b, _ := json.Marshal(result.Data.List)
  99. inventoryStatisticsList := make([]*model.InventoryStatisticsList, 0)
  100. if err = json.Unmarshal(b, &inventoryStatisticsList); err != nil {
  101. return nil, xerr.Customf("牧场端返回数据错误")
  102. }
  103. file := excelize.NewFile()
  104. defer file.Close()
  105. streamWriter, err := file.NewStreamWriter(model.DefaultSheetName)
  106. if err != nil {
  107. return nil, xerr.WithStack(err)
  108. }
  109. // 表头
  110. titles := map[string][]interface{}{
  111. "A1": {"饲料名称", "期初", nil, "用料", nil, nil, nil, "期末"},
  112. "A2": {nil, "期初库存(kg)", "期初金额(元)", "入库重量(kg)", "系统出库重量(kg)", "人工用料重量(kg)", "损耗重量", "期末库存(kg)", "期末金额(kg)"},
  113. }
  114. for cell, values := range titles {
  115. if err = streamWriter.SetRow(cell, values); err != nil {
  116. return nil, xerr.WithStack(err)
  117. }
  118. }
  119. for i, item := range inventoryStatisticsList {
  120. cell, err := excelize.CoordinatesToCellName(1, i+3)
  121. if err != nil {
  122. zaplog.Error("InventoryStatisticsExcelExport CoordinatesToCellName", zap.Any("Err", err))
  123. continue
  124. }
  125. row := make([]interface{}, 0)
  126. row = append(row, item.FeedName, item.StartSum, item.StartPrice, item.LaidSum, item.UseSumXT, item.UseSumRG, item.UseSumXH, item.StopSum, item.StopPrice)
  127. if err = streamWriter.SetRow(cell, row); err != nil {
  128. return nil, xerr.WithStack(err)
  129. }
  130. }
  131. hvCell := map[string]string{
  132. "A1": "A2",
  133. "B1": "C1",
  134. "D1": "G1",
  135. "H1": "I1",
  136. }
  137. // 合并单元格
  138. for h, v := range hvCell {
  139. if err = streamWriter.MergeCell(h, v); err != nil {
  140. return nil, xerr.WithStack(err)
  141. }
  142. }
  143. // 修改样式
  144. /*style1, err := file.NewStyle(&excelize.Style{
  145. Fill: excelize.Fill{Type: "pattern", Color: []string{"#DFEBF6"}, Pattern: 1},
  146. Alignment: &excelize.Alignment{Horizontal: "center"},
  147. })
  148. if err != nil {
  149. return nil, xerr.WithStack(err)
  150. }
  151. style1, err := file.NewStyle(&excelize.Style{
  152. //Fill: excelize.Fill{Type: "pattern", Color: []string{"#DFEBF6"}, Pattern: 1},
  153. Alignment: &excelize.Alignment{Horizontal: "center"},
  154. })
  155. if err != nil {
  156. return nil, xerr.WithStack(err)
  157. }
  158. if err = file.SetCellStyle(model.DefaultSheetName, "A1", "A1", style1); err != nil {
  159. return nil, xerr.WithStack(err)
  160. }*/
  161. if err = streamWriter.Flush(); err != nil {
  162. return nil, xerr.WithStack(err)
  163. }
  164. return file.WriteToBuffer()
  165. }
  166. // SearchUserMaterialsStatistics 库存管理-用料分析
  167. func (s *StoreEntry) SearchUserMaterialsStatistics(ctx context.Context, req *operationPb.SearchUserMaterialsStatisticsRequest) (*model.PastureCommonResponse, error) {
  168. body := &model.PastureCommonRequest{
  169. Name: req.ApiName,
  170. Page: req.Pagination.Page,
  171. Offset: req.Pagination.PageOffset,
  172. PageCount: req.Pagination.PageSize,
  173. ReturnType: "Map",
  174. Checked: req.ErrorCheck,
  175. ParamMaps: &model.UserMaterialsStatisticsParams{
  176. PastureId: fmt.Sprintf("%d", req.PastureId),
  177. StartTime: req.StartTime,
  178. StopTime: req.EndTime,
  179. FeedName: req.FeedName,
  180. Typea: fmt.Sprintf("%d", req.TypeCheck),
  181. },
  182. }
  183. response := &model.PastureCommonResponse{Data: &model.PastureCommonData{
  184. List: &model.UserMaterialsList{},
  185. }}
  186. if err := s.PastureHttpClient(ctx, model.UrlReportForm, int64(req.PastureId), body, response); err != nil {
  187. return nil, xerr.WithStack(err)
  188. }
  189. return response, nil
  190. }
  191. func (s *StoreEntry) UserMaterialsStatisticsExcelExport(ctx context.Context, req *operationPb.SearchUserMaterialsStatisticsRequest) (*bytes.Buffer, error) {
  192. result, err := s.SearchUserMaterialsStatistics(ctx, req)
  193. if err != nil {
  194. return nil, xerr.WithStack(err)
  195. }
  196. b, _ := json.Marshal(result.Data.List)
  197. userMaterialsList := &model.UserMaterialsList{}
  198. if err = json.Unmarshal(b, userMaterialsList); err != nil {
  199. return nil, xerr.Customf("牧场端返回数据错误")
  200. }
  201. file := excelize.NewFile()
  202. defer file.Close()
  203. streamWriter, err := file.NewStreamWriter(model.DefaultSheetName)
  204. if err != nil {
  205. return nil, xerr.WithStack(err)
  206. }
  207. // 表数据
  208. excelValuesList := map[int][]interface{}{}
  209. cProp := make([]string, 0)
  210. for _, data2 := range userMaterialsList.Data2 {
  211. excelValuesList[1] = append(excelValuesList[1], data2.Label)
  212. for _, c := range data2.Children {
  213. excelValuesList[2] = append(excelValuesList[2], c.Label)
  214. cProp = append(cProp, c.Prop)
  215. }
  216. }
  217. for cell, values := range excelValuesList {
  218. if err = streamWriter.SetRow(fmt.Sprintf("A%d", cell), values); err != nil {
  219. return nil, xerr.WithStack(err)
  220. }
  221. delete(excelValuesList, cell)
  222. }
  223. for i, data1 := range userMaterialsList.Data1 {
  224. data1Map, ok := data1.(map[string]interface{})
  225. if ok {
  226. for _, prop := range cProp {
  227. newValue := ""
  228. if value, yes := data1Map[prop]; yes {
  229. if value != nil {
  230. switch v := value.(type) {
  231. case string:
  232. newValue = v
  233. case int, int8, int16, int32, int64, uint, uint8, uint16, uint32, uint64:
  234. newValue = fmt.Sprintf("%d", v)
  235. default:
  236. newValue = fmt.Sprintf("%v", v)
  237. }
  238. }
  239. excelValuesList[i+3] = append(excelValuesList[i+3], newValue)
  240. }
  241. }
  242. }
  243. }
  244. excelValuesKeys := make([]int, 0)
  245. for k, _ := range excelValuesList {
  246. excelValuesKeys = append(excelValuesKeys, k)
  247. }
  248. sort.Ints(excelValuesKeys)
  249. for _, v := range excelValuesKeys {
  250. if err = streamWriter.SetRow(fmt.Sprintf("A%d", v), excelValuesList[v]); err != nil {
  251. return nil, xerr.WithStack(err)
  252. }
  253. }
  254. if err = streamWriter.Flush(); err != nil {
  255. return nil, xerr.WithStack(err)
  256. }
  257. return file.WriteToBuffer()
  258. }
  259. // SearchPriceStatistics 库存管理-价格分析
  260. func (s *StoreEntry) SearchPriceStatistics(ctx context.Context, req *operationPb.SearchPriceStatisticsRequest) (*model.PastureCommonResponse, error) {
  261. body := &model.PastureCommonRequest{
  262. Name: req.ApiName,
  263. Page: req.Pagination.Page,
  264. Offset: req.Pagination.PageOffset,
  265. PageCount: req.Pagination.PageSize,
  266. ReturnType: "Map",
  267. ParamMaps: &model.PriceStatisticsParams{
  268. PastureId: fmt.Sprintf("%d", req.PastureId),
  269. StartTime: req.StartTime,
  270. StopTime: req.EndTime,
  271. FeedName: req.FeedName,
  272. },
  273. }
  274. response := &model.PastureCommonResponse{Data: &model.PastureCommonData{}}
  275. if err := s.PastureHttpClient(ctx, model.UrlReportForm, int64(req.PastureId), body, response); err != nil {
  276. return nil, xerr.WithStack(err)
  277. }
  278. return response, nil
  279. }
  280. // SearchFeedStatistics 饲喂效率-效率统计
  281. func (s *StoreEntry) SearchFeedStatistics(ctx context.Context, req *operationPb.SearchFeedStatisticsRequest) (*model.FeedStatisticsResponse, error) {
  282. res := &model.FeedStatisticsResponse{
  283. Code: http.StatusOK,
  284. Msg: "ok",
  285. Data: &model.FeedStatisticsData{
  286. List: make(map[int32]interface{}),
  287. },
  288. }
  289. if len(req.PastureId) <= 0 {
  290. return res, nil
  291. }
  292. wg := sync.WaitGroup{}
  293. wg.Add(len(req.PastureId))
  294. for _, v := range req.PastureId {
  295. go func(pastureId int32) {
  296. body := &model.PastureCommonRequest{
  297. Name: req.ApiName,
  298. Page: req.Pagination.Page,
  299. Offset: req.Pagination.PageOffset,
  300. PageCount: req.Pagination.PageSize,
  301. ReturnType: "Map",
  302. ParamMaps: &model.FeedStatisticsParams{
  303. PastureId: fmt.Sprintf("%d", pastureId),
  304. StartTime: req.StartTime,
  305. StopTime: req.StartTime,
  306. Date: req.StartTime,
  307. FeedTName: req.FormulaTemplate,
  308. BarName: req.BarnName,
  309. CowClass: req.CattleCategoryName,
  310. Times: fmt.Sprintf("%d", req.ClassNumber),
  311. },
  312. }
  313. response := &model.PastureCommonResponse{
  314. Data: &model.PastureCommonData{},
  315. }
  316. if err := s.PastureHttpClient(ctx, model.UrlDataByName, int64(pastureId), body, response); err != nil {
  317. zaplog.Error("SearchFeedStatistics", zap.Any("pastureId", pastureId), zap.Any("body", body), zap.Any("response", response))
  318. }
  319. if response.Code == http.StatusOK && response.Data.List != nil {
  320. res.Data.List[pastureId] = response.Data.List
  321. }
  322. }(v)
  323. wg.Done()
  324. }
  325. wg.Wait()
  326. return res, nil
  327. }
  328. // FeedChartStatistics 饲喂效率图表分析
  329. func (s *StoreEntry) FeedChartStatistics(ctx context.Context, req *operationPb.FeedChartStatisticsRequest) (*model.PastureCommonResponse, error) {
  330. body := &model.FeedChartParams{
  331. ParamMaps: &model.ParamMaps{
  332. PastureId: fmt.Sprintf("%d", req.PastureId),
  333. StartTime: req.StartTime,
  334. StopTime: req.StartTime,
  335. Status: req.Status,
  336. },
  337. }
  338. url, ok := model.UrlChart[req.ApiType]
  339. if !ok {
  340. return nil, xerr.Customf("错误的接口类型:%s", req.ApiType)
  341. }
  342. response := &model.PastureCommonResponse{Data: &model.PastureCommonData{}}
  343. if err := s.PastureHttpClient(ctx, url, int64(req.PastureId), body, response); err != nil {
  344. return nil, xerr.WithStack(err)
  345. }
  346. return response, nil
  347. }
  348. // CowsAnalysis 饲喂效率-牛群评估
  349. func (s *StoreEntry) CowsAnalysis(ctx context.Context, req *operationPb.CowsAnalysisRequest) (*model.PastureCommonResponse, error) {
  350. body := &model.PastureCommonRequest{
  351. Name: req.ApiName,
  352. Page: req.Pagination.Page,
  353. Offset: req.Pagination.PageOffset,
  354. PageCount: req.Pagination.PageSize,
  355. ReturnType: "Map",
  356. ParamMaps: &model.MixFeedStatisticsParams{
  357. PastureId: fmt.Sprintf("%d", req.PastureId),
  358. StartTime: req.StartTime,
  359. StopTime: req.StartTime,
  360. },
  361. }
  362. response := &model.PastureCommonResponse{Data: &model.PastureCommonData{}}
  363. if err := s.PastureHttpClient(ctx, model.UrlDataByName, int64(req.PastureId), body, response); err != nil {
  364. return nil, xerr.WithStack(err)
  365. }
  366. return response, nil
  367. }
  368. // SearchAccuracyAggStatistics 准确性分析-汇总分析
  369. func (s *StoreEntry) SearchAccuracyAggStatistics(ctx context.Context, req *operationPb.AccuracyAggStatisticsRequest) (*model.PastureCommonResponse, error) {
  370. body := &model.FeedChartParams{
  371. ParamMaps: &model.AccuracyAggParams{
  372. PastureId: fmt.Sprintf("%d", req.PastureId),
  373. StartTime: req.StartTime,
  374. StopTime: req.EndTime,
  375. FName: req.Fname,
  376. Sort: req.Sort,
  377. Status: req.Status,
  378. Times: req.Times,
  379. Genre: req.Genre,
  380. IsDate: req.Isdate,
  381. Hlwc1: req.Hlwc1,
  382. Hlwc2: req.Hlwc2,
  383. Hlzq1: req.Hlzq1,
  384. Hlzq2: req.Hlzq2,
  385. Hlzql1: req.Hlzql1,
  386. Hlzql2: req.Hlzql2,
  387. Slwc1: req.Slwc1,
  388. Slwc2: req.Slwc2,
  389. Slzq1: req.Slzq1,
  390. Slzq2: req.Slzq2,
  391. Slzql1: req.Slzql1,
  392. Slzql2: req.Slzql2,
  393. Error: req.IsError,
  394. },
  395. }
  396. response := &model.PastureCommonResponse{Data: &model.PastureCommonData{}}
  397. if err := s.PastureHttpClient(ctx, model.UrlSummary, int64(req.PastureId), body, response); err != nil {
  398. return nil, xerr.WithStack(err)
  399. }
  400. return response, nil
  401. }
  402. // SearchMixFeedStatistics 准确性分析-混料统计
  403. func (s *StoreEntry) SearchMixFeedStatistics(ctx context.Context, req *operationPb.MixFeedStatisticsRequest) (*model.PastureCommonResponse, error) {
  404. times := ""
  405. if req.ClassNumber > 0 {
  406. times = fmt.Sprintf("%d", req.ClassNumber)
  407. }
  408. body := &model.PastureCommonRequest{
  409. Name: req.ApiName,
  410. Page: req.Pagination.Page,
  411. Offset: req.Pagination.PageOffset,
  412. PageCount: req.Pagination.PageSize,
  413. ReturnType: "Map",
  414. ParamMaps: &model.MixFeedStatisticsParams{
  415. PastureId: fmt.Sprintf("%d", req.PastureId),
  416. StartTime: req.StartTime,
  417. StopTime: req.StartTime,
  418. TmrTName: req.EquipmentName,
  419. ProjName: req.TrainNumber,
  420. Times: times,
  421. ButtonType: req.ButtonType,
  422. TempletName: req.FormulationName,
  423. Isuse: req.IsUse,
  424. Hlwc1: req.Hlwc1,
  425. Hlwc2: req.Hlwc2,
  426. Hlzq1: req.Hlzq1,
  427. Hlzq2: req.Hlzq2,
  428. Hlzql1: req.Hlzql1,
  429. Hlzql2: req.Hlzql2,
  430. Error: req.IsError,
  431. },
  432. }
  433. response := &model.PastureCommonResponse{Data: &model.PastureCommonData{}}
  434. if err := s.PastureHttpClient(ctx, model.UrlDataByName, int64(req.PastureId), body, response); err != nil {
  435. return nil, xerr.WithStack(err)
  436. }
  437. return response, nil
  438. }
  439. // SearchSprinkleStatistics 准确性分析-撒料统计
  440. func (s *StoreEntry) SearchSprinkleStatistics(ctx context.Context, req *operationPb.SprinkleStatisticsRequest) (*model.PastureCommonResponse, error) {
  441. times := ""
  442. if req.ClassNumber > 0 {
  443. times = fmt.Sprintf("%d", req.ClassNumber)
  444. }
  445. body := &model.PastureCommonRequest{
  446. Name: req.ApiName,
  447. Page: req.Pagination.Page,
  448. Offset: req.Pagination.PageOffset,
  449. PageCount: req.Pagination.PageSize,
  450. ReturnType: "Map",
  451. ParamMaps: &model.SprinkleStatisticsParams{
  452. PastureId: fmt.Sprintf("%d", req.PastureId),
  453. StartTime: req.StartTime,
  454. StopTime: req.StartTime,
  455. TmrTName: req.EquipmentName,
  456. ProjName: req.TrainNumber,
  457. Times: times,
  458. ButtonType: req.ButtonType,
  459. TempletName: req.FormulationName,
  460. Isuse: req.IsUse,
  461. Fname: req.BarnName,
  462. Slwc1: req.Slwc1,
  463. Slwc2: req.Slwc2,
  464. Slzq2: req.Slzq2,
  465. Slzq1: req.Slzq1,
  466. Slzql1: req.Slzql1,
  467. Slzql2: req.Slzql2,
  468. Error: req.IsError,
  469. },
  470. }
  471. response := &model.PastureCommonResponse{Data: &model.PastureCommonData{}}
  472. if err := s.PastureHttpClient(ctx, model.UrlDataByName, int64(req.PastureId), body, response); err != nil {
  473. return nil, xerr.WithStack(err)
  474. }
  475. return response, nil
  476. }
  477. // SearchProcessAnalysis 过程分析
  478. func (s *StoreEntry) SearchProcessAnalysis(ctx context.Context, req *operationPb.ProcessAnalysisRequest) (*model.PastureCommonResponse, error) {
  479. body := &model.PastureCommonRequest{
  480. Name: req.ApiName,
  481. Page: req.Pagination.Page,
  482. Offset: req.Pagination.PageOffset,
  483. PageCount: req.Pagination.PageSize,
  484. ReturnType: "Map",
  485. ParamMaps: &model.ProcessAnalysisParams{
  486. PastureId: fmt.Sprintf("%d", req.PastureId),
  487. StartTime: req.StartTime,
  488. StopTime: req.StartTime,
  489. TmrTName: req.TmrName,
  490. IsCompleted: "",
  491. LpPlanType: fmt.Sprintf("%d", req.PlanType),
  492. FClassId: req.MixFeedType,
  493. Hlzq1: req.Hlzq1,
  494. Hlzq2: req.Hlzq2,
  495. Hlwc1: req.Hlwc1,
  496. Hlwc2: req.Hlwc2,
  497. Slwc1: req.Slwc1,
  498. Slwc2: req.Slwc2,
  499. Slzq2: req.Slzq2,
  500. Slzq1: req.Slzq1,
  501. Error: req.ErrorRange,
  502. },
  503. }
  504. response := &model.PastureCommonResponse{Data: &model.PastureCommonData{}}
  505. if err := s.PastureHttpClient(ctx, model.UrlProcess, int64(req.PastureId), body, response); err != nil {
  506. return nil, xerr.WithStack(err)
  507. }
  508. return response, nil
  509. }
  510. // GetDataByName 共同接口
  511. func (s *StoreEntry) GetDataByName(ctx context.Context, req *operationPb.GetDataByNameRequest) (*model.PastureCommonResponse, error) {
  512. body := &model.PastureCommonRequest{
  513. Name: req.ApiName,
  514. ParamMaps: &model.GetDataByNameParams{
  515. PastureId: fmt.Sprintf("%d", req.PastureId),
  516. StartTime: req.StartTime,
  517. StopTime: req.StartTime,
  518. },
  519. }
  520. response := &model.PastureCommonResponse{Data: &model.PastureCommonData{}}
  521. if err := s.PastureHttpClient(ctx, model.UrlDataByName, int64(req.PastureId), body, response); err != nil {
  522. return nil, xerr.WithStack(err)
  523. }
  524. return response, nil
  525. }
  526. // GetTrainNumber 获取班次
  527. func (s *StoreEntry) GetTrainNumber(ctx context.Context, req *operationPb.TrainNumberRequest) (*operationPb.TrainNumberResponse, error) {
  528. body := &model.PastureCommonRequest{
  529. Name: req.ApiName,
  530. Page: req.Pagination.Page,
  531. Offset: req.Pagination.PageOffset,
  532. PageCount: req.Pagination.PageSize,
  533. ReturnType: "Map",
  534. ParamMaps: &model.TrainNumberParams{
  535. PastureId: fmt.Sprintf("%d", req.PastureId),
  536. InfoRName: req.InfoName,
  537. },
  538. }
  539. response := &model.PastureCommonResponse{Data: &model.PastureCommonData{}}
  540. if err := s.PastureHttpClient(ctx, model.UrlDataByName, int64(req.PastureId), body, response); err != nil {
  541. return nil, xerr.WithStack(err)
  542. }
  543. result := &operationPb.TrainNumberResponse{
  544. Code: http.StatusOK,
  545. Msg: "ok",
  546. Data: &operationPb.TrainNumberData{List: make([]*operationPb.FormulaOptionEnum, 0)},
  547. }
  548. if response.Data.List == nil {
  549. return result, nil
  550. }
  551. b, _ := json.Marshal(response.Data.List)
  552. trainNumberList := make([]*model.TrainNumberList, 0)
  553. if err := json.Unmarshal(b, &trainNumberList); err != nil {
  554. return nil, xerr.WithStack(err)
  555. }
  556. formulaOption := make([]*operationPb.FormulaOptionEnum, 0)
  557. if len(trainNumberList) > 0 {
  558. infoValue := trainNumberList[0].InfoValue
  559. switch infoValue {
  560. case "1":
  561. formulaOption = append(formulaOption, &operationPb.FormulaOptionEnum{
  562. Value: 1,
  563. Label: "第一班",
  564. })
  565. case "2":
  566. formulaOption = append(formulaOption, &operationPb.FormulaOptionEnum{
  567. Value: 1,
  568. Label: "第一班",
  569. }, &operationPb.FormulaOptionEnum{
  570. Value: 2,
  571. Label: "第二班",
  572. })
  573. case "3":
  574. formulaOption = append(formulaOption, &operationPb.FormulaOptionEnum{
  575. Value: 1,
  576. Label: "第一班",
  577. }, &operationPb.FormulaOptionEnum{
  578. Value: 2,
  579. Label: "第二班",
  580. }, &operationPb.FormulaOptionEnum{
  581. Value: 3,
  582. Label: "第三班",
  583. })
  584. }
  585. }
  586. result.Data.List = formulaOption
  587. return result, nil
  588. }