event_breed.go 24 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752
  1. package backend
  2. import (
  3. "context"
  4. "encoding/json"
  5. "fmt"
  6. "kpt-pasture/model"
  7. "kpt-pasture/util"
  8. "net/http"
  9. "strings"
  10. "time"
  11. "gitee.com/xuyiping_admin/pkg/logger/zaplog"
  12. "go.uber.org/zap"
  13. pasturePb "gitee.com/xuyiping_admin/go_proto/proto/go/backend/cow"
  14. "gitee.com/xuyiping_admin/pkg/xerr"
  15. "gorm.io/gorm"
  16. )
  17. func (s *StoreEntry) CalvingList(ctx context.Context, req *pasturePb.SearchEventRequest, pagination *pasturePb.PaginationModel) (*pasturePb.SearchLavingEventResponse, error) {
  18. lavingList := make([]*model.EventCalvingList, 0)
  19. var count int64 = 0
  20. pref := s.DB.Table(fmt.Sprintf("%s as a", new(model.EventCalving).TableName())).
  21. Select(`a.*,b.name as pen_name,c.name as staff_member_name`).
  22. Joins(fmt.Sprintf("JOIN %s AS b on a.pen_id = b.id", new(model.Pen).TableName())).
  23. Joins(fmt.Sprintf("JOIN %s AS c on a.operation_id = c.id", new(model.SystemUser).TableName()))
  24. if len(req.CowId) > 0 {
  25. cowIds := strings.Split(req.CowId, ",")
  26. pref.Where("a.cow_id IN ?", cowIds)
  27. }
  28. if err := pref.Order("a.id desc").
  29. Count(&count).Limit(int(pagination.PageSize)).
  30. Offset(int(pagination.PageOffset)).
  31. Find(&lavingList).Error; err != nil {
  32. return nil, xerr.WithStack(err)
  33. }
  34. calvingIds := make([]int64, 0)
  35. for _, v := range lavingList {
  36. calvingIds = append(calvingIds, v.Id)
  37. }
  38. calvingCalfList := make([]*model.CalvingCalf, 0)
  39. if err := s.DB.Model(new(model.CalvingCalf)).
  40. Where("calving_id IN ?", calvingIds).
  41. Find(&calvingCalfList).Error; err != nil {
  42. return nil, xerr.WithStack(err)
  43. }
  44. return &pasturePb.SearchLavingEventResponse{
  45. Code: http.StatusOK,
  46. Message: "ok",
  47. Data: &pasturePb.SearchLavingData{
  48. List: model.EventCalvingListSlice(lavingList).ToPB(calvingCalfList),
  49. Total: int32(count),
  50. PageSize: pagination.PageSize,
  51. Page: pagination.Page,
  52. },
  53. }, nil
  54. }
  55. func (s *StoreEntry) CalvingCreate(ctx context.Context, req *pasturePb.EventCalving) error {
  56. if len(req.CowId) <= 0 {
  57. return xerr.Custom("请选择相关牛只")
  58. }
  59. cowList, err := s.ParseCowIds(ctx, req.CowId)
  60. if err != nil {
  61. return xerr.WithStack(err)
  62. }
  63. if len(cowList) <= 0 {
  64. return xerr.Custom("请选择相关牛只")
  65. }
  66. cow := cowList[0]
  67. newEventCalving := &model.EventCalving{}
  68. if err = s.DB.Model(new(model.EventCalving)).Where("cow_id = ?", cow.Id).Where("lact = ?", cow.Lact).First(newEventCalving).Error; err != nil {
  69. return xerr.Custom("该母牛信息不存在")
  70. }
  71. if err = s.DB.Transaction(func(tx *gorm.DB) error {
  72. // 母牛信息
  73. if err = tx.Model(new(model.EventCalving)).
  74. Where("id = ?", newEventCalving.Id).
  75. Updates(map[string]interface{}{
  76. "reality_day": int64(req.CalvingAt),
  77. "day_age": cow.DayAge,
  78. "lact": cow.Lact + 1,
  79. "pregnancy_age": cow.PregnancyAge,
  80. "calving_level": req.CalvingLevel,
  81. "bull_number": cow.LastBullNumber,
  82. "child_number": len(req.CalfItemList),
  83. }).
  84. Error; err != nil {
  85. return xerr.WithStack(err)
  86. }
  87. // 犊牛信息
  88. newCalvingCalfList := model.NewEventCalvingCalf(cow.Id, newEventCalving.Id, req)
  89. for _, v := range newCalvingCalfList {
  90. if v.IsLive == pasturePb.IsShow_No || v.IsAdoption == pasturePb.IsShow_No {
  91. continue
  92. }
  93. // 留养犊牛
  94. newCow := model.NewCalfCow(cow.Id, cow.LastBullNumber, v)
  95. if err = tx.Create(newCow).Error; err != nil {
  96. return xerr.WithStack(err)
  97. }
  98. v.CowId = newCow.Id
  99. }
  100. if err = tx.Create(newCalvingCalfList).Error; err != nil {
  101. return xerr.WithStack(err)
  102. }
  103. if err = tx.Model(new(model.Cow)).Where("id = ?", cow.Id).
  104. Updates(map[string]interface{}{
  105. "calving_age": 0,
  106. "mating_times": 0,
  107. "lact": cow.Lact + 1,
  108. "breed_status": pasturePb.BreedStatus_Calving,
  109. "is_pregnant": pasturePb.IsShow_No,
  110. "last_calving_at": int64(req.CalvingAt),
  111. }).Error; err != nil {
  112. return xerr.WithStack(err)
  113. }
  114. return nil
  115. }); err != nil {
  116. return xerr.WithStack(err)
  117. }
  118. return nil
  119. }
  120. func (s *StoreEntry) PregnantCheckList(ctx context.Context, req *pasturePb.SearchEventRequest, pagination *pasturePb.PaginationModel) (*pasturePb.PregnantCheckEventResponse, error) {
  121. pregnantCheckList := make([]*model.EventPregnantCheck, 0)
  122. var count int64 = 0
  123. pref := s.DB.Table(new(model.EventPregnantCheck).TableName())
  124. if len(req.CowId) > 0 {
  125. cowIds := strings.Split(req.CowId, ",")
  126. pref.Where("cow_id IN ?", cowIds)
  127. }
  128. if err := pref.Order("id desc").
  129. Count(&count).Limit(int(pagination.PageSize)).
  130. Offset(int(pagination.PageOffset)).
  131. Find(&pregnantCheckList).Error; err != nil {
  132. return nil, xerr.WithStack(err)
  133. }
  134. pregnantCheckResultMap := s.PregnantCheckResultMap()
  135. pregnantCheckMethodMap := s.PregnantCheckMethodMap()
  136. return &pasturePb.PregnantCheckEventResponse{
  137. Code: http.StatusOK,
  138. Message: "ok",
  139. Data: &pasturePb.SearchPregnantCheckData{
  140. List: model.EventPregnantCheckSlice(pregnantCheckList).ToPB(pregnantCheckResultMap, pregnantCheckMethodMap),
  141. Total: int32(count),
  142. PageSize: pagination.PageSize,
  143. Page: pagination.Page,
  144. },
  145. }, nil
  146. }
  147. func (s *StoreEntry) PregnantCheckCreate(ctx context.Context, req *pasturePb.EventPregnantCheck) error {
  148. eventCheckModel, err := s.PregnantCheckCreateCheck(ctx, req)
  149. if err != nil {
  150. return xerr.WithStack(err)
  151. }
  152. // 更新怀孕牛只
  153. updatePregnantCowIds := make([]int64, 0)
  154. // 更新流产牛只
  155. updateAbortCowIds := make([]int64, 0)
  156. // 更新空怀牛只
  157. updateEmptyCowIds := make([]int64, 0)
  158. cowIds := make([]int64, 0)
  159. for _, cow := range eventCheckModel.CowList {
  160. itemEventPregnantCheck := &model.EventPregnantCheck{}
  161. if err = s.DB.Model(new(model.EventPregnantCheck)).
  162. Where("lact = ?", cow.Lact).
  163. Where("cow_id = ?", cow.Id).
  164. Order("id desc").
  165. First(itemEventPregnantCheck).Error; err != nil {
  166. return xerr.WithStack(err)
  167. }
  168. if req.PregnantCheckResult == pasturePb.PregnantCheckResult_Pregnant {
  169. updatePregnantCowIds = append(updatePregnantCowIds, cow.Id)
  170. }
  171. if req.PregnantCheckResult == pasturePb.PregnantCheckResult_UnPregnant {
  172. if itemEventPregnantCheck.PregnantCheckName == model.PregnantCheckForFirst {
  173. updateAbortCowIds = append(updateAbortCowIds, cow.Id)
  174. }
  175. if itemEventPregnantCheck.PregnantCheckName == model.PregnantCheckForSecond {
  176. updateEmptyCowIds = append(updateEmptyCowIds, cow.Id)
  177. }
  178. }
  179. cowIds = append(cowIds, cow.Id)
  180. }
  181. if err = s.DB.Transaction(func(tx *gorm.DB) error {
  182. // 更新孕检事件表
  183. if err = tx.Model(new(model.EventPregnantCheck)).
  184. Where("id IN ?", cowIds).
  185. Updates(map[string]interface{}{
  186. "reality_day": req.PregnantCheckAt,
  187. "pregnant_check_result": req.PregnantCheckResult,
  188. "pregnant_check_method": req.PregnantCheckMethod,
  189. "operation_id": eventCheckModel.OperationUser.Id,
  190. "operation_name": eventCheckModel.OperationUser.Name,
  191. "message_id": eventCheckModel.CurrentUser.Id,
  192. "message_name": eventCheckModel.CurrentUser.Name,
  193. "remarks": req.Remarks,
  194. "is_pregnant": req.PregnantCheckResult,
  195. }).Error; err != nil {
  196. return xerr.WithStack(err)
  197. }
  198. // 更新牛只信息和配种事件表怀孕状态
  199. if len(updatePregnantCowIds) > 0 {
  200. if err = tx.Model(&model.Cow{}).
  201. Where("cow_id IN ?", updatePregnantCowIds).
  202. Updates(map[string]interface{}{
  203. "breed_status": pasturePb.BreedStatus_Pregnant,
  204. "last_pregnant_check_at": req.PregnantCheckAt,
  205. "is_pregnant": pasturePb.IsShow_Ok,
  206. }).Error; err != nil {
  207. return xerr.WithStack(err)
  208. }
  209. if err = tx.Model(&model.EventMating{}).Where("cow_id IN ?", updatePregnantCowIds).
  210. Group("cow_id").Select("MAX(id) as id").
  211. Updates(map[string]interface{}{
  212. "mating_result": pasturePb.MatingResult_Pregnant,
  213. "mating_result_at": req.PregnantCheckAt,
  214. }).Error; err != nil {
  215. return xerr.WithStack(err)
  216. }
  217. }
  218. // 更新牛只信息和配种事件表流产状态
  219. if len(updateAbortCowIds) > 0 {
  220. if err = tx.Model(&model.Cow{}).Where("cow_id IN ?", updateAbortCowIds).
  221. Updates(map[string]interface{}{
  222. "breed_status": pasturePb.BreedStatus_Abort,
  223. "last_pregnant_check_at": req.PregnantCheckAt,
  224. "is_pregnant": pasturePb.IsShow_Ok,
  225. }).Error; err != nil {
  226. return xerr.WithStack(err)
  227. }
  228. if err = tx.Model(&model.EventMating{}).Where("cow_id IN ?", updateAbortCowIds).
  229. Group("cow_id").Select("MAX(id) as id").
  230. Updates(map[string]interface{}{
  231. "mating_result": pasturePb.MatingResult_Abort,
  232. "mating_result_at": req.PregnantCheckAt,
  233. }).Error; err != nil {
  234. return xerr.WithStack(err)
  235. }
  236. }
  237. // 更新牛只信息和配种事件表空怀状态
  238. if len(updateEmptyCowIds) > 0 {
  239. if err = tx.Model(&model.Cow{}).Where("cow_id IN ?", updateEmptyCowIds).
  240. Updates(map[string]interface{}{
  241. "breed_status": pasturePb.BreedStatus_Empty,
  242. "last_pregnant_check_at": req.PregnantCheckAt,
  243. "is_pregnant": pasturePb.IsShow_Ok,
  244. }).Error; err != nil {
  245. return xerr.WithStack(err)
  246. }
  247. if err = tx.Model(&model.EventMating{}).Where("cow_id IN ?", updateEmptyCowIds).
  248. Group("cow_id").Select("MAX(id) as id").
  249. Updates(map[string]interface{}{
  250. "mating_result": pasturePb.MatingResult_Empty,
  251. "mating_result_at": req.PregnantCheckAt,
  252. }).Error; err != nil {
  253. return xerr.WithStack(err)
  254. }
  255. }
  256. return nil
  257. }); err != nil {
  258. return xerr.WithStack(err)
  259. }
  260. return nil
  261. }
  262. func (s *StoreEntry) MatingList(ctx context.Context, req *pasturePb.SearchEventRequest, pagination *pasturePb.PaginationModel) (*pasturePb.MatingEventResponse, error) {
  263. matingList := make([]*model.EventMating, 0)
  264. var count int64 = 0
  265. pref := s.DB.Model(new(model.EventMating)).
  266. Where("mating_result > ?", pasturePb.MatingResult_Invalid).
  267. Where("status = ?", pasturePb.IsShow_No)
  268. if len(req.CowId) > 0 {
  269. cowIds := strings.Split(req.CowId, ",")
  270. pref.Where("cow_id IN ?", cowIds)
  271. }
  272. if err := pref.Order("id desc").
  273. Count(&count).Limit(int(pagination.PageSize)).
  274. Offset(int(pagination.PageOffset)).
  275. Find(&matingList).Error; err != nil {
  276. return nil, xerr.WithStack(err)
  277. }
  278. exposeEstrusTypeMap := s.ExposeEstrusTypeMap()
  279. return &pasturePb.MatingEventResponse{
  280. Code: http.StatusOK,
  281. Message: "ok",
  282. Data: &pasturePb.SearchMatingData{
  283. List: model.EventMatingSlice(matingList).ToPB(exposeEstrusTypeMap),
  284. Total: int32(count),
  285. PageSize: pagination.PageSize,
  286. Page: pagination.Page,
  287. },
  288. }, nil
  289. }
  290. func (s *StoreEntry) MatingCreate(ctx context.Context, req *pasturePb.EventMating) error {
  291. eventCheckModel, err := s.MatingCreateCheck(ctx, req)
  292. if err != nil {
  293. return xerr.WithStack(err)
  294. }
  295. frozenSemen := &model.FrozenSemen{}
  296. if err = s.DB.Where("bull_id = ?", req.FrozenSemenNumber).
  297. First(frozenSemen).Error; err != nil {
  298. return xerr.Custom("未找到冻精信息")
  299. }
  300. // 更新复配的牛只
  301. matingReMatchIds := make([]int64, 0)
  302. // 新建配种信息的牛只
  303. newMatingList := make([]*model.EventMating, 0)
  304. // 更新配种信息牛只
  305. updateMatingList := make([]int64, 0)
  306. // 所有牛只
  307. cowIds := make([]int64, 0)
  308. // 需要更新配次的牛只
  309. matingTimes := make([]*model.MatingTimes, 0)
  310. // 需要更新空怀的牛只
  311. emptyCowIds := make([]int64, 0)
  312. for _, cow := range eventCheckModel.CowList {
  313. // 1. 第一次配种,创建配种信息(自然发情牛只,未经过同期的牛只)
  314. if cow.LastMatingAt <= 0 {
  315. newMatingList = append(newMatingList, model.NewEventMating2(cow, req, eventCheckModel.CurrentUser))
  316. continue
  317. }
  318. eventMatingHistory, err := s.GetEventMatingIsExIstByCowId(ctx, cow)
  319. if err != nil {
  320. return xerr.WithStack(err)
  321. }
  322. // 2. 同期更新配种信息 牛号&& 胎次 && 未配 && 配种结果未知 ==> 同期初配
  323. if eventMatingHistory.Status == pasturePb.IsShow_No &&
  324. eventMatingHistory.MatingResult == pasturePb.MatingResult_Unknown {
  325. updateMatingList = append(updateMatingList, eventMatingHistory.Id)
  326. continue
  327. }
  328. lastMatingAt := time.Unix(cow.LastMatingAt, 0)
  329. currentMatingAt := time.Unix(int64(req.MatingAt), 0)
  330. daysBetween := util.DaysBetween(currentMatingAt.Unix(), lastMatingAt.Unix())
  331. // 3. 如何两次配种时间为连续两天之内&&有过一次配种记录,则更新为复配状态 牛号&& 胎次 && 已配 && 配种结果未知 ==> 复配
  332. if (currentMatingAt.Format(model.LayoutDate2) == lastMatingAt.Format(model.LayoutDate2) || daysBetween == 1) &&
  333. eventMatingHistory.Status == pasturePb.IsShow_Ok &&
  334. eventMatingHistory.MatingResult == pasturePb.MatingResult_Unknown {
  335. matingReMatchIds = append(matingReMatchIds, eventMatingHistory.Id)
  336. } else {
  337. matingTimes = append(matingTimes, &model.MatingTimes{
  338. Mt: cow.MatingTimes + 1,
  339. CowId: cow.Id,
  340. EventMatingId: eventMatingHistory.Id,
  341. })
  342. }
  343. // 4. 提交的配种数据中,如何定位出空怀的牛只,
  344. if (eventMatingHistory.MatingResult == pasturePb.MatingResult_Unknown || eventMatingHistory.MatingResult == pasturePb.MatingResult_ReMatch) &&
  345. daysBetween >= 2 {
  346. emptyCowIds = append(emptyCowIds, eventMatingHistory.Id)
  347. newMatingList = append(newMatingList, model.NewEventMating2(cow, req, eventCheckModel.CurrentUser))
  348. }
  349. cowIds = append(cowIds, cow.Id)
  350. }
  351. zaplog.Info("MatingCreate",
  352. zap.Any("cowIds", cowIds),
  353. zap.Any("updateMatingList", updateMatingList),
  354. zap.Any("newMatingList", newMatingList),
  355. zap.Any("matingReMatchIds", matingReMatchIds),
  356. zap.Any("req", req),
  357. )
  358. if len(cowIds) != len(updateMatingList)+len(matingReMatchIds)+len(newMatingList) {
  359. return xerr.Custom("配种信息有误")
  360. }
  361. itemFrozenSemenLog := model.NewEventFrozenSemenLog(req)
  362. if err = s.DB.Transaction(func(tx *gorm.DB) error {
  363. // 更新配种事件数据(初配)
  364. if len(updateMatingList) > 0 {
  365. if err = tx.Model(new(model.EventMating)).
  366. Where("id IN ?", updateMatingList).
  367. Where("status = ?", pasturePb.IsShow_No).
  368. Updates(map[string]interface{}{
  369. "mating_result": pasturePb.MatingResult_Unknown,
  370. "status": pasturePb.IsShow_Ok,
  371. "reality_day": int64(req.MatingAt),
  372. "frozen_semen_number": req.FrozenSemenNumber,
  373. "operation_id": eventCheckModel.OperationUser.Id,
  374. "operation_name": eventCheckModel.OperationUser.Name,
  375. "message_id": eventCheckModel.CurrentUser.Id,
  376. "message_name": eventCheckModel.CurrentUser.Name,
  377. }).Error; err != nil {
  378. return xerr.WithStack(err)
  379. }
  380. }
  381. // 更新已配种的牛只为复配状态
  382. if len(matingReMatchIds) > 0 {
  383. if err = tx.Model(new(model.EventMating)).
  384. Where("id IN ?", matingReMatchIds).
  385. Where("mating_result = ?", pasturePb.MatingResult_Unknown).
  386. Where("status = ?", pasturePb.IsShow_Ok).
  387. Update("mating_result", pasturePb.MatingResult_ReMatch).
  388. Error; err != nil {
  389. return xerr.WithStack(err)
  390. }
  391. }
  392. // 更新上次已配种的牛只为空怀
  393. if len(emptyCowIds) > 0 {
  394. if err = tx.Model(new(model.EventMating)).
  395. Where("id IN ?", emptyCowIds).
  396. Update("mating_result", pasturePb.BreedStatus_Empty).
  397. Error; err != nil {
  398. return xerr.WithStack(err)
  399. }
  400. }
  401. // 创建配种事件数据
  402. if len(newMatingList) > 0 {
  403. if err = tx.Create(newMatingList).Error; err != nil {
  404. return xerr.WithStack(err)
  405. }
  406. }
  407. // 创建冻精使用记录日志
  408. if err = tx.Create(itemFrozenSemenLog).Error; err != nil {
  409. return xerr.WithStack(err)
  410. }
  411. // 如果有同期牛只,则修改为已结束状态
  412. if err = tx.Model(new(model.CowSameTime)).
  413. Where("cow_id IN ?", cowIds).
  414. UpdateColumn("same_time_status", pasturePb.SameTimeStatus_End).Error; err != nil {
  415. return xerr.WithStack(err)
  416. }
  417. // 更新牛只的繁殖状态为配种
  418. if err = tx.Model(new(model.Cow)).
  419. Where("id IN ?", cowIds).
  420. Where("admission_status = ?", pasturePb.AdmissionStatus_Admission).
  421. Updates(map[string]interface{}{
  422. "breed_status": pasturePb.BreedStatus_Breeding,
  423. "last_bull_number": req.FrozenSemenNumber,
  424. "last_mating_at": int64(req.MatingAt),
  425. }).Error; err != nil {
  426. return xerr.WithStack(err)
  427. }
  428. // 减去精液的数量
  429. if err = tx.Model(frozenSemen).
  430. Where("id = ?", frozenSemen.Id).
  431. Where("quantity > 0").
  432. UpdateColumn("quantity",
  433. gorm.Expr("quantity - ?", req.FrozenSemenCount),
  434. ).Error; err != nil {
  435. return xerr.WithStack(err)
  436. }
  437. // 更新配次
  438. if len(matingTimes) > 0 {
  439. for _, mt := range matingTimes {
  440. if err = tx.Model(new(model.EventMating)).
  441. Where("id = ?", mt.EventMatingId).
  442. Update("mating_times", mt.Mt).Error; err != nil {
  443. return xerr.WithStack(err)
  444. }
  445. if err = tx.Model(new(model.Cow)).
  446. Where("id = ?", mt.CowId).
  447. Where("admission_status = ?", pasturePb.AdmissionStatus_Admission).
  448. Where("breed_status = ?", pasturePb.BreedStatus_Breeding).
  449. Update("mating_times", mt.Mt).Error; err != nil {
  450. return xerr.WithStack(err)
  451. }
  452. }
  453. }
  454. return nil
  455. }); err != nil {
  456. return xerr.WithStack(err)
  457. }
  458. return nil
  459. }
  460. func (s *StoreEntry) EstrusList(ctx context.Context, req *pasturePb.SearchEventRequest, pagination *pasturePb.PaginationModel) (*pasturePb.EstrusEventResponse, error) {
  461. estrusList := make([]*model.EventEstrus, 0)
  462. var count int64 = 0
  463. pref := s.DB.Table(new(model.EventEstrus).TableName())
  464. if len(req.CowId) > 0 {
  465. cowIds := strings.Split(req.CowId, ",")
  466. pref.Where("cow_id IN ?", cowIds)
  467. }
  468. if err := pref.Order("id desc").
  469. Count(&count).Limit(int(pagination.PageSize)).
  470. Offset(int(pagination.PageOffset)).
  471. Find(&estrusList).Error; err != nil {
  472. return nil, xerr.WithStack(err)
  473. }
  474. return &pasturePb.EstrusEventResponse{
  475. Code: http.StatusOK,
  476. Message: "ok",
  477. Data: &pasturePb.SearchEstrusData{
  478. List: model.EstrusSlice(estrusList).ToPB(),
  479. Total: int32(count),
  480. PageSize: pagination.PageSize,
  481. Page: pagination.Page,
  482. },
  483. }, nil
  484. }
  485. func (s *StoreEntry) EstrusCreate(ctx context.Context, req *pasturePb.EventEstrus) error {
  486. if len(req.CowId) <= 0 {
  487. return xerr.Custom("请选择相关牛只")
  488. }
  489. cowList, err := s.ParseCowIds(ctx, req.CowId)
  490. if err != nil {
  491. return xerr.WithStack(err)
  492. }
  493. estrusList := make([]*model.EventEstrus, 0)
  494. currentUser, err := s.GetCurrentSystemUser(ctx)
  495. if err != nil {
  496. return xerr.Custom("获取当前登录用户失败")
  497. }
  498. for _, cow := range cowList {
  499. operationUser, err := s.GetSystemUserById(ctx, int64(req.OperationId))
  500. if err != nil {
  501. return xerr.WithStack(err)
  502. }
  503. req.OperationName = operationUser.Name
  504. estrusList = append(estrusList, model.NewEventEstrus(cow, currentUser, req))
  505. }
  506. if err = s.DB.Create(estrusList).Error; err != nil {
  507. return xerr.WithStack(err)
  508. }
  509. return nil
  510. }
  511. func (s *StoreEntry) SameTimeCreate(ctx context.Context, req *pasturePb.EventSameTime) error {
  512. if _, err := s.GetSameTimeById(ctx, int64(req.SameTimeId)); err != nil {
  513. return xerr.WithStack(err)
  514. }
  515. drugs := &model.Drugs{}
  516. var err error
  517. if req.DrugsId > 0 {
  518. if drugs, err = s.GetDrugsById(ctx, int64(req.DrugsId)); err != nil {
  519. return xerr.WithStack(err)
  520. }
  521. }
  522. operationUser, err := s.GetSystemUserById(ctx, int64(req.OperationId))
  523. if err != nil {
  524. return xerr.WithStack(err)
  525. }
  526. if err = s.DB.Transaction(func(tx *gorm.DB) error {
  527. if err = tx.Model(new(model.CowSameTime)).
  528. Where("cow_id = ?", req.CowId).
  529. Where("same_time_id = ?", req.SameTimeId).
  530. Update("same_time_status", pasturePb.SameTimeStatus_In_Progress).Error; err != nil {
  531. return xerr.WithStack(err)
  532. }
  533. if err = tx.Model(new(model.EventCowSameTime)).
  534. Where("cow_id = ?", req.CowId).
  535. Where("same_time_id = ?", req.SameTimeId).
  536. Where("same_time_type = ?", req.SameTimeType).
  537. Updates(map[string]interface{}{
  538. "status": pasturePb.IsShow_Ok,
  539. "drug_id": drugs.Id,
  540. "unit": drugs.Unit,
  541. "usage": req.Usage,
  542. "remarks": req.Remarks,
  543. "operation_id": operationUser.Id,
  544. "operation_name": operationUser.Name,
  545. }).Error; err != nil {
  546. return xerr.WithStack(err)
  547. }
  548. return nil
  549. }); err != nil {
  550. return xerr.WithStack(err)
  551. }
  552. return nil
  553. }
  554. func (s *StoreEntry) SameTimeList(
  555. ctx context.Context,
  556. req *pasturePb.SearchEventRequest,
  557. pagination *pasturePb.PaginationModel,
  558. ) (*pasturePb.SearchSameTimeResponse, error) {
  559. cowSameTimeList := make([]*model.EventCowSameTime, 0)
  560. var count int64 = 0
  561. pref := s.DB.Table(new(model.EventCowSameTime).TableName())
  562. if req.CowId != "" {
  563. cowIds := strings.Split(req.CowId, ",")
  564. pref.Where("cow_id IN ?", cowIds)
  565. }
  566. if err := pref.Count(&count).
  567. Limit(int(pagination.PageSize)).
  568. Offset(int(pagination.PageOffset)).
  569. Find(&cowSameTimeList).Error; err != nil {
  570. return nil, xerr.WithStack(err)
  571. }
  572. return &pasturePb.SearchSameTimeResponse{
  573. Code: http.StatusOK,
  574. Message: "ok",
  575. Data: &pasturePb.SameTimeData{
  576. List: model.EventCowSameTimeSlice(cowSameTimeList).ToPB(),
  577. Total: int32(count),
  578. PageSize: pagination.PageSize,
  579. Page: pagination.Page,
  580. },
  581. }, nil
  582. }
  583. func (s *StoreEntry) AbortionCreate(ctx context.Context, req *pasturePb.EventAbortionRequest) error {
  584. cow, err := s.GetCowInfoByCowId(ctx, int64(req.CowId))
  585. if err != nil {
  586. return xerr.WithStack(err)
  587. }
  588. systemUser, err := s.GetSystemUserById(ctx, int64(req.OperationId))
  589. if err != nil {
  590. return xerr.WithStack(err)
  591. }
  592. req.OperationName = systemUser.Name
  593. abortionReasonsMap := s.AbortionReasonsMap()
  594. newEventAbortion := model.NewEventAbortion(cow, req, abortionReasonsMap)
  595. lastCowMating := &model.EventMating{}
  596. if err = s.DB.Model(new(model.EventMating)).Where("cow_id = ?", cow.Id).
  597. Where("mating_result = ?", pasturePb.MatingResult_Pregnant).
  598. Order("id desc").First(lastCowMating).Error; err != nil {
  599. return xerr.WithStack(err)
  600. }
  601. if err = s.DB.Transaction(func(tx *gorm.DB) error {
  602. // 创建牛只流产事件数据
  603. if err = tx.Create(newEventAbortion).Error; err != nil {
  604. return xerr.WithStack(err)
  605. }
  606. // 更新牛只状态
  607. if err = tx.Model(new(model.Cow)).Where("id = ?", req.CowId).
  608. Updates(map[string]interface{}{
  609. "is_pregnant": pasturePb.IsShow_No,
  610. "last_abortion_at": req.AbortionAt,
  611. "breed_status": pasturePb.BreedStatus_Abort,
  612. }).Error; err != nil {
  613. return xerr.WithStack(err)
  614. }
  615. // 更新最近一次配种结果为流产
  616. if err = tx.Model(new(model.EventMating)).
  617. Where("id = ?", lastCowMating.Id).
  618. Updates(map[string]interface{}{
  619. "mating_result": pasturePb.MatingResult_Abort,
  620. "mating_result_at": req.AbortionAt,
  621. }).Error; err != nil {
  622. }
  623. return nil
  624. }); err != nil {
  625. return xerr.WithStack(err)
  626. }
  627. return nil
  628. }
  629. func (s *StoreEntry) AbortionCreateSlice(ctx context.Context, req *pasturePb.EventAbortionSlice) error {
  630. if len(req.Item) <= 0 {
  631. return xerr.WithStack(xerr.New("流产数据不能为空"))
  632. }
  633. errors := make(map[int32]error)
  634. for _, v := range req.Item {
  635. if err := s.AbortionCreate(ctx, v); err != nil {
  636. errors[v.CowId] = err
  637. }
  638. }
  639. if len(errors) > 0 {
  640. b, _ := json.Marshal(errors)
  641. return xerr.WithStack(xerr.Customf("部分流产数据创建失败: %s", string(b)))
  642. }
  643. return nil
  644. }
  645. func (s *StoreEntry) AbortionList(
  646. ctx context.Context,
  647. req *pasturePb.SearchEventRequest,
  648. pagination *pasturePb.PaginationModel,
  649. ) (*pasturePb.EventAbortionResponse, error) {
  650. abortionList := make([]*model.EventAbortion, 0)
  651. var count int64 = 0
  652. pref := s.DB.Model(new(model.EventAbortion))
  653. if len(req.CowId) > 0 {
  654. cowIds := strings.Split(req.CowId, ",")
  655. pref.Where("cow_id IN ?", cowIds)
  656. }
  657. if err := pref.Order("id desc").
  658. Count(&count).Limit(int(pagination.PageSize)).
  659. Offset(int(pagination.PageOffset)).
  660. Find(&abortionList).Error; err != nil {
  661. return nil, xerr.WithStack(err)
  662. }
  663. return &pasturePb.EventAbortionResponse{
  664. Code: http.StatusOK,
  665. Message: "ok",
  666. Data: &pasturePb.EventAbortionData{
  667. List: model.AbortionSlice(abortionList).ToPB(),
  668. Total: int32(count),
  669. PageSize: pagination.PageSize,
  670. Page: pagination.Page,
  671. },
  672. }, nil
  673. }