event_breed.go 22 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703
  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.OperationId))
  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. currentUser, err := s.GetCurrentSystemUser(ctx)
  301. if err != nil {
  302. return xerr.Customf("获取当前用户失败: %s", err.Error())
  303. }
  304. operationUser, err := s.GetSystemUserById(ctx, int64(req.OperationId))
  305. if err != nil {
  306. return xerr.WithStack(err)
  307. }
  308. frozenSemen := &model.FrozenSemen{}
  309. if err = s.DB.Where("bull_id = ?", req.FrozenSemenNumber).First(frozenSemen).Error; err != nil {
  310. return xerr.WithStack(err)
  311. }
  312. matingReMatchIds := make([]int64, 0)
  313. newMatingList := make([]*model.EventMating, 0)
  314. updateMatingList := make([]int64, 0)
  315. eventFrozenSemenLogList := make([]*model.FrozenSemenLog, 0)
  316. cowIds := make([]int64, 0)
  317. for _, cow := range cowList {
  318. if cow.Sex != pasturePb.Genders_Female {
  319. return xerr.Customf("牛只 %d 不是母牛", cow.Id)
  320. }
  321. // 获取当前牛只最后一次配种时间,判断是否为复配牛只
  322. if cow.LastMatingAt <= 0 {
  323. // 第一次配种,创建配种信息(自然发情牛只,未经过同期的牛只)
  324. newMatingList = append(newMatingList, model.NewEventMating2(cow, req))
  325. } else {
  326. // 如何两次配种时间为连续两天,则更新为复配状态
  327. lastMatingAt := time.Unix(cow.LastMatingAt, 0).AddDate(0, 0, 1).Format(model.LayoutDate2)
  328. currentMatingAt := time.Unix(int64(req.MatingAt), 0).Format(model.LayoutDate2)
  329. if cow.LastMatingAt != int64(req.MatingAt) && lastMatingAt == currentMatingAt {
  330. matingReMatchIds = append(matingReMatchIds, cow.Id)
  331. } else {
  332. updateMatingList = append(updateMatingList, cow.Id)
  333. }
  334. }
  335. newFrozenSemenItem := &model.FrozenSemenItem{
  336. CowId: cow.Id,
  337. Quantity: req.FrozenSemenCount,
  338. MatingAt: int64(req.MatingAt),
  339. OperationId: operationUser.Id,
  340. OperationName: operationUser.Name,
  341. BullId: req.FrozenSemenNumber,
  342. Remarks: req.Remarks,
  343. }
  344. itemFrozenSemenLog := model.NewEventFrozenSemenLog(newFrozenSemenItem)
  345. eventFrozenSemenLogList = append(eventFrozenSemenLogList, itemFrozenSemenLog)
  346. cowIds = append(cowIds, cow.Id)
  347. }
  348. if err = s.DB.Transaction(func(tx *gorm.DB) error {
  349. if len(updateMatingList) > 0 {
  350. // 创建配种事件数据
  351. if err = tx.Model(new(model.EventMating)).
  352. Where("cow_id IN ?", updateMatingList).
  353. Where("status = ?", pasturePb.IsShow_No).
  354. Updates(map[string]interface{}{
  355. "mating_result": pasturePb.MatingResult_Unknown,
  356. "status": pasturePb.IsShow_Ok,
  357. "reality_day": time.Unix(int64(req.MatingAt), 0).Format(model.LayoutTime),
  358. "frozen_semen_count": 1,
  359. "frozen_semen_number": req.FrozenSemenNumber,
  360. "operation_id": operationUser.Id,
  361. "operation_name": operationUser.Name,
  362. "message_id": currentUser.Id,
  363. "message_name": currentUser.Name,
  364. }).Error; err != nil {
  365. return xerr.WithStack(err)
  366. }
  367. }
  368. // 更新已配种的牛只为复配状态
  369. if len(matingReMatchIds) > 0 {
  370. if err = tx.Model(new(model.EventMating)).
  371. Where("id IN ?", matingReMatchIds).
  372. Where("mating_result = ?", pasturePb.MatingResult_Unknown).
  373. Where("status = ?", pasturePb.IsShow_Ok).
  374. Update("mating_result", pasturePb.MatingResult_ReMatch).
  375. Error; err != nil {
  376. return xerr.WithStack(err)
  377. }
  378. }
  379. // 创建配种事件数据
  380. if len(newMatingList) > 0 {
  381. if err = tx.Create(newMatingList).Error; err != nil {
  382. return xerr.WithStack(err)
  383. }
  384. }
  385. // 创建冻精使用记录日志
  386. if err = tx.Create(eventFrozenSemenLogList).Error; err != nil {
  387. return xerr.WithStack(err)
  388. }
  389. // 更新牛只的繁殖状态为配种
  390. if len(cowIds) > 0 {
  391. // 如果有同期牛只,则修改为已结束状态
  392. if err = tx.Table(new(model.CowSameTime).TableName()).
  393. Where("cow_id IN ?", cowIds).
  394. UpdateColumn("status", pasturePb.SameTimeStatus_End).Error; err != nil {
  395. }
  396. if err = tx.Model(new(model.Cow)).
  397. Where("id IN ?", cowIds).
  398. Where("admission_status = ?", pasturePb.AdmissionStatus_Admission).
  399. Updates(map[string]interface{}{
  400. "breed_status": pasturePb.BreedStatus_Breeding,
  401. "last_bull_number": req.FrozenSemenNumber,
  402. "last_mating_at": int64(req.MatingAt),
  403. }).Error; err != nil {
  404. return xerr.WithStack(err)
  405. }
  406. // 减去精液的数量
  407. if err = tx.Model(new(model.FrozenSemen)).
  408. Where("id = ?", frozenSemen.Id).
  409. Where("quantity > 0").
  410. UpdateColumn("quantity",
  411. gorm.Expr("quantity - ?", req.FrozenSemenCount),
  412. ).Error; err != nil {
  413. return xerr.WithStack(err)
  414. }
  415. // todo 更新每头牛只的胎次配次
  416. }
  417. return nil
  418. }); err != nil {
  419. return xerr.WithStack(err)
  420. }
  421. return nil
  422. }
  423. func (s *StoreEntry) EstrusList(ctx context.Context, req *pasturePb.SearchEventRequest, pagination *pasturePb.PaginationModel) (*pasturePb.EstrusEventResponse, error) {
  424. estrusList := make([]*model.EventEstrus, 0)
  425. var count int64 = 0
  426. pref := s.DB.Table(new(model.EventEstrus).TableName())
  427. if len(req.CowId) > 0 {
  428. cowIds := strings.Split(req.CowId, ",")
  429. pref.Where("cow_id IN ?", cowIds)
  430. }
  431. if err := pref.Order("id desc").
  432. Count(&count).Limit(int(pagination.PageSize)).
  433. Offset(int(pagination.PageOffset)).
  434. Find(&estrusList).Error; err != nil {
  435. return nil, xerr.WithStack(err)
  436. }
  437. return &pasturePb.EstrusEventResponse{
  438. Code: http.StatusOK,
  439. Message: "ok",
  440. Data: &pasturePb.SearchEstrusData{
  441. List: model.EstrusSlice(estrusList).ToPB(),
  442. Total: int32(count),
  443. PageSize: pagination.PageSize,
  444. Page: pagination.Page,
  445. },
  446. }, nil
  447. }
  448. func (s *StoreEntry) EstrusCreate(ctx context.Context, req *pasturePb.EventEstrus) error {
  449. if len(req.CowId) <= 0 {
  450. return xerr.Custom("请选择相关牛只")
  451. }
  452. cowList, err := s.ParseCowIds(ctx, req.CowId)
  453. if err != nil {
  454. return xerr.WithStack(err)
  455. }
  456. estrusList := make([]*model.EventEstrus, 0)
  457. currentUser, err := s.GetCurrentSystemUser(ctx)
  458. if err != nil {
  459. return xerr.Custom("获取当前登录用户失败")
  460. }
  461. for _, cow := range cowList {
  462. operationUser, err := s.GetSystemUserById(ctx, int64(req.OperationId))
  463. if err != nil {
  464. return xerr.WithStack(err)
  465. }
  466. req.OperationName = operationUser.Name
  467. estrusList = append(estrusList, model.NewEventEstrus(cow, currentUser, req))
  468. }
  469. if err = s.DB.Create(estrusList).Error; err != nil {
  470. return xerr.WithStack(err)
  471. }
  472. return nil
  473. }
  474. func (s *StoreEntry) SameTimeCreate(ctx context.Context, req *pasturePb.EventSameTime) error {
  475. if _, err := s.GetSameTimeById(ctx, int64(req.SameTimeId)); err != nil {
  476. return xerr.WithStack(err)
  477. }
  478. drugs := &model.Drugs{}
  479. var err error
  480. if req.DrugsId > 0 {
  481. if drugs, err = s.GetDrugsById(ctx, int64(req.DrugsId)); err != nil {
  482. return xerr.WithStack(err)
  483. }
  484. }
  485. operationUser, err := s.GetSystemUserById(ctx, int64(req.OperationId))
  486. if err != nil {
  487. return xerr.WithStack(err)
  488. }
  489. if err = s.DB.Transaction(func(tx *gorm.DB) error {
  490. if err = tx.Model(new(model.CowSameTime)).
  491. Where("cow_id = ?", req.CowId).
  492. Where("same_time_id = ?", req.SameTimeId).
  493. Update("same_time_status", pasturePb.SameTimeStatus_In_Progress).Error; err != nil {
  494. return xerr.WithStack(err)
  495. }
  496. if err = tx.Model(new(model.EventCowSameTime)).
  497. Where("cow_id = ?", req.CowId).
  498. Where("same_time_id = ?", req.SameTimeId).
  499. Where("same_time_type = ?", req.SameTimeType).
  500. Updates(map[string]interface{}{
  501. "status": pasturePb.IsShow_Ok,
  502. "drug_id": drugs.Id,
  503. "unit": drugs.Unit,
  504. "usage": req.Usage,
  505. "remarks": req.Remarks,
  506. "operation_id": operationUser.Id,
  507. "operation_name": operationUser.Name,
  508. }).Error; err != nil {
  509. return xerr.WithStack(err)
  510. }
  511. return nil
  512. }); err != nil {
  513. return xerr.WithStack(err)
  514. }
  515. return nil
  516. }
  517. func (s *StoreEntry) SameTimeList(
  518. ctx context.Context,
  519. req *pasturePb.SearchEventRequest,
  520. pagination *pasturePb.PaginationModel,
  521. ) (*pasturePb.SearchSameTimeResponse, error) {
  522. cowSameTimeList := make([]*model.EventCowSameTime, 0)
  523. var count int64 = 0
  524. pref := s.DB.Table(new(model.EventCowSameTime).TableName())
  525. if req.CowId != "" {
  526. cowIds := strings.Split(req.CowId, ",")
  527. pref.Where("cow_id IN ?", cowIds)
  528. }
  529. if err := pref.Count(&count).
  530. Limit(int(pagination.PageSize)).
  531. Offset(int(pagination.PageOffset)).
  532. Find(&cowSameTimeList).Error; err != nil {
  533. return nil, xerr.WithStack(err)
  534. }
  535. return &pasturePb.SearchSameTimeResponse{
  536. Code: http.StatusOK,
  537. Message: "ok",
  538. Data: &pasturePb.SameTimeData{
  539. List: model.EventCowSameTimeSlice(cowSameTimeList).ToPB(),
  540. Total: int32(count),
  541. PageSize: pagination.PageSize,
  542. Page: pagination.Page,
  543. },
  544. }, nil
  545. }
  546. func (s *StoreEntry) AbortionCreate(ctx context.Context, req *pasturePb.EventAbortionRequest) error {
  547. cow, err := s.GetCowInfoByCowId(ctx, int64(req.CowId))
  548. if err != nil {
  549. return xerr.WithStack(err)
  550. }
  551. systemUser, err := s.GetSystemUserById(ctx, int64(req.OperationId))
  552. if err != nil {
  553. return xerr.WithStack(err)
  554. }
  555. req.OperationName = systemUser.Name
  556. abortionReasonsMap := s.AbortionReasonsMap()
  557. newEventAbortion := model.NewEventAbortion(cow, req, abortionReasonsMap)
  558. lastCowMating := &model.EventMating{}
  559. if err = s.DB.Where("cow_id = ?", cow.Id).Order("id desc").First(lastCowMating).Error; err != nil {
  560. return xerr.WithStack(err)
  561. }
  562. if err = s.DB.Transaction(func(tx *gorm.DB) error {
  563. // 创建牛只流产事件数据
  564. if err = tx.Create(newEventAbortion).Error; err != nil {
  565. return xerr.WithStack(err)
  566. }
  567. // 更新牛只状态
  568. if err = tx.Model(new(model.Cow)).Where("id = ?", req.CowId).
  569. Updates(map[string]interface{}{
  570. "is_pregnant": pasturePb.IsShow_No,
  571. "last_abortion_at": req.AbortionAt,
  572. "breed_status": pasturePb.BreedStatus_Abort,
  573. "lact": cow.Lact + 1,
  574. }).Error; err != nil {
  575. return xerr.WithStack(err)
  576. }
  577. // 更新最近一次配种结果为流产
  578. if err = tx.Model(new(model.EventMating)).Where("id = ?", lastCowMating.Id).
  579. Updates(map[string]interface{}{
  580. "mating_result": pasturePb.MatingResult_Abort,
  581. "mating_result_at": req.AbortionAt,
  582. }).Error; err != nil {
  583. }
  584. return nil
  585. }); err != nil {
  586. return xerr.WithStack(err)
  587. }
  588. return nil
  589. }
  590. func (s *StoreEntry) AbortionCreateSlice(ctx context.Context, req *pasturePb.EventAbortionSlice) error {
  591. if len(req.Item) <= 0 {
  592. return xerr.WithStack(xerr.New("流产数据不能为空"))
  593. }
  594. errors := make(map[int32]error)
  595. for _, v := range req.Item {
  596. if err := s.AbortionCreate(ctx, v); err != nil {
  597. errors[v.CowId] = err
  598. }
  599. }
  600. if len(errors) > 0 {
  601. b, _ := json.Marshal(errors)
  602. return xerr.WithStack(xerr.Customf("部分流产数据创建失败: %s", string(b)))
  603. }
  604. return nil
  605. }
  606. func (s *StoreEntry) AbortionList(
  607. ctx context.Context,
  608. req *pasturePb.SearchEventRequest,
  609. pagination *pasturePb.PaginationModel,
  610. ) (*pasturePb.EventAbortionResponse, error) {
  611. abortionList := make([]*model.EventAbortion, 0)
  612. var count int64 = 0
  613. pref := s.DB.Model(new(model.EventAbortion))
  614. if len(req.CowId) > 0 {
  615. cowIds := strings.Split(req.CowId, ",")
  616. pref.Where("cow_id IN ?", cowIds)
  617. }
  618. if err := pref.Order("id desc").
  619. Count(&count).Limit(int(pagination.PageSize)).
  620. Offset(int(pagination.PageOffset)).
  621. Find(&abortionList).Error; err != nil {
  622. return nil, xerr.WithStack(err)
  623. }
  624. return &pasturePb.EventAbortionResponse{
  625. Code: http.StatusOK,
  626. Message: "ok",
  627. Data: &pasturePb.EventAbortionData{
  628. List: model.AbortionSlice(abortionList).ToPB(),
  629. Total: int32(count),
  630. PageSize: pagination.PageSize,
  631. Page: pagination.Page,
  632. },
  633. }, nil
  634. }