From 02a84fb6fb2a39bfe7fc5cf6c0137bbf231b17fe Mon Sep 17 00:00:00 2001 From: zhangqian <zhangqian@123.com> Date: 星期四, 09 十一月 2023 14:14:17 +0800 Subject: [PATCH] 设备列表返回是否设置工艺参数字段 --- service/device.go | 15 +++++++++++++-- 1 files changed, 13 insertions(+), 2 deletions(-) diff --git a/service/device.go b/service/device.go index e11b959..f09559e 100644 --- a/service/device.go +++ b/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) +} -- Gitblit v1.8.0