event_health.go 20 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618
  1. package backend
  2. import (
  3. "context"
  4. "fmt"
  5. "kpt-pasture/model"
  6. "net/http"
  7. "strings"
  8. "time"
  9. "gitee.com/xuyiping_admin/pkg/logger/zaplog"
  10. "go.uber.org/zap"
  11. "gorm.io/gorm"
  12. pasturePb "gitee.com/xuyiping_admin/go_proto/proto/go/backend/cow"
  13. "gitee.com/xuyiping_admin/pkg/xerr"
  14. )
  15. // CowDiseaseCreate 牛只发病提交
  16. func (s *StoreEntry) CowDiseaseCreate(ctx context.Context, req *pasturePb.EventCowDiseaseRequest, source string) error {
  17. userModel, err := s.GetUserModel(ctx)
  18. if err != nil {
  19. return xerr.WithStack(err)
  20. }
  21. // 牛只信息
  22. cow, err := s.GetCowInfoByEarNumber(ctx, userModel.AppPasture.Id, req.EarNumber)
  23. if err != nil {
  24. return xerr.Customf("牛只信息错误: %d", req.CowId)
  25. }
  26. operationUser, err := s.GetSystemUserById(ctx, int64(req.OperationId))
  27. if err != nil {
  28. return xerr.Customf("请检查操作人信息")
  29. }
  30. disease, err := s.GetDiseaseById(ctx, userModel.AppPasture.Id, req.DiseaseId)
  31. if err != nil {
  32. return xerr.WithStack(err)
  33. }
  34. newEventCowDisease := model.NewEventCowDisease(userModel.AppPasture.Id, cow, disease, req, operationUser, userModel.SystemUser)
  35. defer func() {
  36. // 更新牛只健康状态
  37. if newEventCowDisease.HealthStatus == pasturePb.HealthStatus_Disease || newEventCowDisease.HealthStatus == pasturePb.HealthStatus_Treatment {
  38. cow.EventHealthStatusUpdate(pasturePb.HealthStatus_Disease)
  39. if err = s.DB.Model(new(model.Cow)).
  40. Select("health_status").
  41. Where("id = ?", req.CowId).
  42. Updates(cow).Error; err != nil {
  43. zaplog.Error("CowDiseaseCreate", zap.Any("EventHealthStatusUpdate", err))
  44. }
  45. }
  46. if req.PenId > 0 {
  47. penMap := s.PenMap(ctx, userModel.AppPasture.Id)
  48. penData, ok := penMap[req.PenId]
  49. if !ok {
  50. return
  51. }
  52. cow.EventPenUpdate(penData)
  53. if err = s.DB.Model(new(model.Cow)).
  54. Select("pen_id", "pen_name").
  55. Where("id = ?", cow.Id).
  56. Updates(cow).Error; err != nil {
  57. zaplog.Error("CowDiseaseCreate", zap.Any("EventPenUpdate", err))
  58. }
  59. }
  60. }()
  61. // PC端直接跳过诊断过程
  62. if source == model.SourcePC {
  63. newEventCowDisease.DiagnosedResult = pasturePb.IsShow_Ok
  64. newEventCowDisease.DiagnoseOperationId = int32(operationUser.Id)
  65. newEventCowDisease.DiagnoseOperationName = operationUser.Name
  66. newEventCowDisease.Source = model.SourcePC
  67. newEventCowDisease.DiagnoseId = req.DiseaseId
  68. newEventCowDisease.DiagnoseName = disease.Name
  69. newEventCowDisease.HealthStatus = pasturePb.HealthStatus_Disease
  70. }
  71. if req.PrescriptionId > 0 || len(req.PrescriptionDetail) > 0 {
  72. newEventCowDisease.HealthStatus = pasturePb.HealthStatus_Treatment
  73. newEventCowDisease.DiagnosedResult = pasturePb.IsShow_Ok
  74. newEventCowDisease.FirstTreatmentAt = int64(req.DiseaseAt)
  75. newEventCowDisease.LastTreatmentAt = int64(req.DiseaseAt)
  76. newEventCowDisease.DiagnoseId = int32(disease.Id)
  77. newEventCowDisease.DiagnoseName = disease.Name
  78. }
  79. newEventCowTreatment := &model.EventCowTreatment{}
  80. newCowTreatmentRequest := &pasturePb.CowTreatmentRequest{}
  81. var isCreatePrescription bool
  82. diseaseTypeMap := s.DiseaseTypeMap()
  83. prescription := &model.Prescription{}
  84. if req.PrescriptionId > 0 || len(req.PrescriptionDetail) > 0 {
  85. isCreatePrescription = true
  86. if req.PrescriptionId > 0 {
  87. prescription, err = s.GetPrescriptionById(ctx, userModel.AppPasture.Id, req.PrescriptionId)
  88. if err != nil {
  89. return xerr.WithStack(err)
  90. }
  91. prescriptionDrugs, err := s.PrescriptionDrugsByPrescriptionId(ctx, userModel.AppPasture.Id, prescription.Id)
  92. if err != nil {
  93. return xerr.WithStack(err)
  94. }
  95. req.PrescriptionDetail = model.PrescriptionDrugsSlice(prescriptionDrugs).ToPB()
  96. }
  97. }
  98. if err = s.DB.Transaction(func(tx *gorm.DB) error {
  99. lastPrescriptionName := ""
  100. // 已有的处方使用次数+1
  101. if req.PrescriptionId > 0 {
  102. prescription.EventUseCountUpdate()
  103. if err = tx.Model(new(model.Prescription)).
  104. Select("use_count").
  105. Where("id = ?", prescription.Id).
  106. Updates(prescription).Error; err != nil {
  107. return xerr.WithStack(err)
  108. }
  109. lastPrescriptionName = prescription.Name
  110. }
  111. // 新的临时处方
  112. if req.PrescriptionId <= 0 && len(req.PrescriptionDetail) > 0 {
  113. newPrescriptionRequest := &pasturePb.PrescriptionRequest{
  114. Name: fmt.Sprintf("%s-%s-%s", disease.Name, time.Now().Format("20060102"), operationUser.Name),
  115. ApplicableDiseaseIds: []int32{req.DiseaseId},
  116. IsShow: pasturePb.IsShow_Ok,
  117. }
  118. newPrescription := model.NewPrescription(
  119. userModel.AppPasture.Id,
  120. newPrescriptionRequest,
  121. fmt.Sprintf("%d", disease.Id),
  122. 1,
  123. 0,
  124. 0,
  125. userModel.SystemUser,
  126. )
  127. newPrescription.UseCount += 1
  128. if err = tx.Model(new(model.Prescription)).Create(newPrescription).Error; err != nil {
  129. return xerr.WithStack(err)
  130. }
  131. prescription = newPrescription
  132. lastPrescriptionName = newPrescription.Name
  133. newPrescriptionDrugs := model.NewPrescriptionDrugs(userModel.AppPasture.Id, prescription.Id, req.PrescriptionDetail)
  134. if err = tx.Model(new(model.PrescriptionDrugs)).Create(newPrescriptionDrugs).Error; err != nil {
  135. return xerr.WithStack(err)
  136. }
  137. // 记录事件日志
  138. cowLogs := s.SubmitEventLog(ctx, userModel.AppPasture.Id, cow, pasturePb.EventType_Disease, newEventCowDisease)
  139. if err = tx.Table(cowLogs.TableName()).Create(cowLogs).Error; err != nil {
  140. return xerr.WithStack(err)
  141. }
  142. }
  143. // 创建 CowDisease
  144. newEventCowDisease.LastPrescriptionName = lastPrescriptionName
  145. if err = tx.Model(new(model.EventCowDisease)).Create(newEventCowDisease).Error; err != nil {
  146. return xerr.WithStack(err)
  147. }
  148. // 创建治疗记录
  149. if isCreatePrescription {
  150. newCowTreatmentRequest = &pasturePb.CowTreatmentRequest{
  151. CowId: req.CowId,
  152. PrescriptionId: prescription.Id,
  153. DiseaseId: req.DiseaseId,
  154. DiseaseName: disease.Name,
  155. DiseaseType: disease.DiseaseType,
  156. PrescriptionDetail: req.PrescriptionDetail,
  157. TreatmentResult: pasturePb.TreatmentResult_GoOn,
  158. Remarks: req.Remarks,
  159. TreatmentAt: req.DiseaseAt,
  160. }
  161. newEventCowTreatment = model.NewEventCowTreatment(userModel.AppPasture.Id, prescription, newCowTreatmentRequest, diseaseTypeMap, operationUser, userModel.SystemUser)
  162. newEventCowTreatment.CowDiseaseId = newEventCowDisease.Id
  163. // 创建治疗记录
  164. if err = tx.Model(new(model.EventCowTreatment)).Create(newEventCowTreatment).Error; err != nil {
  165. return xerr.WithStack(err)
  166. }
  167. }
  168. return nil
  169. }); err != nil {
  170. return xerr.WithStack(err)
  171. }
  172. return nil
  173. }
  174. // CowDiseaseList 发病牛只清单
  175. func (s *StoreEntry) CowDiseaseList(ctx context.Context, req *pasturePb.SearchEventCowTreatmentRequest, pagination *pasturePb.PaginationModel) (*pasturePb.EventCowDiseaseResponse, error) {
  176. userModel, err := s.GetUserModel(ctx)
  177. if err != nil {
  178. return nil, xerr.WithStack(err)
  179. }
  180. cowDiseaseList := make([]*model.EventCowDisease, 0)
  181. var count int64 = 0
  182. pref := s.DB.Table(fmt.Sprintf("%s as a", new(model.EventCowDisease).TableName())).
  183. Joins(fmt.Sprintf("JOIN %s AS b on a.cow_id = b.id", new(model.Cow).TableName())).
  184. Select("a.*,b.pen_name").
  185. Where("a.pasture_id = ?", userModel.AppPasture.Id).
  186. Where("a.health_status != ?", pasturePb.HealthStatus_Curable)
  187. if len(req.CowIds) > 0 {
  188. pref.Where("a.cow_id IN ?", req.CowIds)
  189. }
  190. if req.EarNumber != "" {
  191. pref.Where("a.ear_number = ?", req.EarNumber)
  192. }
  193. if req.DiseaseId > 0 {
  194. pref.Where("a.disease_id = ?", req.DiseaseId)
  195. }
  196. if req.PenId > 0 {
  197. pref.Where("b.pen_id = ?", req.PenId)
  198. }
  199. if req.HealthStatus > 0 {
  200. pref.Where("a.health_status = ?", req.HealthStatus)
  201. }
  202. if req.DiseasedStartAt > 0 && req.DiseaseEndAt > 0 && req.DiseaseEndAt >= req.DiseasedStartAt {
  203. pref.Where("a.disease_at BETWEEN ? AND ?", req.DiseasedStartAt, req.DiseaseEndAt)
  204. }
  205. if err = pref.Order("a.id DESC").
  206. Count(&count).Limit(int(pagination.PageSize)).
  207. Offset(int(pagination.PageOffset)).
  208. Find(&cowDiseaseList).Error; err != nil {
  209. return nil, xerr.WithStack(err)
  210. }
  211. healthStatusMap := s.HealthStatusMap()
  212. return &pasturePb.EventCowDiseaseResponse{
  213. Code: http.StatusOK,
  214. Msg: "ok",
  215. Data: &pasturePb.EventCowDiseaseData{
  216. List: model.EventCowDiseaseSlice(cowDiseaseList).ToPB(healthStatusMap),
  217. Total: int32(count),
  218. PageSize: pagination.PageSize,
  219. Page: pagination.Page,
  220. },
  221. }, nil
  222. }
  223. // CowDiseaseDiagnose 发病牛只诊断
  224. func (s *StoreEntry) CowDiseaseDiagnose(ctx context.Context, req *pasturePb.CowDiagnosedRequest) error {
  225. userModel, err := s.GetUserModel(ctx)
  226. if err != nil {
  227. return xerr.WithStack(err)
  228. }
  229. cow, err := s.GetCowInfoByCowId(ctx, userModel.AppPasture.Id, int64(req.CowId))
  230. if err != nil {
  231. return xerr.Customf("错误的牛只信息: %d", req.CowId)
  232. }
  233. eventCowDisease := &model.EventCowDisease{}
  234. if err = s.DB.Model(new(model.EventCowDisease)).
  235. Where("cow_id = ?", req.CowId).
  236. Where("id = ?", req.Id).
  237. Where("health_status = ?", pasturePb.HealthStatus_Health).
  238. Where("pasture_id = ?", userModel.AppPasture.Id).
  239. First(eventCowDisease).Error; err != nil {
  240. zaplog.Error("CowDiseaseDiagnose", zap.Any("req", req), zap.Any("userModel", userModel))
  241. return xerr.Custom("异常牛只数据")
  242. }
  243. if eventCowDisease == nil || eventCowDisease.Id <= 0 {
  244. return xerr.Custom("异常牛只数据")
  245. }
  246. if req.DiagnosedResult == pasturePb.IsShow_No {
  247. // 未发病更新
  248. eventCowDisease.EventUnDiseaseUpdate(userModel.SystemUser, req.Remarks)
  249. if err = s.DB.Model(eventCowDisease).
  250. Select("diagnosed_result", "diagnose_operation_id", "diagnose_operation_name", "remarks").
  251. Where("id = ?", req.Id).
  252. Updates(eventCowDisease).Error; err != nil {
  253. return xerr.WithStack(err)
  254. }
  255. return nil
  256. }
  257. // 已发病
  258. disease, err := s.GetDiseaseById(ctx, userModel.AppPasture.Id, req.DiseaseId)
  259. if err != nil {
  260. return xerr.WithStack(err)
  261. }
  262. systemUser, err := s.GetSystemUserById(ctx, int64(req.OperationId))
  263. if err != nil {
  264. return xerr.WithStack(err)
  265. }
  266. if err = s.DB.Transaction(func(tx *gorm.DB) error {
  267. eventCowDisease.EventDiseaseUpdate(disease, systemUser, req.Temperature)
  268. if err = tx.Model(eventCowDisease).
  269. Select("health_status", "diagnosed_result", "diagnose_id", "diagnose_name", "temperature", "diagnose_operation_id", "diagnose_operation_name", "diagnosed_at").
  270. Where("id = ?", req.Id).
  271. Where("cow_id = ?", req.CowId).
  272. Updates(eventCowDisease).Error; err != nil {
  273. return xerr.WithStack(err)
  274. }
  275. cow.EventHealthStatusUpdate(pasturePb.HealthStatus_Disease)
  276. if err = tx.Model(cow).
  277. Select("health_status").
  278. Where("id = ?", cow.Id).
  279. Updates(cow).Error; err != nil {
  280. return xerr.WithStack(err)
  281. }
  282. // 记录事件日志
  283. cowLogs := s.SubmitEventLog(ctx, userModel.AppPasture.Id, cow, pasturePb.EventType_Disease, eventCowDisease)
  284. if err = tx.Table(cowLogs.TableName()).Create(cowLogs).Error; err != nil {
  285. return xerr.WithStack(err)
  286. }
  287. return nil
  288. }); err != nil {
  289. return xerr.WithStack(err)
  290. }
  291. return nil
  292. }
  293. // CowDiseaseTreatment 发病牛只治疗
  294. func (s *StoreEntry) CowDiseaseTreatment(ctx context.Context, req *pasturePb.CowTreatmentRequest) error {
  295. userModel, err := s.GetUserModel(ctx)
  296. if err != nil {
  297. return xerr.WithStack(err)
  298. }
  299. // 牛只疾病信息
  300. eventCowDisease := &model.EventCowDisease{}
  301. if err = s.DB.Where("cow_id = ?", req.CowId).
  302. Where("pasture_id = ?", userModel.AppPasture.Id).
  303. Where("id = ?", req.Id).
  304. First(eventCowDisease).Error; err != nil {
  305. return xerr.WithStack(err)
  306. }
  307. if eventCowDisease.HealthStatus != pasturePb.HealthStatus_Disease &&
  308. eventCowDisease.HealthStatus != pasturePb.HealthStatus_Treatment {
  309. return xerr.Custom("异常牛只数据")
  310. }
  311. cow, err := s.GetCowInfoByCowId(ctx, userModel.AppPasture.Id, int64(req.CowId))
  312. if err != nil {
  313. return xerr.WithStack(err)
  314. }
  315. // 操作人信息
  316. operationUser, err := s.GetSystemUserById(ctx, int64(req.OperationId))
  317. if err != nil {
  318. return xerr.Customf("操作人数据异常: %d", req.OperationId)
  319. }
  320. // 处方信息
  321. prescription, err := s.GetPrescriptionById(ctx, userModel.AppPasture.Id, req.PrescriptionId)
  322. if err != nil {
  323. return xerr.WithStack(err)
  324. }
  325. // 疾病信息
  326. disease, err := s.GetDiseaseById(ctx, userModel.AppPasture.Id, eventCowDisease.DiseaseId)
  327. if err != nil {
  328. return xerr.WithStack(err)
  329. }
  330. req.DiseaseName = disease.Name
  331. req.DiseaseType = disease.DiseaseType
  332. // 处方信息
  333. prescriptionDrugs, err := s.PrescriptionDrugsByPrescriptionId(ctx, userModel.AppPasture.Id, prescription.Id)
  334. if err != nil {
  335. return xerr.WithStack(err)
  336. }
  337. // 处方详情
  338. unitMap := s.UnitMap()
  339. prescriptionDetail := make([]*pasturePb.PrescriptionDrugsList, len(prescriptionDrugs))
  340. for i, v := range prescriptionDrugs {
  341. prescriptionDetail[i] = &pasturePb.PrescriptionDrugsList{
  342. DrugsId: int32(v.DrugsId),
  343. DrugsName: v.DrugsName,
  344. Unit: v.Unit,
  345. UnitName: unitMap[v.Unit],
  346. Dosages: v.Dosages,
  347. }
  348. }
  349. req.PrescriptionDetail = prescriptionDetail
  350. healthStatus := pasturePb.HealthStatus_Treatment
  351. if req.TreatmentResult == pasturePb.TreatmentResult_Curable {
  352. healthStatus = pasturePb.HealthStatus_Curable
  353. }
  354. if req.TreatmentResult == pasturePb.TreatmentResult_Out {
  355. healthStatus = pasturePb.HealthStatus_Out
  356. }
  357. if req.TreatmentResult == pasturePb.TreatmentResult_Dead {
  358. healthStatus = pasturePb.HealthStatus_Dead
  359. }
  360. diseaseTypeMap := s.DiseaseTypeMap()
  361. newEventCowTreatment := model.NewEventCowTreatment(userModel.AppPasture.Id, prescription, req, diseaseTypeMap, operationUser, userModel.SystemUser)
  362. if err = s.DB.Transaction(func(tx *gorm.DB) error {
  363. if err = tx.Create(newEventCowTreatment).Error; err != nil {
  364. return xerr.WithStack(err)
  365. }
  366. eventCowDisease.EventTreatmentUpdate(healthStatus, int64(req.TreatmentAt), prescription.Name)
  367. if err = tx.Model(eventCowDisease).
  368. Select("health_status", "first_treatment_at", "last_treatment_at", "last_prescription_name").
  369. Where("id = ?", req.Id).
  370. Where("cow_id = ?", req.CowId).
  371. Updates(eventCowDisease).Error; err != nil {
  372. return xerr.WithStack(err)
  373. }
  374. cow.EventHealthStatusUpdate(healthStatus)
  375. if err = tx.Model(new(model.Cow)).
  376. Select("health_status").
  377. Where("id = ?", req.CowId).
  378. Updates(cow).Error; err != nil {
  379. return xerr.WithStack(err)
  380. }
  381. prescription.EventUseCountUpdate()
  382. if err = tx.Model(prescription).
  383. Select("use_count").
  384. Where("id = ?", prescription.Id).
  385. Updates(prescription).Error; err != nil {
  386. return xerr.WithStack(err)
  387. }
  388. return nil
  389. }); err != nil {
  390. return xerr.WithStack(err)
  391. }
  392. return nil
  393. }
  394. func (s *StoreEntry) DiseaseSuggestPrescription(ctx context.Context, diseaseId int64) (*pasturePb.ConfigOptionsListResponse, error) {
  395. userModel, err := s.GetUserModel(ctx)
  396. if err != nil {
  397. return nil, xerr.WithStack(err)
  398. }
  399. res := make([]*pasturePb.ConfigOptionsList, 0)
  400. prescriptionList := make([]*model.Prescription, 0)
  401. if err = s.DB.Model(new(model.Prescription)).
  402. Where("is_show = ?", pasturePb.IsShow_Ok).
  403. Where("pasture_id = ?", userModel.AppPasture.Id).
  404. Find(&prescriptionList).Error; err != nil {
  405. return nil, xerr.WithStack(err)
  406. }
  407. for _, v := range prescriptionList {
  408. disabled := false
  409. if strings.Contains(v.ApplicableDisease, fmt.Sprintf("%d", diseaseId)) {
  410. disabled = true
  411. }
  412. res = append(res, &pasturePb.ConfigOptionsList{
  413. Value: v.Id,
  414. Label: v.Name,
  415. Disabled: disabled,
  416. })
  417. }
  418. return &pasturePb.ConfigOptionsListResponse{
  419. Code: http.StatusOK,
  420. Msg: "ok",
  421. Data: res,
  422. }, nil
  423. }
  424. // CowDiseaseTreatmentDetail 发病牛只治疗详情列表
  425. func (s *StoreEntry) CowDiseaseTreatmentDetail(ctx context.Context, req *pasturePb.EventCowTreatmentDetailRequest, pagination *pasturePb.PaginationModel) (*pasturePb.EventCowTreatmentDetailResponse, error) {
  426. userModel, err := s.GetUserModel(ctx)
  427. if err != nil {
  428. return nil, xerr.WithStack(err)
  429. }
  430. eventCowDisease := &model.EventCowDisease{}
  431. var count int64 = 0
  432. pref := s.DB.Model(new(model.EventCowDisease)).
  433. Where("cow_id = ?", req.CowId).
  434. Where("pasture_id = ?", userModel.AppPasture.Id).
  435. Where("id = ?", req.Id)
  436. if req.DiseaseId > 0 {
  437. pref.Where("disease_id = ?", req.DiseaseId)
  438. }
  439. if req.DiseaseStartAt > 0 && req.DiseaseEndAt > 0 && req.DiseaseStartAt <= req.DiseaseEndAt {
  440. pref.Where("disease_at BETWEEN ? AND ?", req.DiseaseStartAt, req.DiseaseEndAt)
  441. }
  442. if err = pref.Count(&count).
  443. Limit(int(pagination.PageSize)).
  444. Offset(int(pagination.PageOffset)).
  445. Order("id desc").
  446. First(&eventCowDisease).Error; err != nil {
  447. return nil, xerr.WithStack(err)
  448. }
  449. eventCowTreatmentList := make([]*model.EventCowTreatment, 0)
  450. if err = s.DB.Model(new(model.EventCowTreatment)).
  451. Where("cow_disease_id = ?", req.Id).
  452. Where("cow_id = ?", req.CowId).
  453. Order("id desc").
  454. Find(&eventCowTreatmentList).Error; err != nil {
  455. return nil, xerr.WithStack(err)
  456. }
  457. return &pasturePb.EventCowTreatmentDetailResponse{
  458. Code: http.StatusOK,
  459. Msg: "ok",
  460. Data: &pasturePb.EventCowTreatmentDetail{
  461. List: model.EventCowTreatmentSlice(eventCowTreatmentList).ToPB(eventCowDisease),
  462. Total: int32(count),
  463. PageSize: pagination.Page,
  464. Page: pagination.PageSize,
  465. },
  466. }, nil
  467. }
  468. func (s *StoreEntry) CowDiseaseCurable(ctx context.Context, req *pasturePb.EventCowCurableRequest) error {
  469. userModel, err := s.GetUserModel(ctx)
  470. if err != nil {
  471. return xerr.WithStack(err)
  472. }
  473. eventCowDiseaseList := make([]*model.EventCowDisease, 0)
  474. if err = s.DB.Where("id IN ?", req.Ids).
  475. Where("health_status = ?", pasturePb.HealthStatus_Treatment).
  476. Where("pasture_id = ?", userModel.AppPasture.Id).
  477. Find(&eventCowDiseaseList).Error; err != nil {
  478. zaplog.Error("GetEventCowDiseaseList", zap.Any("err", err), zap.Any("req", req))
  479. return xerr.Custom("异常数据")
  480. }
  481. if len(eventCowDiseaseList) == 0 {
  482. return nil
  483. }
  484. operationUser, err := s.GetSystemUserById(ctx, int64(req.OperationId))
  485. if err != nil {
  486. return xerr.Customf("该用户不存在: %d", req.OperationId)
  487. }
  488. eventCowTreatmentList := make([]*model.EventCowTreatment, 0)
  489. for _, v := range eventCowDiseaseList {
  490. newEventCowTreatment := model.NewEventCowCurableTreatment(userModel.AppPasture.Id, userModel.SystemUser, operationUser, v, req.Remarks, int64(req.CurableAt))
  491. eventCowTreatmentList = append(eventCowTreatmentList, newEventCowTreatment)
  492. }
  493. if len(eventCowTreatmentList) <= 0 {
  494. return nil
  495. }
  496. if err = s.DB.Transaction(func(tx *gorm.DB) error {
  497. for _, eventCowDisease := range eventCowDiseaseList {
  498. eventCowDisease.EventCurableUpdate(int64(req.CurableAt))
  499. if err = tx.Model(eventCowDisease).
  500. Select("health_status", "diagnosed_result", "curable_at").
  501. Where("id = ?", eventCowDisease.Id).
  502. Where("health_status = ?", pasturePb.HealthStatus_Treatment).
  503. Updates(eventCowDisease).Error; err != nil {
  504. return xerr.WithStack(err)
  505. }
  506. cow, err := s.GetCowInfoByCowId(ctx, userModel.AppPasture.Id, eventCowDisease.CowId)
  507. if err != nil {
  508. return xerr.WithStack(err)
  509. }
  510. // 更新牛只健康状态
  511. cow.EventHealthStatusUpdate(pasturePb.HealthStatus_Curable)
  512. if err = tx.Model(cow).
  513. Select("health_status").
  514. Where("id = ?", eventCowDisease.CowId).
  515. Updates(cow).Error; err != nil {
  516. return xerr.WithStack(err)
  517. }
  518. // 更新发病事件日志
  519. cowLogs := &model.EventCowLog{CowId: cow.Id}
  520. curableAtFormat := ""
  521. if eventCowDisease.CurableAt > 0 {
  522. curableAtFormat = time.Unix(eventCowDisease.CurableAt, 0).Format(model.LayoutDate2)
  523. }
  524. curableAtParse := time.Unix(eventCowDisease.CurableAt, 0)
  525. diseaseAtParse := time.Unix(eventCowDisease.DiseaseAt, 0)
  526. curableDays := int64(curableAtParse.Sub(diseaseAtParse).Hours() / 24)
  527. if err = tx.Table(cowLogs.TableName()).
  528. Where("event_type = ?", pasturePb.EventType_Disease).
  529. Where("event_at = ?", eventCowDisease.DiseaseAt).
  530. Update("event_description", fmt.Sprintf("疾病名称: %s; 状态: 已治愈; 治愈时间: %s; 治疗天数: %d;", eventCowDisease.DiseaseName, curableAtFormat, curableDays)).
  531. Error; err != nil {
  532. return xerr.WithStack(err)
  533. }
  534. }
  535. if err = tx.Model(new(model.EventCowTreatment)).Create(eventCowTreatmentList).Error; err != nil {
  536. return xerr.WithStack(err)
  537. }
  538. return nil
  539. }); err != nil {
  540. return xerr.WithStack(err)
  541. }
  542. return nil
  543. }