Fix watch clean event

DaoCloud · Feb 11, 2025 · 0e75fad · 0e75fad
1 parent fefd765
commit 0e75fad
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/queue/controller/message.go b/queue/controller/message.go
@@ -254,7 +254,13 @@ func (mc *MessageController) Schedule(ctx context.Context, logger *slog.Logger)
 					if err != nil {
 						logger.Error("ResetToPending", "error", err)
 					} else {
-						data := MessageResponse{MessageID: item.MessageID, Content: item.Content, Priority: item.Priority, Status: model.StatusPending}
+						data := MessageResponse{
+							MessageID: item.MessageID,
+							Content:   item.Content,
+							Priority:  item.Priority,
+							Status:    model.StatusPending,
+							Data:      item.Data,
+						}
 						mc.appendWatchChannel(item.MessageID, data)
 						mc.appendWatchListChannels(data)
 					}
@@ -273,7 +279,9 @@ func (mc *MessageController) Schedule(ctx context.Context, logger *slog.Logger)
 					mc.appendWatchListChannels(MessageResponse{
 						MessageID: item.MessageID,
 						Content:   item.Content,
-						Status:    model.StatusCleanup,
+						Priority:  item.Priority,
+						Status:    model.StatusPending,
+						Data:      item.Data,
 					})
 				}
 			}