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

diff --git a/service/device.go b/service/device.go
index f09559e..e0ff8ab 100644
--- a/service/device.go
+++ b/service/device.go
@@ -93,5 +93,16 @@
 		return errors.New("璁惧涓嶅瓨鍦�")
 	}
 	device.NeedSetProcessParams = needSetProcessParams
-	return model.NewDeviceSearch().SetID(device.ID).Save(device)
+	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