From bc7ee359dfe4e66a05c2cca9deb7a945534009f3 Mon Sep 17 00:00:00 2001
From: zhangqian <zhangqian@123.com>
Date: 星期五, 20 十月 2023 18:30:26 +0800
Subject: [PATCH] 删除不再用的topic,增加初始化的时候,设备拉取

---
 model/device.go |   18 ++++++++++++++----
 1 files changed, 14 insertions(+), 4 deletions(-)

diff --git a/model/device.go b/model/device.go
index 46e2665..59c7859 100644
--- a/model/device.go
+++ b/model/device.go
@@ -19,10 +19,11 @@
 
 	DeviceSearch struct {
 		Device
-		Order    string
-		PageNum  int
-		PageSize int
-		Orm      *gorm.DB
+		Order     string
+		PageNum   int
+		PageSize  int
+		Orm       *gorm.DB
+		DeviceIDs []string
 	}
 )
 
@@ -64,6 +65,11 @@
 	return slf
 }
 
+func (slf *DeviceSearch) SetDeviceIds(deviceIds []string) *DeviceSearch {
+	slf.DeviceIDs = deviceIds
+	return slf
+}
+
 func (slf *DeviceSearch) build() *gorm.DB {
 	var db = slf.Orm.Table(slf.TableName())
 
@@ -75,6 +81,10 @@
 		db = db.Where("device_id = ?", slf.DeviceID)
 	}
 
+	if len(slf.DeviceIDs) != 0 {
+		db = db.Where("device_id in (?)", slf.DeviceIDs)
+	}
+
 	if slf.Order != "" {
 		db = db.Order(slf.Order)
 	}

--
Gitblit v1.8.0