event_breed.go 22 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654
  1. package backend
  2. import (
  3. "context"
  4. "errors"
  5. "fmt"
  6. "kpt-pasture/model"
  7. "kpt-pasture/util"
  8. "net/http"
  9. "strings"
  10. "gitee.com/xuyiping_admin/pkg/logger/zaplog"
  11. "go.uber.org/zap"
  12. pasturePb "gitee.com/xuyiping_admin/go_proto/proto/go/backend/cow"
  13. "gitee.com/xuyiping_admin/pkg/xerr"
  14. "gorm.io/gorm"
  15. )
  16. func (s *StoreEntry) CalvingList(ctx context.Context, req *pasturePb.SearchEventRequest, pagination *pasturePb.PaginationModel) (*pasturePb.SearchLavingEventResponse, error) {
  17. userModel, err := s.GetUserModel(ctx)
  18. if err != nil {
  19. return nil, xerr.WithStack(err)
  20. }
  21. lavingList := make([]*model.EventCalvingList, 0)
  22. var count int64 = 0
  23. pref := s.DB.Table(fmt.Sprintf("%s as a", new(model.EventCalving).TableName())).
  24. Select(`a.*,b.name as pen_name,c.name as staff_member_name`).
  25. Joins(fmt.Sprintf("JOIN %s AS b on a.pen_id = b.id", new(model.Pen).TableName())).
  26. Joins(fmt.Sprintf("JOIN %s AS c on a.operation_id = c.id", new(model.SystemUser).TableName())).
  27. Where("a.pasture_id = ?", userModel.AppPasture.Id).Where("a.status = ?", pasturePb.IsShow_No)
  28. if len(req.CowId) > 0 {
  29. cowIds := strings.Split(req.CowId, ",")
  30. pref.Where("a.cow_id IN ?", cowIds)
  31. }
  32. if err = pref.Order("a.id desc").
  33. Count(&count).Limit(int(pagination.PageSize)).
  34. Offset(int(pagination.PageOffset)).
  35. Find(&lavingList).Error; err != nil {
  36. return nil, xerr.WithStack(err)
  37. }
  38. calvingIds := make([]int64, 0)
  39. for _, v := range lavingList {
  40. calvingIds = append(calvingIds, v.Id)
  41. }
  42. calvingCalfList := make([]*model.CalvingCalf, 0)
  43. if err = s.DB.Model(new(model.CalvingCalf)).
  44. Where("calving_id IN ?", calvingIds).
  45. Find(&calvingCalfList).Error; err != nil {
  46. return nil, xerr.WithStack(err)
  47. }
  48. return &pasturePb.SearchLavingEventResponse{
  49. Code: http.StatusOK,
  50. Msg: "ok",
  51. Data: &pasturePb.SearchLavingData{
  52. List: model.EventCalvingListSlice(lavingList).ToPB(calvingCalfList),
  53. Total: int32(count),
  54. PageSize: pagination.PageSize,
  55. Page: pagination.Page,
  56. },
  57. }, nil
  58. }
  59. func (s *StoreEntry) CalvingCreate(ctx context.Context, req *pasturePb.EventCalving) (err error) {
  60. userModel, err := s.GetUserModel(ctx)
  61. if err != nil {
  62. return xerr.WithStack(err)
  63. }
  64. cow, err := s.GetCowInfoByCowId(ctx, userModel.AppPasture.Id, int64(req.CowId))
  65. if err != nil {
  66. zaplog.Error("CalvingCreate", zap.Any("cow_id", req.CowId), zap.Any("err", err))
  67. return xerr.Custom("请选择相关牛只")
  68. }
  69. if len(req.CalfItemList) != int(req.ChildNumber) {
  70. return xerr.Custom("犊牛信息与产子数不相符")
  71. }
  72. if cow.IsPregnant != pasturePb.IsShow_Ok || cow.BreedStatus != pasturePb.BreedStatus_Pregnant {
  73. return xerr.Custom("该母牛未配种")
  74. }
  75. operationUser, err := s.GetSystemUserById(ctx, int64(req.OperationId))
  76. if err != nil {
  77. return xerr.Customf("获取操作人员信息失败: %s", err.Error())
  78. }
  79. req.OperationName = operationUser.Name
  80. penMap := s.PenMap(ctx, userModel.AppPasture.Id)
  81. newEventCalving := &model.EventCalving{}
  82. if err = s.DB.Model(new(model.EventCalving)).
  83. Where("cow_id = ?", cow.Id).
  84. Where("lact = ?", cow.Lact).
  85. Where("status = ?", pasturePb.IsShow_No).
  86. First(newEventCalving).Error; err != nil {
  87. return xerr.Custom("该母牛信息不存在")
  88. }
  89. if err = s.DB.Transaction(func(tx *gorm.DB) error {
  90. // 更新产犊事件表
  91. newEventCalving.EventUpdate(operationUser, userModel.SystemUser, req, cow)
  92. if err = tx.Model(new(model.EventCalving)).
  93. Select("operation_id", "operation_name", "message_id", "message_name", "reality_day", "day_age", "pregnancy_age",
  94. "bull_number", "calving_level", "child_number", "status", "is_inducing_childbirth", "pen_id", "dystocia_reason", "remarks").
  95. Where("id = ?", newEventCalving.Id).
  96. Updates(newEventCalving).Error; err != nil {
  97. return xerr.WithStack(err)
  98. }
  99. for _, calf := range req.CalfItemList {
  100. // 犊牛信息
  101. newCalvingCalf := model.NewEventCalvingCalf(userModel.AppPasture.Id, int64(req.CowId), newEventCalving.Id, int64(req.CalvingAt), penMap, calf)
  102. if err = tx.Create(newCalvingCalf).Error; err != nil {
  103. return xerr.WithStack(err)
  104. }
  105. if calf.IsLive == pasturePb.IsShow_No || calf.IsAdoption == pasturePb.IsShow_No {
  106. continue
  107. }
  108. newCalfCow := model.NewCalfCow(cow.NeckRingNumber, cow.LastBullNumber, newCalvingCalf)
  109. if err = tx.Create(newCalfCow).Error; err != nil {
  110. return xerr.WithStack(err)
  111. }
  112. if err = tx.Model(new(model.CalvingCalf)).Where("id = ?", newCalvingCalf.Id).Update("cow_id", newCalfCow.Id).Error; err != nil {
  113. return xerr.WithStack(err)
  114. }
  115. // 犊牛日志
  116. cowLogs := s.SubmitEventLog(ctx, userModel.AppPasture.Id, newCalfCow, pasturePb.EventType_Birth, pasturePb.ExposeEstrusType_Invalid, calf)
  117. if err = tx.Table(cowLogs.TableName()).Create(cowLogs).Error; err != nil {
  118. return xerr.WithStack(err)
  119. }
  120. }
  121. // 更新母牛信息
  122. cow.EventCalvingUpdate(int64(req.CalvingAt))
  123. if err = tx.Model(cow).
  124. Select("calving_age", "mating_times", "pregnancy_age", "lact", "breed_status", "is_pregnant", "last_calving_at").
  125. Where("id = ?", cow.Id).
  126. Updates(cow).Error; err != nil {
  127. return xerr.WithStack(err)
  128. }
  129. // 母牛事件日志
  130. cowLogs := s.SubmitEventLog(ctx, userModel.AppPasture.Id, cow, pasturePb.EventType_Calving, pasturePb.ExposeEstrusType_Invalid, req)
  131. if err = tx.Table(cowLogs.TableName()).Create(cowLogs).Error; err != nil {
  132. return xerr.WithStack(err)
  133. }
  134. return nil
  135. }); err != nil {
  136. return xerr.WithStack(err)
  137. }
  138. return nil
  139. }
  140. func (s *StoreEntry) SameTimeCreate(ctx context.Context, req *pasturePb.EventSameTime) (err error) {
  141. userModel, err := s.GetUserModel(ctx)
  142. if err != nil {
  143. return xerr.WithStack(err)
  144. }
  145. operationUser, err := s.GetSystemUserById(ctx, int64(req.OperationId))
  146. if err != nil {
  147. return xerr.WithStack(err)
  148. }
  149. eventCowSameTime, err := s.GetEventCowSameTimeByCowId(ctx, userModel.AppPasture.Id, int64(req.CowId))
  150. if err != nil {
  151. zaplog.Error("SameTimeCreate", zap.Any("err", err), zap.Any("req", req))
  152. return xerr.Customf("异常数据")
  153. }
  154. drugs := &model.Drugs{}
  155. if drugs, err = s.GetDrugsById(ctx, userModel.AppPasture.Id, int64(req.DrugsId)); err != nil {
  156. zaplog.Error("SameTimeCreate", zap.Any("err", err), zap.Any("req", req))
  157. return xerr.Customf("该药品不存在: %d", req.DrugsId)
  158. }
  159. req.DrugsName = drugs.Name
  160. eventCowSameTime.EventUpdate(drugs, req.Usage, req.Remarks, operationUser, userModel.SystemUser)
  161. defer func() {
  162. if err == nil {
  163. // 记录牛只事件日志
  164. cow, _ := s.GetCowInfoByCowId(ctx, userModel.AppPasture.Id, eventCowSameTime.CowId)
  165. cowLogs := s.SubmitEventLog(ctx, userModel.AppPasture.Id, cow, pasturePb.EventType_Seme_Time, pasturePb.ExposeEstrusType_Same_Time, req)
  166. s.DB.Table(cowLogs.TableName()).Create(cowLogs)
  167. }
  168. }()
  169. if err = s.DB.Model(new(model.EventCowSameTime)).
  170. Select("status", "drugs_id", "unit", "usage", "remarks", "operation_id", "operation_name").
  171. Where("id = ?", eventCowSameTime.Id).
  172. Updates(eventCowSameTime).Error; err != nil {
  173. return xerr.WithStack(err)
  174. }
  175. return nil
  176. }
  177. func (s *StoreEntry) SameTimeBatch(ctx context.Context, req *pasturePb.EventSameTimeBatch) (err error) {
  178. userModel, err := s.GetUserModel(ctx)
  179. if err != nil {
  180. return xerr.WithStack(err)
  181. }
  182. operationUser, err := s.GetSystemUserById(ctx, int64(req.OperationId))
  183. if err != nil {
  184. return xerr.Customf("异常数据")
  185. }
  186. req.OperationName = operationUser.Name
  187. drugs := &model.Drugs{}
  188. if drugs, err = s.GetDrugsById(ctx, userModel.AppPasture.Id, int64(req.DrugsId)); err != nil {
  189. zaplog.Error("SameTimeBatch", zap.Any("err", err), zap.Any("req", req))
  190. return xerr.Customf("该药物不存在: %d", req.DrugsId)
  191. }
  192. req.DrugsName = drugs.Name
  193. eventCowSameTimeList := make([]*model.EventCowSameTime, 0)
  194. for _, v := range req.CowIds {
  195. eventCowSameTime, err := s.GetEventCowSameTimeByCowId(ctx, userModel.AppPasture.Id, int64(v))
  196. if err != nil {
  197. zaplog.Error("SameTimeCreate", zap.Any("err", err), zap.Any("req", req))
  198. return xerr.WithStack(err)
  199. }
  200. eventCowSameTimeList = append(eventCowSameTimeList, eventCowSameTime)
  201. }
  202. defer func() {
  203. // 记录牛只事件日志
  204. if err == nil {
  205. for _, v := range eventCowSameTimeList {
  206. cow, _ := s.GetCowInfoByCowId(ctx, userModel.AppPasture.Id, v.CowId)
  207. cowLogs := s.SubmitEventLog(
  208. ctx,
  209. userModel.AppPasture.Id,
  210. cow,
  211. pasturePb.EventType_Seme_Time,
  212. pasturePb.ExposeEstrusType_Same_Time,
  213. &pasturePb.EventSameTime{
  214. Id: int32(v.Id),
  215. CowId: int32(v.CowId),
  216. SameTimeId: int32(v.SameTimeId),
  217. SameTimeType: v.SameTimeType,
  218. SameTimeTypeName: v.SameTimeName,
  219. DrugsId: int32(v.DrugsId),
  220. DrugsName: drugs.Name,
  221. Usage: req.Usage,
  222. OperationId: int32(operationUser.Id),
  223. OperationName: operationUser.Name,
  224. SameTimeAt: req.SameTimeAt,
  225. })
  226. s.DB.Table(cowLogs.TableName()).Create(cowLogs)
  227. }
  228. }
  229. }()
  230. if err = s.DB.Transaction(func(tx *gorm.DB) error {
  231. for _, v := range eventCowSameTimeList {
  232. // 更新SameTime
  233. v.EventUpdate(drugs, req.Usage, req.Remarks, userModel.SystemUser, operationUser)
  234. if err = tx.Model(new(model.EventCowSameTime)).
  235. Select("status", "drugs_id", "unit", "usage", "remarks", "operation_id", "operation_name").
  236. Where("id = ?", v.Id).
  237. Updates(v).Error; err != nil {
  238. return xerr.WithStack(err)
  239. }
  240. }
  241. return nil
  242. }); err != nil {
  243. return xerr.WithStack(err)
  244. }
  245. return nil
  246. }
  247. func (s *StoreEntry) SameTimeList(
  248. ctx context.Context,
  249. req *pasturePb.SearchEventRequest,
  250. pagination *pasturePb.PaginationModel,
  251. ) (*pasturePb.SearchSameTimeResponse, error) {
  252. userModel, err := s.GetUserModel(ctx)
  253. if err != nil {
  254. return nil, xerr.WithStack(err)
  255. }
  256. cowSameTimeList := make([]*model.EventCowSameTime, 0)
  257. var count int64 = 0
  258. pref := s.DB.Table(new(model.EventCowSameTime).TableName()).Where("pasture_id = ?", userModel.AppPasture.Id)
  259. if req.CowId != "" {
  260. cowIds := strings.Split(req.CowId, ",")
  261. pref.Where("cow_id IN ?", cowIds)
  262. }
  263. if err = pref.Count(&count).
  264. Limit(int(pagination.PageSize)).
  265. Offset(int(pagination.PageOffset)).
  266. Find(&cowSameTimeList).Error; err != nil {
  267. return nil, xerr.WithStack(err)
  268. }
  269. sameTimeTypeMap := s.SameTimeTypeMap()
  270. return &pasturePb.SearchSameTimeResponse{
  271. Code: http.StatusOK,
  272. Msg: "ok",
  273. Data: &pasturePb.SameTimeData{
  274. List: model.EventCowSameTimeSlice(cowSameTimeList).ToPB(sameTimeTypeMap),
  275. Total: int32(count),
  276. PageSize: pagination.PageSize,
  277. Page: pagination.Page,
  278. },
  279. }, nil
  280. }
  281. func (s *StoreEntry) EstrusList(ctx context.Context, req *pasturePb.EstrusItemsRequest, pagination *pasturePb.PaginationModel) (*pasturePb.SearchEventEstrusResponse, error) {
  282. userModel, err := s.GetUserModel(ctx)
  283. if err != nil {
  284. return nil, xerr.WithStack(err)
  285. }
  286. estrusList := make([]*model.EventEstrus, 0)
  287. var count int64 = 0
  288. pref := s.DB.Table(new(model.EventEstrus).TableName()).
  289. Where("is_show = ?", pasturePb.IsShow_Ok).
  290. Where("pasture_id = ?", userModel.AppPasture.Id).
  291. Where("expose_estrus_type = ?", pasturePb.ExposeEstrusType_Natural_Estrus)
  292. if len(req.CowIds) > 0 {
  293. cowIds := strings.Split(util.ArrayInt32ToStrings(req.CowIds, ","), ",")
  294. pref.Where("cow_id IN ?", cowIds)
  295. }
  296. if req.Level > 0 {
  297. pref.Where("level = ?", req.Level)
  298. }
  299. if len(req.PenIds) > 0 {
  300. penIds := strings.Split(util.ArrayInt32ToStrings(req.PenIds, ","), ",")
  301. pref.Where("pen_id IN ?", penIds)
  302. }
  303. if err = pref.Order("id desc").
  304. Count(&count).Limit(int(pagination.PageSize)).
  305. Offset(int(pagination.PageOffset)).
  306. Find(&estrusList).Error; err != nil {
  307. return nil, xerr.WithStack(err)
  308. }
  309. return &pasturePb.SearchEventEstrusResponse{
  310. Code: http.StatusOK,
  311. Msg: "ok",
  312. Data: &pasturePb.SearchEventEstrusData{
  313. List: model.EventEstrusSlice(estrusList).ToPB(),
  314. Total: int32(count),
  315. PageSize: pagination.PageSize,
  316. Page: pagination.Page,
  317. },
  318. }, nil
  319. }
  320. func (s *StoreEntry) EstrusBatchMating(ctx context.Context, req *pasturePb.EventNaturalEstrusBatch) (err error) {
  321. userModel, err := s.GetUserModel(ctx)
  322. if err != nil {
  323. return xerr.WithStack(err)
  324. }
  325. eventEstrusList, eventMatingList, err := s.EstrusCheckDataCheck(ctx, userModel, req.Items)
  326. if err != nil {
  327. return xerr.WithStack(err)
  328. }
  329. if err = s.DB.Transaction(func(tx *gorm.DB) error {
  330. if len(eventMatingList) > 0 {
  331. if err = tx.Model(new(model.EventMating)).Create(eventMatingList).Error; err != nil {
  332. return xerr.WithStack(err)
  333. }
  334. }
  335. if len(eventEstrusList) > 0 {
  336. if err = tx.Model(new(model.EventEstrus)).Create(eventEstrusList).Error; err != nil {
  337. return xerr.WithStack(err)
  338. }
  339. }
  340. // 记录事件日志
  341. for _, v := range eventEstrusList {
  342. cow, _ := s.GetCowInfoByCowId(ctx, userModel.AppPasture.Id, v.CowId)
  343. cowLogs := s.SubmitEventLog(ctx, userModel.AppPasture.Id, cow, pasturePb.EventType_Estrus, v.ExposeEstrusType, v)
  344. if err = tx.Table(cowLogs.TableName()).Create(cowLogs).Error; err != nil {
  345. return xerr.WithStack(err)
  346. }
  347. }
  348. return nil
  349. }); err != nil {
  350. return xerr.WithStack(err)
  351. }
  352. return nil
  353. }
  354. func (s *StoreEntry) MatingList(ctx context.Context, req *pasturePb.SearchEventRequest, pagination *pasturePb.PaginationModel) (*pasturePb.EventMatingResponse, error) {
  355. userModel, err := s.GetUserModel(ctx)
  356. if err != nil {
  357. return nil, xerr.WithStack(err)
  358. }
  359. matingList := make([]*model.EventMating, 0)
  360. var count int64 = 0
  361. pref := s.DB.Model(new(model.EventMating)).
  362. Where("mating_result > ?", pasturePb.MatingResult_Invalid).
  363. Where("status = ?", pasturePb.IsShow_Ok).
  364. Where("pasture_id = ?", userModel.AppPasture.Id)
  365. if len(req.CowId) > 0 {
  366. cowIds := strings.Split(req.CowId, ",")
  367. pref.Where("cow_id IN ?", cowIds)
  368. }
  369. if err = pref.Order("id desc").
  370. Count(&count).Limit(int(pagination.PageSize)).
  371. Offset(int(pagination.PageOffset)).
  372. Find(&matingList).Error; err != nil {
  373. return nil, xerr.WithStack(err)
  374. }
  375. exposeEstrusTypeMap := s.ExposeEstrusTypeMap()
  376. return &pasturePb.EventMatingResponse{
  377. Code: http.StatusOK,
  378. Msg: "ok",
  379. Data: &pasturePb.SearchMatingData{
  380. List: model.EventMatingSlice(matingList).ToPB(exposeEstrusTypeMap),
  381. Total: int32(count),
  382. PageSize: pagination.PageSize,
  383. Page: pagination.Page,
  384. },
  385. }, nil
  386. }
  387. // MatingCreate 牛只配种
  388. func (s *StoreEntry) MatingCreate(ctx context.Context, req *pasturePb.EventMating) (err error) {
  389. userModel, err := s.GetUserModel(ctx)
  390. if err != nil {
  391. return xerr.WithStack(err)
  392. }
  393. eventMatingCheckModel, err := s.MatingCreateCheck(ctx, userModel.AppPasture.Id, req)
  394. if err != nil {
  395. return xerr.WithStack(err)
  396. }
  397. req.OperationName = eventMatingCheckModel.OperationUser.Name
  398. if err = s.DB.Transaction(func(tx *gorm.DB) error {
  399. for _, cow := range eventMatingCheckModel.CowList {
  400. // 获取牛只最近一次配种信息
  401. lastEventMating, err := s.FindLastEventMatingByCowId(ctx, userModel.AppPasture.Id, cow.Id)
  402. if err != nil {
  403. // 1. 没有配种信息,(第一次参加配种的牛只,并且没有参与同期的牛只 )
  404. if errors.Is(err, gorm.ErrRecordNotFound) {
  405. newMating := model.NewEventMatingNaturalEstrus(userModel.AppPasture.Id, cow, req, userModel.SystemUser)
  406. if err = tx.Create(newMating).Error; err != nil {
  407. return xerr.WithStack(err)
  408. }
  409. if err = s.MatingCowUpdate(ctx, userModel.AppPasture.Id, cow, req, false); err != nil {
  410. return xerr.WithStack(err)
  411. }
  412. continue
  413. } else {
  414. return xerr.WithStack(err)
  415. }
  416. }
  417. // 2. 所有有配种数据的牛只
  418. if lastEventMating == nil || lastEventMating.Id <= 0 || lastEventMating.Status != pasturePb.IsShow_No {
  419. zaplog.Error("MatingCreate", zap.Any("cow", cow), zap.Any("UserModel", userModel))
  420. return xerr.Customf("牛只配种数据异常: %d", cow.EarNumber)
  421. }
  422. // 2.1 复配 => 本胎次配次不加1
  423. isReMating := lastEventMating.IsReMating(cow, int64(req.MatingAt))
  424. if isReMating {
  425. lastEventMating.EventReMatingUpdate(int64(req.MatingAt))
  426. if err = tx.Model(lastEventMating).
  427. Select("mating_result", "mating_result_at", "status").
  428. Where("id = ?", lastEventMating.Id).
  429. Updates(lastEventMating).Error; err != nil {
  430. }
  431. }
  432. // 2.2. 同期初配
  433. IsMatingUpdate := lastEventMating.IsMatingUpdate()
  434. if IsMatingUpdate {
  435. lastEventMating.EventUpdate(int64(req.MatingAt), req.FrozenSemenNumber, isReMating, eventMatingCheckModel.OperationUser, userModel.SystemUser)
  436. if err = tx.Model(lastEventMating).
  437. Select("mating_at", "status", "reality_day", "frozen_semen_number", "operation_id", "operation_name", "message_id", "message_name").
  438. Where("id = ?", lastEventMating.Id).
  439. Updates(lastEventMating).Error; err != nil {
  440. return xerr.WithStack(err)
  441. }
  442. }
  443. // 2.3. 提交的配种数据中,定位出空怀的牛只,
  444. isEmptyMating := lastEventMating.IsEmptyMating(cow, int64(req.MatingAt))
  445. if isEmptyMating {
  446. // 把上次配种结果信息更新成空怀
  447. lastEventMating.EventMatingResultUpdate(pasturePb.MatingResult_Empty, int64(req.MatingAt))
  448. if err = tx.Model(lastEventMating).
  449. Select("mating_result", "mating_result_at").
  450. Where("id = ?", lastEventMating.Id).
  451. Updates(lastEventMating).Error; err != nil {
  452. return xerr.WithStack(err)
  453. }
  454. // 先创建一条新的配种数据
  455. newMating := model.NewEventMating(userModel.AppPasture.Id, cow, int64(req.MatingAt), req.ExposeEstrusType)
  456. newMating.EventUpdate(int64(req.MatingAt), req.FrozenSemenNumber, isReMating, eventMatingCheckModel.OperationUser, userModel.SystemUser)
  457. if err = tx.Model(newMating).Create(newMating).Error; err != nil {
  458. return xerr.WithStack(err)
  459. }
  460. // 更新牛只配种结果日志
  461. if err = s.UpdateMatingResultEventCowLogByCowId(ctx, cow.Id, s.MatingResultMap()[pasturePb.MatingResult_Empty]); err != nil {
  462. zaplog.Error("MatingCreate", zap.Any("UpdateEventCowLogByCowId", err), zap.Any("cow", cow))
  463. }
  464. }
  465. // 牛只基本中配种信息更新
  466. if err = s.MatingCowUpdate(ctx, userModel.AppPasture.Id, cow, req, isReMating); err != nil {
  467. return xerr.WithStack(err)
  468. }
  469. }
  470. // 创建冻精使用记录日志
  471. itemFrozenSemenLog := model.NewEventFrozenSemenLog(eventMatingCheckModel.FrozenSemen.PastureId, req)
  472. if err = tx.Create(itemFrozenSemenLog).Error; err != nil {
  473. return xerr.WithStack(err)
  474. }
  475. // 减去精液的数量
  476. eventMatingCheckModel.FrozenSemen.EventQuantityUpdate(req.FrozenSemenCount)
  477. if err = tx.Model(eventMatingCheckModel.FrozenSemen).
  478. Select("quantity").
  479. Where("id = ?", eventMatingCheckModel.FrozenSemen.Id).
  480. Updates(eventMatingCheckModel.FrozenSemen).Error; err != nil {
  481. return xerr.WithStack(err)
  482. }
  483. return nil
  484. }); err != nil {
  485. return xerr.WithStack(err)
  486. }
  487. return nil
  488. }
  489. func (s *StoreEntry) MatingCowUpdate(ctx context.Context, pastureId int64, cow *model.Cow, req *pasturePb.EventMating, isReMating bool) error {
  490. // 牛只基本中配种信息更新
  491. cow.EventMatingUpdate(int64(req.MatingAt), req.FrozenSemenNumber, isReMating)
  492. if err := s.DB.Model(cow).
  493. Select("last_mating_at", "mating_times", "last_bull_number", "first_mating_at", "is_pregnant", "breed_status").
  494. Where("id = ?", cow.Id).
  495. Updates(cow).Error; err != nil {
  496. return xerr.WithStack(err)
  497. }
  498. // 记录日志
  499. cowLogs := s.SubmitEventLog(ctx, pastureId, cow, pasturePb.EventType_Mating, req.ExposeEstrusType, req)
  500. if err := s.DB.Table(cowLogs.TableName()).Create(cowLogs).Error; err != nil {
  501. return xerr.WithStack(err)
  502. }
  503. return nil
  504. }
  505. func (s *StoreEntry) WeaningBatch(ctx context.Context, req *pasturePb.EventWeaningBatchRequest) (err error) {
  506. userModel, err := s.GetUserModel(ctx)
  507. if err != nil {
  508. return xerr.WithStack(err)
  509. }
  510. if len(req.Item) <= 0 {
  511. return nil
  512. }
  513. cowIds := make([]int64, 0)
  514. cowWeightMap := make(map[int64]float64)
  515. for _, item := range req.Item {
  516. cowIds = append(cowIds, int64(item.CowId))
  517. cowWeightMap[int64(item.CowId)] = float64(item.Weight)
  518. }
  519. eventWeaningList := make([]*model.EventWeaning, 0)
  520. if err = s.DB.Model(new(model.EventWeaning)).
  521. Where("cow_id IN ?", cowIds).
  522. Where("status = ?", pasturePb.IsShow_No).
  523. Find(&eventWeaningList).Error; err != nil {
  524. return xerr.WithStack(err)
  525. }
  526. operation, err := s.GetSystemUserById(ctx, int64(req.OperationId))
  527. if err != nil {
  528. return xerr.WithStack(err)
  529. }
  530. defer func() {
  531. if err != nil {
  532. for _, cowId := range cowIds {
  533. cow, _ := s.GetCowInfoByCowId(ctx, userModel.AppPasture.Id, cowId)
  534. cowLogs := s.SubmitEventLog(ctx, userModel.AppPasture.Id, cow, pasturePb.EventType_Weaning, pasturePb.ExposeEstrusType_Invalid, req)
  535. s.DB.Table(cowLogs.TableName()).Create(cowLogs)
  536. }
  537. }
  538. }()
  539. if err = s.DB.Transaction(func(tx *gorm.DB) error {
  540. cowInfo := &model.Cow{}
  541. for _, eventWeaning := range eventWeaningList {
  542. eventWeaning.EventUpdate(int64(req.WeaningAt), int32(cowWeightMap[cowInfo.Id]*1000), req.Remarks, req.PenId, operation, userModel.SystemUser)
  543. if err = tx.Model(new(model.EventWeaning)).
  544. Select("status", "reality_day", "operation_id", "operation_name", "message_id", "message_name", "remarks", "after_pen_id").
  545. Where("id = ?", eventWeaning.Id).
  546. Updates(eventWeaning).Error; err != nil {
  547. return xerr.WithStack(err)
  548. }
  549. cowInfo, err = s.GetCowInfoByCowId(ctx, userModel.AppPasture.Id, eventWeaning.CowId)
  550. if err != nil {
  551. return xerr.WithStack(err)
  552. }
  553. cowInfo.EventWeaningUpdate(int64(req.WeaningAt), req.PenId, int64(cowWeightMap[cowInfo.Id]*1000))
  554. if err = tx.Model(new(model.Cow)).
  555. Select("pen_id", "current_weight", "weaning_at", "last_weight_at").
  556. Where("id = ?", cowInfo.Id).
  557. Updates(cowInfo).Error; err != nil {
  558. return xerr.WithStack(err)
  559. }
  560. // 创建牛只的体重记录
  561. newEventWeight := model.NewEventWeight(
  562. userModel.AppPasture.Id,
  563. cowInfo,
  564. userModel.SystemUser,
  565. &pasturePb.EventWeight{
  566. WeightAt: req.WeaningAt,
  567. OperationId: req.OperationId,
  568. OperationName: req.OperationName,
  569. Remarks: req.Remarks,
  570. Weight: float32(cowWeightMap[cowInfo.Id]),
  571. Height: 0,
  572. })
  573. if err = tx.Create(newEventWeight).Error; err != nil {
  574. return xerr.WithStack(err)
  575. }
  576. }
  577. return nil
  578. }); err != nil {
  579. return xerr.WithStack(err)
  580. }
  581. return nil
  582. }