From d7ba01be2d1c89e76b7e41d7a70cc7dd34188776 Mon Sep 17 00:00:00 2001
From: chuzhongzai <chuzhongzai@hotmail.com>
Date: Sat, 22 Mar 2025 23:47:21 +0800
Subject: [PATCH] =?UTF-8?q?=E5=AD=90=E8=8A=82=E7=82=B9=E6=81=A2=E5=A4=8D?=
 =?UTF-8?q?=E6=97=B6=E5=8F=91=E9=80=81=E6=9C=AA=E5=AE=8C=E6=88=90=E4=BB=BB?=
 =?UTF-8?q?=E5=8A=A1?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../lion/lionwebsite/Service/RemoteService.java    | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/src/main/java/com/lion/lionwebsite/Service/RemoteService.java b/src/main/java/com/lion/lionwebsite/Service/RemoteService.java
index 5748dbb..c22b86b 100644
--- a/src/main/java/com/lion/lionwebsite/Service/RemoteService.java
+++ b/src/main/java/com/lion/lionwebsite/Service/RemoteService.java
@@ -87,6 +87,8 @@ public class RemoteService {
             if(pushService != null)
                 pushService.storageNodeOnline();
 
+            //子节点上线时，发送未完成的任务
+            resetUndone();
             channel = channelFuture.channel();
             channel.writeAndFlush(new IdentityMessage("lionwebsite"));
             return true;
@@ -101,6 +103,18 @@ public class RemoteService {
         return channelFuture.channel() == null || !channelFuture.channel().isActive();
     }
 
+    public void resetUndone(){
+        if (channelFuture.channel() == null || !channelFuture.channel().isActive())
+            return;
+
+        Gallery[] galleries = galleryMapper.selectUnDoneGalleries();
+        if(galleries != null && galleries.length != 0) {
+            log.info("发送未下载完成图片至节点,{}本", galleries.length);
+            for (Gallery gallery : galleries)
+                addGalleryToQueue(gallery);
+        }
+    }
+
     public byte addGalleryToQueue(Gallery gallery){
         GalleryTask galleryTask = new GalleryTask();
         galleryTask.setGid(gallery.getGid());