event_health.go 22 KB

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