设备列表返回是否设置工艺参数字段

zhangqian

2023-11-09 02a84fb6fb2a39bfe7fc5cf6c0137bbf231b17fe

 service/device.go

@@ -5,6 +5,7 @@
   "apsClient/model"
   "apsClient/model/response"
   "apsClient/pkg/logx"
   "errors"
   "github.com/jinzhu/gorm"
   "os"
   "strings"
@@ -36,8 +37,9 @@
   deviceList = make([]*response.Device, 0, len(devices))
   for _, device := range devices {
      deviceList = append(deviceList, &response.Device{
         DeviceID:   device.DeviceID,
         DeviceName: device.DeviceName,
         DeviceID:             device.DeviceID,
         DeviceName:           device.DeviceName,
         NeedSetProcessParams: device.NeedSetProcessParams,
      })
   }
   return deviceList, nil
@@ -84,3 +86,12 @@
   deviceId = strings.Trim(deviceId, "\n")
   return deviceId
}

func UpdateDevice(deviceId string, needSetProcessParams bool) (err error) {
   device, err := model.NewDeviceSearch().SetDeviceId(deviceId).First()
   if err == gorm.ErrRecordNotFound {
      return errors.New("设备不存在")
   }
   device.NeedSetProcessParams = needSetProcessParams
   return model.NewDeviceSearch().SetID(device.ID).Save(device)
}