event_breed.go 20 KB

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