From c69f721df41c9c7c1e2e9a5f167c881c04536986 Mon Sep 17 00:00:00 2001
From: zhangzengfei <zhangzengfei@smartai.com>
Date: 星期一, 03 六月 2024 19:44:30 +0800
Subject: [PATCH] 添加人员订阅

---
 service/notification.go |   41 +++++++++++++++++++++++++++++++++++++++--
 1 files changed, 39 insertions(+), 2 deletions(-)

diff --git a/service/notification.go b/service/notification.go
index 1605025..32b67a1 100644
--- a/service/notification.go
+++ b/service/notification.go
@@ -115,7 +115,44 @@
 	task.mutex.Unlock()
 
 	var req vo.RequestSubscribeNotification
-	req.SubscribeNotificationListObject.SubscribeNotificationObject[0] = notification
+	req.SubscribeNotificationListObject.SubscribeNotificationObject = append(req.SubscribeNotificationListObject.SubscribeNotificationObject, notification)
+
+	return &req
+}
+
+func (task *NotificationTask) packPersonNotificationMsg() *vo.RequestSubscribeNotification {
+	triggerTime := time.Now().Format("20060102150405")
+	if len(task.faceList) == 0 {
+		return nil
+	}
+
+	var notification = vo.PersonNotification{
+		NotificationID:   triggerTime + snowflake.GenerateIdStr(),
+		SubscribeID:      task.conf.Id,
+		Title:            task.conf.Ext.Title,
+		TriggerTime:      triggerTime,
+		ExecuteOperation: 1,
+	}
+
+	var ids []string
+	for idx, _ := range task.personList {
+		// 鎸夋捣搴风ず渚嬪~鍏呬慨鏀瑰瓧娈�
+		for i, _ := range task.personList[idx].SubImageList.SubImageInfoObject {
+			task.personList[idx].SubImageList.SubImageInfoObject[i].EventSort = 2
+		}
+
+		ids = append(ids, task.personList[idx].PersonID)
+		notification.PersonObjectList.PersonObject = append(notification.PersonObjectList.PersonObject, *task.personList[idx])
+	}
+
+	notification.InfoIDs = strings.Join(ids, ";")
+
+	task.mutex.Lock()
+	task.personList = []*vo.PersonObject{}
+	task.mutex.Unlock()
+
+	var req vo.RequestSubscribeNotification
+	req.SubscribeNotificationListObject.SubscribeNotificationObject = append(req.SubscribeNotificationListObject.SubscribeNotificationObject, notification)
 
 	return &req
 }
@@ -154,7 +191,7 @@
 	notification.InfoIDs = strings.Join(ids, ";")
 
 	var req vo.RequestSubscribeNotification
-	req.SubscribeNotificationListObject.SubscribeNotificationObject[0] = notification
+	req.SubscribeNotificationListObject.SubscribeNotificationObject = append(req.SubscribeNotificationListObject.SubscribeNotificationObject, notification)
 
 	return &req
 }

--
Gitblit v1.8.0