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