From 09aad454d6374af45eb851584a14120195105ff6 Mon Sep 17 00:00:00 2001
From: azw <vcl_delphi@126.com>
Date: Mon, 24 Apr 2023 08:31:16 +0000
Subject: [PATCH] 1

---
 server/wsproxy/app.cc | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/server/wsproxy/app.cc b/server/wsproxy/app.cc
index a72c214..dea8306 100644
--- a/server/wsproxy/app.cc
+++ b/server/wsproxy/app.cc
@@ -307,6 +307,20 @@ bool App::HasTask()
             return true;
         }
     }
+    {
+        if (!udp_work_node_) {
+            udp_msg_mutex_->lock();
+            if (!udp_work_node_ && udp_top_node_) {
+                udp_work_node_ = udp_top_node_;
+                udp_top_node_ = nullptr;
+                udp_bot_node_ = nullptr;
+            }
+            udp_msg_mutex_->unlock();
+        }
+        if (udp_work_node_) {
+            return true;
+        }
+    }
     return false;
 }