event_breed.go 20 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652
  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. "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. lavingList := make([]*model.EventCalvingList, 0)
  18. var count int64 = 0
  19. pref := s.DB.Table(fmt.Sprintf("%s as a", new(model.EventCalving).TableName())).
  20. Select(`a.*,b.name as pen_name,c.name as staff_member_name`).
  21. Joins(fmt.Sprintf("JOIN %s AS b on a.pen_id = b.id", new(model.Pen).TableName())).
  22. Joins(fmt.Sprintf("JOIN %s AS c on a.operation_id = c.id", new(model.SystemUser).TableName()))
  23. if len(req.CowId) > 0 {
  24. cowIds := strings.Split(req.CowId, ",")
  25. pref.Where("a.cow_id IN ?", cowIds)
  26. }
  27. if err := pref.Order("a.id desc").
  28. Count(&count).Limit(int(pagination.PageSize)).
  29. Offset(int(pagination.PageOffset)).
  30. Find(&lavingList).Error; err != nil {
  31. return nil, xerr.WithStack(err)
  32. }
  33. calvingIds := make([]int64, 0)
  34. for _, v := range lavingList {
  35. calvingIds = append(calvingIds, v.Id)
  36. }
  37. calvingCalfList := make([]*model.CalvingCalf, 0)
  38. if err := s.DB.Model(new(model.CalvingCalf)).
  39. Where("calving_id IN ?", calvingIds).
  40. Find(&calvingCalfList).Error; err != nil {
  41. return nil, xerr.WithStack(err)
  42. }
  43. return &pasturePb.SearchLavingEventResponse{
  44. Code: http.StatusOK,
  45. Message: "ok",
  46. Data: &pasturePb.SearchLavingData{
  47. List: model.EventCalvingListSlice(lavingList).ToPB(calvingCalfList),
  48. Total: int32(count),
  49. PageSize: pagination.PageSize,
  50. Page: pagination.Page,
  51. },
  52. }, nil
  53. }
  54. func (s *StoreEntry) CalvingCreate(ctx context.Context, req *pasturePb.EventCalving) error {
  55. if len(req.CowId) <= 0 {
  56. return xerr.Custom("请选择相关牛只")
  57. }
  58. cowList, err := s.ParseCowIds(ctx, req.CowId)
  59. if err != nil {
  60. return xerr.WithStack(err)
  61. }
  62. if len(cowList) <= 0 {
  63. return xerr.Custom("请选择相关牛只")
  64. }
  65. cow := cowList[0]
  66. newEventCalving := &model.EventCalving{}
  67. if err = s.DB.Model(new(model.EventCalving)).Where("cow_id = ?", cow.Id).Where("lact = ?", cow.Lact).First(newEventCalving).Error; err != nil {
  68. return xerr.Custom("该母牛信息不存在")
  69. }
  70. if err = s.DB.Transaction(func(tx *gorm.DB) error {
  71. // 母牛信息
  72. if err = tx.Model(new(model.EventCalving)).
  73. Where("id = ?", newEventCalving.Id).
  74. Updates(map[string]interface{}{
  75. "reality_day": int64(req.CalvingAt),
  76. "day_age": cow.DayAge,
  77. "lact": cow.Lact + 1,
  78. "pregnancy_age": cow.PregnancyAge,
  79. "calving_level": req.CalvingLevel,
  80. "bull_number": cow.LastBullNumber,
  81. "child_number": len(req.CalfItemList),
  82. }).
  83. Error; err != nil {
  84. return xerr.WithStack(err)
  85. }
  86. // 犊牛信息
  87. newCalvingCalfList := model.NewEventCalvingCalf(cow.Id, newEventCalving.Id, req)
  88. for _, v := range newCalvingCalfList {
  89. if v.IsLive == pasturePb.IsShow_No || v.IsAdoption == pasturePb.IsShow_No {
  90. continue
  91. }
  92. // 留养犊牛
  93. newCow := model.NewCalfCow(cow.Id, cow.LastBullNumber, v)
  94. if err = tx.Create(newCow).Error; err != nil {
  95. return xerr.WithStack(err)
  96. }
  97. v.CowId = newCow.Id
  98. }
  99. if err = tx.Create(newCalvingCalfList).Error; err != nil {
  100. return xerr.WithStack(err)
  101. }
  102. if err = tx.Model(new(model.Cow)).Where("id = ?", cow.Id).
  103. Updates(map[string]interface{}{
  104. "calving_age": 0,
  105. "mating_times": 0,
  106. "lact": cow.Lact + 1,
  107. "breed_status": pasturePb.BreedStatus_Calving,
  108. "is_pregnant": pasturePb.IsShow_No,
  109. "last_calving_at": int64(req.CalvingAt),
  110. }).Error; err != nil {
  111. return xerr.WithStack(err)
  112. }
  113. return nil
  114. }); err != nil {
  115. return xerr.WithStack(err)
  116. }
  117. return nil
  118. }
  119. func (s *StoreEntry) EstrusList(ctx context.Context, req *pasturePb.SearchEventRequest, pagination *pasturePb.PaginationModel) (*pasturePb.EstrusEventResponse, error) {
  120. estrusList := make([]*model.EventEstrus, 0)
  121. var count int64 = 0
  122. pref := s.DB.Table(new(model.EventEstrus).TableName())
  123. if len(req.CowId) > 0 {
  124. cowIds := strings.Split(req.CowId, ",")
  125. pref.Where("cow_id IN ?", cowIds)
  126. }
  127. if err := pref.Order("id desc").
  128. Count(&count).Limit(int(pagination.PageSize)).
  129. Offset(int(pagination.PageOffset)).
  130. Find(&estrusList).Error; err != nil {
  131. return nil, xerr.WithStack(err)
  132. }
  133. return &pasturePb.EstrusEventResponse{
  134. Code: http.StatusOK,
  135. Message: "ok",
  136. Data: &pasturePb.SearchEstrusData{
  137. List: model.EstrusSlice(estrusList).ToPB(),
  138. Total: int32(count),
  139. PageSize: pagination.PageSize,
  140. Page: pagination.Page,
  141. },
  142. }, nil
  143. }
  144. func (s *StoreEntry) EstrusCreate(ctx context.Context, req *pasturePb.EventEstrus) error {
  145. estrusInfo, err := s.EventEstrusCheck(ctx, req)
  146. if err != nil {
  147. return xerr.Custom("获取当前登录用户失败")
  148. }
  149. if err = s.DB.Transaction(func(tx *gorm.DB) error {
  150. if err = tx.Create(estrusInfo).Error; err != nil {
  151. return xerr.WithStack(err)
  152. }
  153. if estrusInfo.IsMating == pasturePb.IsShow_Ok {
  154. cow, _ := s.GetCowInfoByCowId(ctx, int64(req.CowId))
  155. newEventMating := model.NewEventMating(cow, estrusInfo.EstrusAt, estrusInfo.ExposeEstrusType)
  156. if err = tx.Create(newEventMating).Error; err != nil {
  157. return xerr.WithStack(err)
  158. }
  159. }
  160. return nil
  161. }); err != nil {
  162. return xerr.WithStack(err)
  163. }
  164. return nil
  165. }
  166. func (s *StoreEntry) EstrusBatch(ctx context.Context, req *pasturePb.EventEstrusBatch) error {
  167. estrusList := make([]*model.EventEstrus, 0)
  168. eventMatingList := make([]*model.EventMating, 0)
  169. for _, v := range req.Item {
  170. estrusInfo, err := s.EventEstrusCheck(ctx, v)
  171. if err != nil {
  172. return xerr.Custom("获取当前登录用户失败")
  173. }
  174. estrusList = append(estrusList, estrusInfo)
  175. cow, _ := s.GetCowInfoByCowId(ctx, int64(v.CowId))
  176. eventMatingList = append(eventMatingList, model.NewEventMating(cow, estrusInfo.EstrusAt, estrusInfo.ExposeEstrusType))
  177. }
  178. if err := s.DB.Transaction(func(tx *gorm.DB) error {
  179. if err := tx.Create(estrusList).Error; err != nil {
  180. return xerr.WithStack(err)
  181. }
  182. if len(eventMatingList) > 0 {
  183. if err := tx.Create(eventMatingList).Error; err != nil {
  184. return xerr.WithStack(err)
  185. }
  186. }
  187. return nil
  188. }); err != nil {
  189. return xerr.WithStack(err)
  190. }
  191. return nil
  192. }
  193. func (s *StoreEntry) SameTimeCreate(ctx context.Context, req *pasturePb.EventSameTime) error {
  194. eventCowSameTime, err := s.GetEventCowSameTimeById(ctx, int64(req.Id))
  195. if err != nil {
  196. return xerr.WithStack(err)
  197. }
  198. if eventCowSameTime.Status == pasturePb.IsShow_Ok {
  199. return xerr.Custom("该事件已处理")
  200. }
  201. if eventCowSameTime.CowId != int64(req.CowId) {
  202. return xerr.Custom("牛只Id 不匹配")
  203. }
  204. drugs := &model.Drugs{}
  205. if req.DrugsId > 0 {
  206. if drugs, err = s.GetDrugsById(ctx, int64(req.DrugsId)); err != nil {
  207. return xerr.WithStack(err)
  208. }
  209. }
  210. operationUser, err := s.GetSystemUserById(ctx, int64(req.OperationId))
  211. if err != nil {
  212. return xerr.WithStack(err)
  213. }
  214. if err = s.DB.Transaction(func(tx *gorm.DB) error {
  215. if err = tx.Model(new(model.CowSameTime)).
  216. Where("cow_id = ?", eventCowSameTime.CowId).
  217. Where("same_time_id = ?", eventCowSameTime.Id).
  218. Where("same_time_status = ?", pasturePb.SameTimeStatus_No_Start).
  219. Update("same_time_status", pasturePb.SameTimeStatus_In_Progress).Error; err != nil {
  220. zaplog.Error("SameTimeCreate", zap.Any("err", err), zap.Any("req", req))
  221. }
  222. if err = tx.Model(new(model.EventCowSameTime)).
  223. Where("id = ?", eventCowSameTime.Id).
  224. Where("cow_id = ?", eventCowSameTime.CowId).
  225. Where("status = ?", pasturePb.IsShow_No).
  226. Updates(map[string]interface{}{
  227. "status": pasturePb.IsShow_Ok,
  228. "drug_id": drugs.Id,
  229. "unit": drugs.Unit,
  230. "usage": req.Usage,
  231. "remarks": req.Remarks,
  232. "operation_id": operationUser.Id,
  233. "operation_name": operationUser.Name,
  234. }).Error; err != nil {
  235. return xerr.WithStack(err)
  236. }
  237. return nil
  238. }); err != nil {
  239. return xerr.WithStack(err)
  240. }
  241. return nil
  242. }
  243. func (s *StoreEntry) SameTimeBatch(ctx context.Context, req *pasturePb.EventSameTimeBatch) error {
  244. drugs := &model.Drugs{}
  245. var err error
  246. if req.DrugsId > 0 {
  247. if drugs, err = s.GetDrugsById(ctx, int64(req.DrugsId)); err != nil {
  248. return xerr.WithStack(err)
  249. }
  250. }
  251. operationUser, err := s.GetSystemUserById(ctx, int64(req.OperationId))
  252. if err != nil {
  253. return xerr.WithStack(err)
  254. }
  255. eventCowSameTimeList := make([]*model.EventCowSameTime, 0)
  256. for _, v := range req.Item {
  257. eventCowSameTime, err := s.GetEventCowSameTimeById(ctx, int64(v.Id))
  258. if err != nil {
  259. return xerr.WithStack(err)
  260. }
  261. if eventCowSameTime.Status == pasturePb.IsShow_Ok {
  262. return xerr.Custom("该事件已处理")
  263. }
  264. if eventCowSameTime.CowId != int64(v.CowId) {
  265. return xerr.Custom(" 牛只Id 不匹配")
  266. }
  267. eventCowSameTimeList = append(eventCowSameTimeList, eventCowSameTime)
  268. }
  269. if err = s.DB.Transaction(func(tx *gorm.DB) error {
  270. for _, v := range eventCowSameTimeList {
  271. if err = tx.Model(new(model.CowSameTime)).
  272. Where("cow_id = ?", v.CowId).
  273. Where("same_time_id = ?", v.Id).
  274. Where("same_time_status = ?", pasturePb.SameTimeStatus_No_Start).
  275. Update("same_time_status", pasturePb.SameTimeStatus_In_Progress).Error; err != nil {
  276. zaplog.Error("SameTimeCreate", zap.Any("err", err), zap.Any("req", req))
  277. }
  278. if err = tx.Model(new(model.EventCowSameTime)).
  279. Where("id = ?", v.Id).
  280. Where("cow_id = ?", v.CowId).
  281. Where("status = ?", pasturePb.IsShow_No).
  282. Updates(map[string]interface{}{
  283. "status": pasturePb.IsShow_Ok,
  284. "drugs_id": drugs.Id,
  285. "unit": drugs.Unit,
  286. "usage": req.Usage,
  287. "remarks": req.Remarks,
  288. "operation_id": operationUser.Id,
  289. "operation_name": operationUser.Name,
  290. }).Error; err != nil {
  291. return xerr.WithStack(err)
  292. }
  293. }
  294. return nil
  295. }); err != nil {
  296. return xerr.WithStack(err)
  297. }
  298. return nil
  299. }
  300. func (s *StoreEntry) SameTimeList(
  301. ctx context.Context,
  302. req *pasturePb.SearchEventRequest,
  303. pagination *pasturePb.PaginationModel,
  304. ) (*pasturePb.SearchSameTimeResponse, error) {
  305. cowSameTimeList := make([]*model.EventCowSameTime, 0)
  306. var count int64 = 0
  307. pref := s.DB.Table(new(model.EventCowSameTime).TableName())
  308. if req.CowId != "" {
  309. cowIds := strings.Split(req.CowId, ",")
  310. pref.Where("cow_id IN ?", cowIds)
  311. }
  312. if err := pref.Count(&count).
  313. Limit(int(pagination.PageSize)).
  314. Offset(int(pagination.PageOffset)).
  315. Find(&cowSameTimeList).Error; err != nil {
  316. return nil, xerr.WithStack(err)
  317. }
  318. sameTimeTypeMap := s.SameTimeTypeMap()
  319. return &pasturePb.SearchSameTimeResponse{
  320. Code: http.StatusOK,
  321. Message: "ok",
  322. Data: &pasturePb.SameTimeData{
  323. List: model.EventCowSameTimeSlice(cowSameTimeList).ToPB(sameTimeTypeMap),
  324. Total: int32(count),
  325. PageSize: pagination.PageSize,
  326. Page: pagination.Page,
  327. },
  328. }, nil
  329. }
  330. func (s *StoreEntry) MatingList(ctx context.Context, req *pasturePb.SearchEventRequest, pagination *pasturePb.PaginationModel) (*pasturePb.MatingEventResponse, error) {
  331. matingList := make([]*model.EventMating, 0)
  332. var count int64 = 0
  333. pref := s.DB.Model(new(model.EventMating)).
  334. Where("mating_result > ?", pasturePb.MatingResult_Invalid).
  335. Where("status = ?", pasturePb.IsShow_No)
  336. if len(req.CowId) > 0 {
  337. cowIds := strings.Split(req.CowId, ",")
  338. pref.Where("cow_id IN ?", cowIds)
  339. }
  340. if err := pref.Order("id desc").
  341. Count(&count).Limit(int(pagination.PageSize)).
  342. Offset(int(pagination.PageOffset)).
  343. Find(&matingList).Error; err != nil {
  344. return nil, xerr.WithStack(err)
  345. }
  346. exposeEstrusTypeMap := s.ExposeEstrusTypeMap()
  347. return &pasturePb.MatingEventResponse{
  348. Code: http.StatusOK,
  349. Message: "ok",
  350. Data: &pasturePb.SearchMatingData{
  351. List: model.EventMatingSlice(matingList).ToPB(exposeEstrusTypeMap),
  352. Total: int32(count),
  353. PageSize: pagination.PageSize,
  354. Page: pagination.Page,
  355. },
  356. }, nil
  357. }
  358. func (s *StoreEntry) MatingCreate(ctx context.Context, req *pasturePb.EventMating) error {
  359. eventCheckModel, err := s.MatingCreateCheck(ctx, req)
  360. if err != nil {
  361. return xerr.WithStack(err)
  362. }
  363. frozenSemen := &model.FrozenSemen{}
  364. if err = s.DB.Where("bull_id = ?", req.FrozenSemenNumber).
  365. First(frozenSemen).Error; err != nil {
  366. return xerr.Custom("未找到冻精信息")
  367. }
  368. if frozenSemen.Quantity < req.FrozenSemenCount {
  369. return xerr.Custom("冻精数量不足")
  370. }
  371. // 更新复配的牛只
  372. matingReMatchIds := make([]int64, 0)
  373. // 新建配种信息的牛只
  374. newMatingList := make([]*model.EventMating, 0)
  375. // 更新配种信息牛只
  376. updateMatingList := make([]int64, 0)
  377. // 所有牛只
  378. cowIds := make([]int64, 0)
  379. // 需要更新配次的牛只
  380. matingTimes := make([]*MatingTimes, 0)
  381. // 需要更新空怀的牛只
  382. emptyCowIds := make([]int64, 0)
  383. for _, cow := range eventCheckModel.CowList {
  384. // 1. 第一次配种,创建配种信息(自然发情牛只,未经过同期的牛只)
  385. if cow.LastMatingAt <= 0 {
  386. newMatingList = append(newMatingList, model.NewEventMating2(cow, req, eventCheckModel.CurrentUser))
  387. continue
  388. }
  389. eventMatingHistory, err := s.GetEventMatingIsExIstByCowId(ctx, cow)
  390. if err != nil {
  391. return xerr.WithStack(err)
  392. }
  393. // 2. 同期更新配种信息 牛号&& 胎次 && 未配 && 配种结果未知 ==> 同期初配
  394. if eventMatingHistory.Status == pasturePb.IsShow_No &&
  395. eventMatingHistory.MatingResult == pasturePb.MatingResult_Unknown {
  396. updateMatingList = append(updateMatingList, eventMatingHistory.Id)
  397. continue
  398. }
  399. lastMatingAt := time.Unix(cow.LastMatingAt, 0)
  400. currentMatingAt := time.Unix(int64(req.MatingAt), 0)
  401. daysBetween := util.DaysBetween(currentMatingAt.Unix(), lastMatingAt.Unix())
  402. // 3. 如何两次配种时间为连续两天之内&&有过一次配种记录,则更新为复配状态 牛号&& 胎次 && 已配 && 配种结果未知 ==> 复配
  403. if (currentMatingAt.Format(model.LayoutDate2) == lastMatingAt.Format(model.LayoutDate2) || daysBetween == 1) &&
  404. eventMatingHistory.Status == pasturePb.IsShow_Ok &&
  405. eventMatingHistory.MatingResult == pasturePb.MatingResult_Unknown {
  406. matingReMatchIds = append(matingReMatchIds, eventMatingHistory.Id)
  407. } else {
  408. matingTimes = append(matingTimes, &MatingTimes{
  409. Mt: cow.MatingTimes + 1,
  410. CowId: cow.Id,
  411. EventMatingId: eventMatingHistory.Id,
  412. })
  413. }
  414. // 4. 提交的配种数据中,如何定位出空怀的牛只,
  415. if (eventMatingHistory.MatingResult == pasturePb.MatingResult_Unknown || eventMatingHistory.MatingResult == pasturePb.MatingResult_ReMatch) &&
  416. daysBetween >= 2 {
  417. emptyCowIds = append(emptyCowIds, eventMatingHistory.Id)
  418. newMatingList = append(newMatingList, model.NewEventMating2(cow, req, eventCheckModel.CurrentUser))
  419. }
  420. cowIds = append(cowIds, cow.Id)
  421. }
  422. if len(cowIds) != len(updateMatingList)+len(matingReMatchIds)+len(newMatingList) {
  423. zaplog.Error("MatingCreate",
  424. zap.Any("cowIds", cowIds),
  425. zap.Any("updateMatingList", updateMatingList),
  426. zap.Any("newMatingList", newMatingList),
  427. zap.Any("matingReMatchIds", matingReMatchIds),
  428. zap.Any("req", req),
  429. )
  430. return xerr.Custom("配种信息有误")
  431. }
  432. itemFrozenSemenLog := model.NewEventFrozenSemenLog(req)
  433. if err = s.DB.Transaction(func(tx *gorm.DB) error {
  434. // 更新配种事件数据(初配)
  435. if len(updateMatingList) > 0 {
  436. if err = tx.Model(new(model.EventMating)).
  437. Where("id IN ?", updateMatingList).
  438. Where("status = ?", pasturePb.IsShow_No).
  439. Updates(map[string]interface{}{
  440. "mating_result": pasturePb.MatingResult_Unknown,
  441. "status": pasturePb.IsShow_Ok,
  442. "reality_day": int64(req.MatingAt),
  443. "frozen_semen_number": req.FrozenSemenNumber,
  444. "operation_id": eventCheckModel.OperationUser.Id,
  445. "operation_name": eventCheckModel.OperationUser.Name,
  446. "message_id": eventCheckModel.CurrentUser.Id,
  447. "message_name": eventCheckModel.CurrentUser.Name,
  448. }).Error; err != nil {
  449. return xerr.WithStack(err)
  450. }
  451. }
  452. // 更新已配种的牛只为复配状态
  453. if len(matingReMatchIds) > 0 {
  454. if err = tx.Model(new(model.EventMating)).
  455. Where("id IN ?", matingReMatchIds).
  456. Where("mating_result = ?", pasturePb.MatingResult_Unknown).
  457. Where("status = ?", pasturePb.IsShow_Ok).
  458. Update("mating_result", pasturePb.MatingResult_ReMatch).
  459. Error; err != nil {
  460. return xerr.WithStack(err)
  461. }
  462. }
  463. // 更新上次已配种的牛只为空怀
  464. if len(emptyCowIds) > 0 {
  465. if err = tx.Model(new(model.EventMating)).
  466. Where("id IN ?", emptyCowIds).
  467. Update("mating_result", pasturePb.BreedStatus_Empty).
  468. Error; err != nil {
  469. return xerr.WithStack(err)
  470. }
  471. }
  472. // 创建配种事件数据
  473. if len(newMatingList) > 0 {
  474. if err = tx.Create(newMatingList).Error; err != nil {
  475. return xerr.WithStack(err)
  476. }
  477. }
  478. // 创建冻精使用记录日志
  479. if err = tx.Create(itemFrozenSemenLog).Error; err != nil {
  480. return xerr.WithStack(err)
  481. }
  482. // 如果有同期牛只,则修改为已结束状态
  483. if err = tx.Model(new(model.CowSameTime)).
  484. Where("cow_id IN ?", cowIds).
  485. UpdateColumn("same_time_status", pasturePb.SameTimeStatus_End).Error; err != nil {
  486. return xerr.WithStack(err)
  487. }
  488. // 更新牛只的繁殖状态为配种
  489. if err = tx.Model(new(model.Cow)).
  490. Where("id IN ?", cowIds).
  491. Where("admission_status = ?", pasturePb.AdmissionStatus_Admission).
  492. Updates(map[string]interface{}{
  493. "breed_status": pasturePb.BreedStatus_Breeding,
  494. "last_bull_number": req.FrozenSemenNumber,
  495. "last_mating_at": int64(req.MatingAt),
  496. }).Error; err != nil {
  497. return xerr.WithStack(err)
  498. }
  499. // 减去精液的数量
  500. if err = tx.Model(frozenSemen).
  501. Where("id = ?", frozenSemen.Id).
  502. Where("quantity > 0").
  503. UpdateColumn("quantity",
  504. gorm.Expr("quantity - ?", req.FrozenSemenCount),
  505. ).Error; err != nil {
  506. return xerr.WithStack(err)
  507. }
  508. // 更新配次
  509. if len(matingTimes) > 0 {
  510. for _, mt := range matingTimes {
  511. if err = tx.Model(new(model.EventMating)).
  512. Where("id = ?", mt.EventMatingId).
  513. Update("mating_times", mt.Mt).Error; err != nil {
  514. return xerr.WithStack(err)
  515. }
  516. if err = tx.Model(new(model.Cow)).
  517. Where("id = ?", mt.CowId).
  518. Where("admission_status = ?", pasturePb.AdmissionStatus_Admission).
  519. Where("breed_status = ?", pasturePb.BreedStatus_Breeding).
  520. Update("mating_times", mt.Mt).Error; err != nil {
  521. return xerr.WithStack(err)
  522. }
  523. }
  524. }
  525. return nil
  526. }); err != nil {
  527. return xerr.WithStack(err)
  528. }
  529. return nil
  530. }
  531. func (s *StoreEntry) WeaningBatch(ctx context.Context, req *pasturePb.EventWeaningBatchRequest) error {
  532. if len(req.Item) <= 0 {
  533. return nil
  534. }
  535. cowIds := make([]int64, 0)
  536. cowWeightMap := make(map[int64]float64)
  537. for _, item := range req.Item {
  538. cowIds = append(cowIds, int64(item.CowId))
  539. cowWeightMap[int64(item.CowId)] = float64(item.Weight)
  540. }
  541. eventWeaningList := make([]*model.EventWeaning, 0)
  542. if err := s.DB.Model(new(model.EventWeaning)).
  543. Where("cow_id IN ?", cowIds).
  544. Where("status = ?", pasturePb.IsShow_No).
  545. Find(&eventWeaningList).Error; err != nil {
  546. return xerr.WithStack(err)
  547. }
  548. operation, err := s.GetSystemUserById(ctx, int64(req.OperationId))
  549. if err != nil {
  550. return xerr.WithStack(err)
  551. }
  552. currentUser, err := s.GetCurrentSystemUser(ctx)
  553. if err != nil {
  554. return xerr.WithStack(err)
  555. }
  556. if err = s.DB.Transaction(func(tx *gorm.DB) error {
  557. for _, v := range eventWeaningList {
  558. if err = tx.Model(new(model.EventWeaning)).Where("id = ?", v.Id).Updates(map[string]interface{}{
  559. "status": pasturePb.IsShow_Ok,
  560. "reality_day": int64(req.WeaningAt),
  561. "operation_id": req.OperationId,
  562. "operation_name": operation.Name,
  563. "message_id": currentUser.Id,
  564. "message_name": currentUser.Name,
  565. "remarks": req.Remarks,
  566. "after_pen_id": req.PenId,
  567. }).Error; err != nil {
  568. return xerr.WithStack(err)
  569. }
  570. if err = tx.Model(new(model.Cow)).Where("id = ?", v.CowId).Updates(map[string]interface{}{
  571. "pen_id": req.PenId,
  572. "current_weight": cowWeightMap[v.CowId],
  573. "weaning_at": req.WeaningAt,
  574. "last_weight_at": req.WeaningAt,
  575. }).Error; err != nil {
  576. return xerr.WithStack(err)
  577. }
  578. }
  579. return nil
  580. }); err != nil {
  581. return xerr.WithStack(err)
  582. }
  583. return nil
  584. }