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