From 8324f872ef3a4d0c978a9b1d062800c6a1701c12 Mon Sep 17 00:00:00 2001
From: zhangqian <zhangqian@123.com>
Date: 星期五, 01 十二月 2023 09:58:17 +0800
Subject: [PATCH] fix

---
 service/device.go |   38 ++++++++++++++++++++++++++++++++------
 1 files changed, 32 insertions(+), 6 deletions(-)

diff --git a/service/device.go b/service/device.go
index c81500b..1906d99 100644
--- a/service/device.go
+++ b/service/device.go
@@ -2,6 +2,7 @@
 
 import (
 	"apsClient/conf"
+	"apsClient/constvar"
 	"apsClient/model"
 	"apsClient/model/response"
 	"apsClient/pkg/logx"
@@ -37,14 +38,15 @@
 	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
 }
 
-func InitCurrentDeviceID() (err error) {
+func InitCurrentDeviceID(ServerID string) (err error) {
 	currentDeviceID := ReadDeviceIDFromFile()
 	if currentDeviceID != "" {
 		conf.Conf.CurrentDeviceID = currentDeviceID
@@ -54,10 +56,12 @@
 	if err != nil {
 		return err
 	}
-	if len(deviceList) == 0 {
+	if len(deviceList) > 0 {
+		conf.Conf.CurrentDeviceID = deviceList[0]
+	} else if conf.Conf.System.DeviceId != "" {
 		conf.Conf.CurrentDeviceID = conf.Conf.System.DeviceId
 	} else {
-		conf.Conf.CurrentDeviceID = deviceList[0]
+		conf.Conf.CurrentDeviceID = ServerID
 	}
 	SetDeviceIDToFile(conf.Conf.CurrentDeviceID)
 	return nil
@@ -92,5 +96,27 @@
 		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
+}
+
+// ReportsSystemDeviceToCloud 鍒涘缓鍚屾璁惧id璁板綍
+func ReportsSystemDeviceToCloud(systemDeviceID string) {
+	err := model.NewReportsToCloudSearch(nil).Create(&model.ReportsToCloud{
+		ReportType: constvar.ReportTypeSystemDeviceID,
+		Content:    systemDeviceID,
+	})
+	if err != nil {
+		logx.Errorf("ReportsSystemDeviceToCloud create record error:%v", err)
+	}
 }

--
Gitblit v1.8.0