statistic_service.go 20 KB

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