event_breed.go 22 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715
  1. package backend
  2. import (
  3. "context"
  4. "encoding/json"
  5. "fmt"
  6. "kpt-pasture/model"
  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.operation_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": int64(req.CalvingAt),
  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).
  101. Updates(map[string]interface{}{
  102. "calving_age": 0,
  103. "lact": cow.Lact + 1,
  104. "breed_status": pasturePb.BreedStatus_Calving,
  105. "is_pregnant": pasturePb.IsShow_No,
  106. "calving_at": int64(req.CalvingAt),
  107. "last_calving_at": time.Now().Unix(),
  108. }).Error; err != nil {
  109. return xerr.WithStack(err)
  110. }
  111. return nil
  112. }); err != nil {
  113. return xerr.WithStack(err)
  114. }
  115. return nil
  116. }
  117. func (s *StoreEntry) PregnantCheckList(ctx context.Context, req *pasturePb.SearchEventRequest, pagination *pasturePb.PaginationModel) (*pasturePb.PregnantCheckEventResponse, error) {
  118. pregnantCheckList := make([]*model.EventPregnantCheck, 0)
  119. var count int64 = 0
  120. pref := s.DB.Table(new(model.EventPregnantCheck).TableName())
  121. if len(req.CowId) > 0 {
  122. cowIds := strings.Split(req.CowId, ",")
  123. pref.Where("cow_id IN ?", cowIds)
  124. }
  125. if err := pref.Order("id desc").
  126. Count(&count).Limit(int(pagination.PageSize)).
  127. Offset(int(pagination.PageOffset)).
  128. Find(&pregnantCheckList).Error; err != nil {
  129. return nil, xerr.WithStack(err)
  130. }
  131. pregnantCheckResultMap := s.PregnantCheckResultMap()
  132. pregnantCheckMethodMap := s.PregnantCheckMethodMap()
  133. return &pasturePb.PregnantCheckEventResponse{
  134. Code: http.StatusOK,
  135. Message: "ok",
  136. Data: &pasturePb.SearchPregnantCheckData{
  137. List: model.EventPregnantCheckSlice(pregnantCheckList).ToPB(pregnantCheckResultMap, pregnantCheckMethodMap),
  138. Total: int32(count),
  139. PageSize: pagination.PageSize,
  140. Page: pagination.Page,
  141. },
  142. }, nil
  143. }
  144. func (s *StoreEntry) PregnantCheckCreate(ctx context.Context, req *pasturePb.EventPregnantCheck) error {
  145. if len(req.CowId) <= 0 {
  146. return xerr.Custom("请选择相关牛只")
  147. }
  148. cowList, err := s.ParseCowIds(ctx, req.CowId)
  149. if err != nil {
  150. return xerr.WithStack(err)
  151. }
  152. systemUser, _ := s.GetSystemUserById(ctx, int64(req.StaffMemberId))
  153. pregnantCheckDay := time.Unix(int64(req.PregnantCheckAt), 0).Format(model.LayoutTime)
  154. // 更新怀孕牛只
  155. updatePregnantCowIds := make([]int64, 0)
  156. // 更新流产牛只
  157. updateAbortCowIds := make([]int64, 0)
  158. // 更新空怀牛只
  159. updateEmptyCowIds := make([]int64, 0)
  160. cowIds := make([]int64, 0)
  161. for _, cow := range cowList {
  162. // 过滤掉没有配种状态的牛只
  163. if cow.BreedStatus != pasturePb.BreedStatus_Breeding {
  164. continue
  165. }
  166. itemEventPregnantCheck := &model.EventPregnantCheck{}
  167. if err = s.DB.Model(new(model.EventPregnantCheck)).
  168. Where("lact = ?", cow.Lact).
  169. Where("cow_id = ?", cow.Id).
  170. Order("id desc").First(itemEventPregnantCheck).Error; err != nil {
  171. return xerr.WithStack(err)
  172. }
  173. if req.PregnantCheckResult == pasturePb.PregnantCheckResult_Pregnant {
  174. updatePregnantCowIds = append(updatePregnantCowIds, cow.Id)
  175. }
  176. if req.PregnantCheckResult == pasturePb.PregnantCheckResult_UnPregnant {
  177. if itemEventPregnantCheck.PregnantCheckName == model.PregnantCheckForFirst {
  178. updateAbortCowIds = append(updateAbortCowIds, cow.Id)
  179. }
  180. if itemEventPregnantCheck.PregnantCheckName == model.PregnantCheckForSecond {
  181. updateEmptyCowIds = append(updateEmptyCowIds, cow.Id)
  182. }
  183. }
  184. cowIds = append(cowIds, cow.Id)
  185. }
  186. if err = s.DB.Transaction(func(tx *gorm.DB) error {
  187. // 更新孕检事件表
  188. if err = tx.Model(new(model.EventPregnantCheck)).
  189. Where("id IN ?", cowIds).
  190. Updates(map[string]interface{}{
  191. "reality_day": pregnantCheckDay,
  192. "pregnant_check_result": req.PregnantCheckResult,
  193. "pregnant_check_method": req.PregnantCheckMethod,
  194. "operation_id": systemUser.Id,
  195. "operation_name": systemUser.Name,
  196. "remarks": req.Remarks,
  197. "is_pregnant": req.PregnantCheckResult,
  198. }).Error; err != nil {
  199. return xerr.WithStack(err)
  200. }
  201. // 更新牛只信息和配种事件表怀孕状态
  202. if len(updatePregnantCowIds) > 0 {
  203. if err = tx.Model(&model.Cow{}).Where("cow_id IN ?", updatePregnantCowIds).
  204. Updates(map[string]interface{}{
  205. "breed_status": pasturePb.BreedStatus_Pregnant,
  206. "last_pregnant_check_at": pregnantCheckDay,
  207. "is_pregnant": pasturePb.IsShow_Ok,
  208. }).Error; err != nil {
  209. return xerr.WithStack(err)
  210. }
  211. if err = tx.Model(&model.EventMating{}).Where("cow_id IN ?", updatePregnantCowIds).
  212. Group("cow_id").Select("MAX(id) as id").
  213. Updates(map[string]interface{}{
  214. "mating_result": pasturePb.MatingResult_Pregnant,
  215. "mating_result_at": req.PregnantCheckAt,
  216. }).Error; err != nil {
  217. return xerr.WithStack(err)
  218. }
  219. }
  220. // 更新牛只信息和配种事件表流产状态
  221. if len(updateAbortCowIds) > 0 {
  222. if err = tx.Model(&model.Cow{}).Where("cow_id IN ?", updateAbortCowIds).
  223. Updates(map[string]interface{}{
  224. "breed_status": pasturePb.BreedStatus_Abort,
  225. "last_pregnant_check_at": pregnantCheckDay,
  226. "is_pregnant": pasturePb.IsShow_Ok,
  227. }).Error; err != nil {
  228. return xerr.WithStack(err)
  229. }
  230. if err = tx.Model(&model.EventMating{}).Where("cow_id IN ?", updateAbortCowIds).
  231. Group("cow_id").Select("MAX(id) as id").
  232. Updates(map[string]interface{}{
  233. "mating_result": pasturePb.MatingResult_Abort,
  234. "mating_result_at": req.PregnantCheckAt,
  235. }).Error; err != nil {
  236. return xerr.WithStack(err)
  237. }
  238. }
  239. // 更新牛只信息和配种事件表空怀状态
  240. if len(updateEmptyCowIds) > 0 {
  241. if err = tx.Model(&model.Cow{}).Where("cow_id IN ?", updateEmptyCowIds).
  242. Updates(map[string]interface{}{
  243. "breed_status": pasturePb.BreedStatus_Empty,
  244. "last_pregnant_check_at": pregnantCheckDay,
  245. "is_pregnant": pasturePb.IsShow_Ok,
  246. }).Error; err != nil {
  247. return xerr.WithStack(err)
  248. }
  249. if err = tx.Model(&model.EventMating{}).Where("cow_id IN ?", updateEmptyCowIds).
  250. Group("cow_id").Select("MAX(id) as id").
  251. Updates(map[string]interface{}{
  252. "mating_result": pasturePb.MatingResult_Empty,
  253. "mating_result_at": req.PregnantCheckAt,
  254. }).Error; err != nil {
  255. return xerr.WithStack(err)
  256. }
  257. }
  258. return nil
  259. }); err != nil {
  260. return xerr.WithStack(err)
  261. }
  262. return nil
  263. }
  264. func (s *StoreEntry) MatingList(ctx context.Context, req *pasturePb.SearchEventRequest, pagination *pasturePb.PaginationModel) (*pasturePb.MatingEventResponse, error) {
  265. matingList := make([]*model.EventMating, 0)
  266. var count int64 = 0
  267. pref := s.DB.Model(new(model.EventMating)).
  268. Where("mating_result > ?", pasturePb.MatingResult_Invalid).
  269. Where("status = ?", pasturePb.IsShow_No)
  270. if len(req.CowId) > 0 {
  271. cowIds := strings.Split(req.CowId, ",")
  272. pref.Where("cow_id IN ?", cowIds)
  273. }
  274. if err := pref.Order("id desc").
  275. Count(&count).Limit(int(pagination.PageSize)).
  276. Offset(int(pagination.PageOffset)).
  277. Find(&matingList).Error; err != nil {
  278. return nil, xerr.WithStack(err)
  279. }
  280. exposeEstrusTypeMap := s.ExposeEstrusTypeMap()
  281. return &pasturePb.MatingEventResponse{
  282. Code: http.StatusOK,
  283. Message: "ok",
  284. Data: &pasturePb.SearchMatingData{
  285. List: model.EventMatingSlice(matingList).ToPB(exposeEstrusTypeMap),
  286. Total: int32(count),
  287. PageSize: pagination.PageSize,
  288. Page: pagination.Page,
  289. },
  290. }, nil
  291. }
  292. func (s *StoreEntry) MatingCreate(ctx context.Context, req *pasturePb.EventMating) error {
  293. if len(req.CowId) <= 0 {
  294. return xerr.Custom("请选择相关牛只")
  295. }
  296. cowList, err := s.ParseCowIds(ctx, req.CowId)
  297. if err != nil {
  298. return xerr.WithStack(err)
  299. }
  300. systemUser, _ := s.GetSystemUserById(ctx, int64(req.StaffMemberId))
  301. eventFrozenSemenLogList := make([]*model.FrozenSemenLog, 0)
  302. frozenSemen := &model.FrozenSemen{}
  303. if err = s.DB.Where("bull_id = ?", req.FrozenSemenNumber).First(frozenSemen).Error; err != nil {
  304. return xerr.WithStack(err)
  305. }
  306. matingUpdateIds := make([]int64, 0)
  307. cowIds := make([]int64, 0)
  308. nowTime := time.Now()
  309. for _, cow := range cowList {
  310. var count int64 = 0
  311. itemEventMating := &model.EventMating{}
  312. if err = s.DB.Where("lact = ?", cow.Lact).
  313. Where("cow_id = ?", cow.Id).
  314. Order("id desc").
  315. First(itemEventMating).Count(&count).Error; err != nil {
  316. return xerr.WithStack(err)
  317. }
  318. // 判断当前输精时间距离上次输精时间是否超过2天,如果超过则更新为复配状态
  319. itemBeforeTwoDays := nowTime.Sub(time.Unix(itemEventMating.RealityDay, 0)).Hours()
  320. if count > 0 && itemBeforeTwoDays > 48 {
  321. matingUpdateIds = append(matingUpdateIds, itemEventMating.Id)
  322. }
  323. eventFrozenSemenLogList = append(
  324. eventFrozenSemenLogList,
  325. model.NewEventFrozenSemenLog(req.FrozenSemenNumber, cow, int64(req.StaffMemberId)),
  326. )
  327. cowIds = append(cowIds, cow.Id)
  328. }
  329. if err = s.DB.Transaction(func(tx *gorm.DB) error {
  330. if len(cowIds) > 0 {
  331. // 如果有同期牛只,则修改为已结束状态
  332. if err = tx.Table(new(model.CowSameTime).TableName()).
  333. Where("cow_id IN ?", cowIds).
  334. UpdateColumn("status", pasturePb.SameTimeStatus_End).Error; err != nil {
  335. }
  336. // 创建配种事件数据
  337. if err = tx.Model(new(model.EventMating)).
  338. Where("cow_id IN ?", cowIds).
  339. Updates(map[string]interface{}{
  340. "mating_result": pasturePb.MatingResult_Unknown,
  341. "status": pasturePb.IsShow_Ok,
  342. "reality_day": time.Unix(int64(req.MatingAt), 0).Format(model.LayoutTime),
  343. "mating_number": 1,
  344. "frozen_semen_number": req.FrozenSemenCount,
  345. "operation_id": req.StaffMemberId,
  346. "operation_name": systemUser.Name,
  347. }).
  348. Error; err != nil {
  349. return xerr.WithStack(err)
  350. }
  351. }
  352. // 更新已配种的牛只为复配状态
  353. if len(matingUpdateIds) > 0 {
  354. if err = tx.Model(new(model.EventMating)).
  355. Where("id IN ?", matingUpdateIds).
  356. Where("mating_result = ?", pasturePb.MatingResult_Unknown).
  357. Updates(map[string]interface{}{
  358. "mating_result": pasturePb.MatingResult_ReMatch,
  359. "mating_number": 2,
  360. "frozen_semen_number": req.FrozenSemenCount,
  361. }).
  362. Error; err != nil {
  363. return xerr.WithStack(err)
  364. }
  365. }
  366. // 创建冻精使用记录日志
  367. if err = tx.Create(eventFrozenSemenLogList).Error; err != nil {
  368. return xerr.WithStack(err)
  369. }
  370. // 减去精液的数量
  371. if err = tx.Model(new(model.FrozenSemen)).
  372. Where("id = ?", frozenSemen.Id).
  373. Where("quantity > 0").
  374. UpdateColumn("quantity",
  375. gorm.Expr("quantity - ?", len(matingUpdateIds)),
  376. ).Error; err != nil {
  377. return xerr.WithStack(err)
  378. }
  379. if err = tx.Table(new(model.Cow).TableName()).
  380. Where("id IN ?", cowIds).
  381. Updates(map[string]interface{}{
  382. "breed_status": pasturePb.BreedStatus_Breeding,
  383. "last_bull_number": req.FrozenSemenNumber,
  384. "last_mating_at": int64(req.MatingAt),
  385. }).Error; err != nil {
  386. return xerr.WithStack(err)
  387. }
  388. return nil
  389. }); err != nil {
  390. return xerr.WithStack(err)
  391. }
  392. return nil
  393. }
  394. func (s *StoreEntry) EstrusList(ctx context.Context, req *pasturePb.SearchEventRequest, pagination *pasturePb.PaginationModel) (*pasturePb.EstrusEventResponse, error) {
  395. estrusList := make([]*model.EventEstrus, 0)
  396. var count int64 = 0
  397. pref := s.DB.Table(new(model.EventEstrus).TableName())
  398. if len(req.CowId) > 0 {
  399. cowIds := strings.Split(req.CowId, ",")
  400. pref.Where("cow_id IN ?", cowIds)
  401. }
  402. if err := pref.Order("id desc").
  403. Count(&count).Limit(int(pagination.PageSize)).
  404. Offset(int(pagination.PageOffset)).
  405. Find(&estrusList).Error; err != nil {
  406. return nil, xerr.WithStack(err)
  407. }
  408. systemUserList, _ := s.SystemUserList(ctx)
  409. return &pasturePb.EstrusEventResponse{
  410. Code: http.StatusOK,
  411. Message: "ok",
  412. Data: &pasturePb.SearchEstrusData{
  413. List: model.EstrusSlice(estrusList).ToPB(systemUserList),
  414. Total: int32(count),
  415. PageSize: pagination.PageSize,
  416. Page: pagination.Page,
  417. },
  418. }, nil
  419. }
  420. func (s *StoreEntry) EstrusCreate(ctx context.Context, req *pasturePb.EventEstrus) error {
  421. if len(req.CowId) <= 0 {
  422. return xerr.Custom("请选择相关牛只")
  423. }
  424. cowList, err := s.ParseCowIds(ctx, req.CowId)
  425. if err != nil {
  426. return xerr.WithStack(err)
  427. }
  428. estrusList := make([]*model.EventEstrus, 0)
  429. currentUser, _ := s.GetCurrentSystemUser(ctx)
  430. for _, cow := range cowList {
  431. estrusList = append(estrusList, model.NewEventEstrus(cow, currentUser, req))
  432. }
  433. if err := s.DB.Create(estrusList).Error; err != nil {
  434. return xerr.WithStack(err)
  435. }
  436. return nil
  437. }
  438. func (s *StoreEntry) FrozenSemenList(ctx context.Context, req *pasturePb.FrozenSemenRequest, pagination *pasturePb.PaginationModel) (*pasturePb.FrozenSemenResponse, error) {
  439. frozenSemenList := make([]*model.FrozenSemen, 0)
  440. var count int64 = 0
  441. pref := s.DB.Table(new(model.FrozenSemen).TableName())
  442. if req.BullId != "" {
  443. pref.Where("bull_id = ?", req.BullId)
  444. }
  445. if req.Producer != "" {
  446. pref.Where("producer = ?", req.Producer)
  447. }
  448. if err := pref.Order("id desc").
  449. Count(&count).Limit(int(pagination.PageSize)).
  450. Offset(int(pagination.PageOffset)).
  451. Find(&frozenSemenList).Error; err != nil {
  452. return nil, xerr.WithStack(err)
  453. }
  454. frozenSemenTypeMap := s.FrozenSemenTypeMap()
  455. unitMap := s.UnitMap()
  456. return &pasturePb.FrozenSemenResponse{
  457. Code: http.StatusOK,
  458. Message: "ok",
  459. Data: &pasturePb.SearchFrozenSemenData{
  460. List: model.FrozenSemenSlice(frozenSemenList).ToPB(frozenSemenTypeMap, unitMap),
  461. Total: int32(count),
  462. PageSize: pagination.PageSize,
  463. Page: pagination.Page,
  464. },
  465. }, nil
  466. }
  467. func (s *StoreEntry) FrozenSemenCreate(ctx context.Context, req *pasturePb.SearchFrozenSemenList) error {
  468. currentUser, _ := s.GetCurrentSystemUser(ctx)
  469. req.CowKindName = s.CowKindMap()[req.CowKind]
  470. newFrozenSemen := model.NewFrozenSemen(req, currentUser)
  471. if err := s.DB.Create(newFrozenSemen).Error; err != nil {
  472. return xerr.WithStack(err)
  473. }
  474. return nil
  475. }
  476. func (s *StoreEntry) SameTimeCreate(ctx context.Context, req *pasturePb.EventSameTime) error {
  477. if _, err := s.GetSameTimeById(ctx, int64(req.SameTimeId)); err != nil {
  478. return xerr.WithStack(err)
  479. }
  480. cowList, err := s.ParseCowIds(ctx, req.CowId)
  481. if err != nil {
  482. return xerr.WithStack(err)
  483. }
  484. cowIds := make([]int64, 0)
  485. for _, v := range cowList {
  486. cowIds = append(cowIds, v.Id)
  487. }
  488. drugs := &model.Drugs{}
  489. if req.DrugsId > 0 {
  490. if drugs, err = s.GetDrugsById(ctx, int64(req.DrugsId)); err != nil {
  491. return xerr.WithStack(err)
  492. }
  493. }
  494. systemUser, _ := s.GetSystemUserById(ctx, int64(req.StaffMemberId))
  495. if err = s.DB.Transaction(func(tx *gorm.DB) error {
  496. if err = tx.Model(new(model.CowSameTime)).
  497. Where("cow_id IN ?", cowIds).
  498. Where("same_time_id = ?", req.SameTimeId).
  499. Update("same_time_status", pasturePb.SameTimeStatus_In_Progress).Error; err != nil {
  500. return xerr.WithStack(err)
  501. }
  502. if err = tx.Model(new(model.EventCowSameTime)).
  503. Where("cow_id IN ?", cowIds).
  504. Where("same_time_id = ?", req.SameTimeId).
  505. Where("same_time_type = ?", req.SameTimeType).
  506. Updates(map[string]interface{}{
  507. "status": pasturePb.IsShow_Ok,
  508. "drug_id": drugs.Id,
  509. "unit": drugs.Unit,
  510. "usage": req.Usage,
  511. "remarks": req.Remarks,
  512. "operation_id": req.StaffMemberId,
  513. "operation_name": systemUser.Name,
  514. }).Error; err != nil {
  515. return xerr.WithStack(err)
  516. }
  517. return nil
  518. }); err != nil {
  519. return xerr.WithStack(err)
  520. }
  521. return nil
  522. }
  523. func (s *StoreEntry) SameTimeList(
  524. ctx context.Context,
  525. req *pasturePb.SearchEventRequest,
  526. pagination *pasturePb.PaginationModel,
  527. ) (*pasturePb.SearchSameTimeResponse, error) {
  528. cowSameTimeList := make([]*model.EventCowSameTime, 0)
  529. var count int64 = 0
  530. pref := s.DB.Table(new(model.EventCowSameTime).TableName())
  531. if req.CowId != "" {
  532. cowIds := strings.Split(req.CowId, ",")
  533. pref.Where("cow_id IN ?", cowIds)
  534. }
  535. if err := pref.Count(&count).
  536. Limit(int(pagination.PageSize)).
  537. Offset(int(pagination.PageOffset)).
  538. Find(&cowSameTimeList).Error; err != nil {
  539. return nil, xerr.WithStack(err)
  540. }
  541. return &pasturePb.SearchSameTimeResponse{
  542. Code: http.StatusOK,
  543. Message: "ok",
  544. Data: &pasturePb.SameTimeData{
  545. List: model.EventCowSameTimeSlice(cowSameTimeList).ToPB(),
  546. Total: int32(count),
  547. PageSize: pagination.PageSize,
  548. Page: pagination.Page,
  549. },
  550. }, nil
  551. }
  552. func (s *StoreEntry) AbortionCreate(ctx context.Context, req *pasturePb.EventAbortionRequest) error {
  553. cow, err := s.GetCowInfoByCowId(ctx, int64(req.CowId))
  554. if err != nil {
  555. return xerr.WithStack(err)
  556. }
  557. systemUser, err := s.GetSystemUserById(ctx, int64(req.OperationId))
  558. if err != nil {
  559. return xerr.WithStack(err)
  560. }
  561. req.OperationName = systemUser.Name
  562. abortionReasonsMap := s.AbortionReasonsMap()
  563. newEventAbortion := model.NewEventAbortion(cow, req, abortionReasonsMap)
  564. lastCowMating := &model.EventMating{}
  565. if err = s.DB.Where("cow_id = ?", cow.Id).Order("id desc").First(lastCowMating).Error; err != nil {
  566. return xerr.WithStack(err)
  567. }
  568. if err = s.DB.Transaction(func(tx *gorm.DB) error {
  569. // 创建牛只流产事件数据
  570. if err = tx.Create(newEventAbortion).Error; err != nil {
  571. return xerr.WithStack(err)
  572. }
  573. // 更新牛只状态
  574. if err = tx.Model(new(model.Cow)).Where("id = ?", req.CowId).
  575. Updates(map[string]interface{}{
  576. "is_pregnant": pasturePb.IsShow_No,
  577. "last_abortion_at": req.AbortionAt,
  578. "breed_status": pasturePb.BreedStatus_Abort,
  579. }).Error; err != nil {
  580. return xerr.WithStack(err)
  581. }
  582. // 更新最近一次配种结果为流产
  583. if err = tx.Model(new(model.EventMating)).Where("id = ?", lastCowMating.Id).
  584. Updates(map[string]interface{}{
  585. "mating_result": pasturePb.MatingResult_Abort,
  586. "mating_result_at": req.AbortionAt,
  587. }).Error; err != nil {
  588. }
  589. return nil
  590. }); err != nil {
  591. return xerr.WithStack(err)
  592. }
  593. return nil
  594. }
  595. func (s *StoreEntry) AbortionCreateSlice(ctx context.Context, req *pasturePb.EventAbortionSlice) error {
  596. if len(req.Item) <= 0 {
  597. return xerr.WithStack(xerr.New("流产数据不能为空"))
  598. }
  599. errors := make(map[int32]error)
  600. for _, v := range req.Item {
  601. if err := s.AbortionCreate(ctx, v); err != nil {
  602. errors[v.CowId] = err
  603. }
  604. }
  605. if len(errors) > 0 {
  606. b, _ := json.Marshal(errors)
  607. return xerr.WithStack(xerr.Customf("部分流产数据创建失败: %s", string(b)))
  608. }
  609. return nil
  610. }
  611. func (s *StoreEntry) AbortionList(
  612. ctx context.Context,
  613. req *pasturePb.SearchEventRequest,
  614. pagination *pasturePb.PaginationModel,
  615. ) (*pasturePb.EventAbortionResponse, error) {
  616. abortionList := make([]*model.EventAbortion, 0)
  617. var count int64 = 0
  618. pref := s.DB.Model(new(model.EventAbortion))
  619. if len(req.CowId) > 0 {
  620. cowIds := strings.Split(req.CowId, ",")
  621. pref.Where("cow_id IN ?", cowIds)
  622. }
  623. if err := pref.Order("id desc").
  624. Count(&count).Limit(int(pagination.PageSize)).
  625. Offset(int(pagination.PageOffset)).
  626. Find(&abortionList).Error; err != nil {
  627. return nil, xerr.WithStack(err)
  628. }
  629. return &pasturePb.EventAbortionResponse{
  630. Code: http.StatusOK,
  631. Message: "ok",
  632. Data: &pasturePb.EventAbortionData{
  633. List: model.AbortionSlice(abortionList).ToPB(),
  634. Total: int32(count),
  635. PageSize: pagination.PageSize,
  636. Page: pagination.Page,
  637. },
  638. }, nil
  639. }