Browse Source

调整 ComfyUIService

tsurumure 2 months ago
parent
commit
f1091e9bb0

+ 2 - 6
src/main/java/com/backendsys/modules/sdk/comfyui/controller/ComfyUIDemoController.java

@@ -24,9 +24,7 @@ public class ComfyUIDemoController {
     private ComfyUIService comfyUIService;
     private ComfyUIService comfyUIService;
 
 
     /**
     /**
-     * 创建WebSocket连接
-     * @param clientId 客户端ID
-     * @return 操作结果
+     * [ComfyUI] 创建 WebSocket 监听连接
      */
      */
     @Anonymous
     @Anonymous
     @PostMapping("/api/comfyui/ws/connect")
     @PostMapping("/api/comfyui/ws/connect")
@@ -36,9 +34,7 @@ public class ComfyUIDemoController {
     }
     }
 
 
     /**
     /**
-     * 断开WebSocket连接
-     * @param clientId 客户端ID
-     * @return 操作结果
+     * [ComfyUI] 断开 WebSocket 监听连接
      */
      */
     @Anonymous
     @Anonymous
     @PostMapping("/api/comfyui/ws/disconnect")
     @PostMapping("/api/comfyui/ws/disconnect")

+ 2 - 2
src/main/java/com/backendsys/modules/sdk/comfyui/service/ComfyUIService.java

@@ -4,9 +4,9 @@ import reactor.core.publisher.Mono;
 
 
 public interface ComfyUIService {
 public interface ComfyUIService {
 
 
-    // 连接到第三方 WebSocket 服务
+    // [ComfyUI] 创建 WebSocket 监听连接
     Mono<Void> connect(String clientId, String wsUrl);
     Mono<Void> connect(String clientId, String wsUrl);
-    // 断开指定连接
+    // [ComfyUI] 断开 WebSocket 监听连接
     Mono<Void> disconnect(String clientId);
     Mono<Void> disconnect(String clientId);
 
 
 }
 }

+ 2 - 4
src/main/java/com/backendsys/modules/sdk/comfyui/service/impl/ComfyUIServiceImpl.java

@@ -50,7 +50,7 @@ public class ComfyUIServiceImpl implements ComfyUIService {
     }
     }
 
 
     /**
     /**
-     * 连接到第三方 WebSocket 服务
+     * [ComfyUI] 创建 WebSocket 监听连接
      * @param clientId 客户端ID(用于标识连接)
      * @param clientId 客户端ID(用于标识连接)
      * @param wsUrl WebSocket 地址
      * @param wsUrl WebSocket 地址
      * @return Mono<Void> 表示连接操作
      * @return Mono<Void> 表示连接操作
@@ -84,8 +84,6 @@ public class ComfyUIServiceImpl implements ComfyUIService {
                     .doFinally(signal -> {
                     .doFinally(signal -> {
                         System.out.println("(doFinally) Connection closed for " + clientId + ": " + signal);
                         System.out.println("(doFinally) Connection closed for " + clientId + ": " + signal);
                         sessions.remove(clientId);
                         sessions.remove(clientId);
-                        // 断开连接
-//                        disconnect(clientId);
                     });
                     });
 
 
                 // 需要返回一个Mono<Void>来表示处理完成
                 // 需要返回一个Mono<Void>来表示处理完成
@@ -96,7 +94,7 @@ public class ComfyUIServiceImpl implements ComfyUIService {
     }
     }
 
 
     /**
     /**
-     * 断开指定连接
+     * [ComfyUI] 断开 WebSocket 监听连接
      * @param clientId 客户端ID
      * @param clientId 客户端ID
      * @return Mono<Void> 表示断开操作
      * @return Mono<Void> 表示断开操作
      */
      */