From cacf363f222e354cad6db18c4a00fb1097d35c2c Mon Sep 17 00:00:00 2001
From: zhangqian <zhangqian@123.com>
Date: 星期四, 09 十一月 2023 14:30:58 +0800
Subject: [PATCH] fix
---
service/device.go | 26 ++++++++++++++++++++++++--
1 files changed, 24 insertions(+), 2 deletions(-)
diff --git a/service/device.go b/service/device.go
index e11b959..e0ff8ab 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,23 @@
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).UpdateByMap(map[string]interface{}{"need_set_process_params": needSetProcessParams})
+}
+
+func GetCurrentDevice() (device *model.Device, err error) {
+ if conf.Conf.CurrentDeviceID == "" {
+ return nil, errors.New("褰撳墠璁惧ID涓嶅瓨鍦紝璇锋鏌�")
+ }
+ device, err = model.NewDeviceSearch().SetDeviceId(conf.Conf.CurrentDeviceID).First()
+ if err == gorm.ErrRecordNotFound {
+ return nil, errors.New("璁惧涓嶅瓨鍦�")
+ }
+ return device, nil
+}
--
Gitblit v1.8.0