valve.go 2.0 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273
  1. package handler
  2. import (
  3. "GtDataStore/app/cmd/organization/internal/logic/job"
  4. "GtDataStore/app/cmd/organization/internal/svc"
  5. "GtDataStore/app/model"
  6. "context"
  7. "github.com/zeromicro/go-zero/core/logx"
  8. "sync"
  9. "time"
  10. )
  11. func DeviceValve(ctx context.Context, task *job.Task, technologyName string) error {
  12. // 1. 查询所有的设备
  13. devices, err := findDeviceValve(task.Job.SvcCtx, int64(task.Id))
  14. if err != nil {
  15. return err
  16. }
  17. // 2. 获得点位信息
  18. var wg sync.WaitGroup
  19. wg.Add(len(devices))
  20. for _, device := range devices {
  21. device := device
  22. go func() {
  23. defer wg.Done()
  24. if err := device.Items.FillCurrentValue(); err != nil {
  25. logx.Errorf("DeviceValve device.Items.FillCurrentValue error: %s", err.Error())
  26. }
  27. }()
  28. }
  29. wg.Wait()
  30. // 3. 转换为存储对象
  31. workings, err := transDeviceValveData(devices)
  32. if err != nil {
  33. return err
  34. }
  35. // 4. 批量入库
  36. if _, err := task.Job.SvcCtx.WorkingValve.MultiInsert(context.Background(), workings); err != nil {
  37. logx.Errorf("DeviceValve task.Job.SvcCtx.WorkingValve.MultiInsert error: %s", err.Error())
  38. }
  39. return nil
  40. }
  41. func findDeviceValve(svcCtx *svc.ServiceContext, projectId int64) ([]model.DcDeviceBind, error) {
  42. if devices, err := svcCtx.DeviceBind.FindByProjectIdDeviceType(context.Background(), projectId, DEVICE_VALVE); err != nil {
  43. logx.Infof("findDeviceValve not found devices")
  44. return nil, err
  45. } else {
  46. return devices, nil
  47. }
  48. }
  49. func transDeviceValveData(datas []model.DcDeviceBind) ([]model.DcWorkingValve, error) {
  50. ts := make([]model.DcWorkingValve, len(datas))
  51. for i, data := range datas {
  52. ts[i] = model.DcWorkingValve{
  53. ProjectId: data.ProjectId,
  54. DeviceCode: data.DeviceCode,
  55. Adjust: data.Items.GetItemInt64Value("adjust"),
  56. Opening: data.Items.GetItemFloat64Value("opening"),
  57. Closed: data.Items.GetItemInt64Value("closed"),
  58. Opened: data.Items.GetItemInt64Value("opened"),
  59. FaultStatus: data.Items.GetItemInt64Value("fault_status"),
  60. CTime: time.Now(),
  61. }
  62. }
  63. return ts, nil
  64. }