event_breed.go 21 KB

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