event_breed.go 20 KB

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