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 |   30 ++++++++++++++++++++++++++++++
 1 files changed, 30 insertions(+), 0 deletions(-)

diff --git a/service/device.go b/service/device.go
index 8ad1d23..f09559e 100644
--- a/service/device.go
+++ b/service/device.go
@@ -3,7 +3,9 @@
 import (
 	"apsClient/conf"
 	"apsClient/model"
+	"apsClient/model/response"
 	"apsClient/pkg/logx"
+	"errors"
 	"github.com/jinzhu/gorm"
 	"os"
 	"strings"
@@ -22,6 +24,25 @@
 		deviceIds = append(deviceIds, device.DeviceID)
 	}
 	return deviceIds, nil
+}
+
+func GetDeviceList() (deviceList []*response.Device, err error) {
+	devices, err := model.NewDeviceSearch().SetDeviceMac(conf.Conf.System.DeviceId).FindNotTotal()
+	if err == gorm.ErrRecordNotFound {
+		return nil, nil
+	}
+	if err != nil {
+		return nil, err
+	}
+	deviceList = make([]*response.Device, 0, len(devices))
+	for _, device := range devices {
+		deviceList = append(deviceList, &response.Device{
+			DeviceID:             device.DeviceID,
+			DeviceName:           device.DeviceName,
+			NeedSetProcessParams: device.NeedSetProcessParams,
+		})
+	}
+	return deviceList, nil
 }
 
 func InitCurrentDeviceID() (err error) {
@@ -65,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