ソースを参照

新增全局控制是否转存(IS_SAVE)

tsurumure 1 ヶ月 前
コミット
4a9e45099a

+ 5 - 1
src/main/java/com/backendsys/modules/crt/service/impl/CrtGenerateServiceImpl.java

@@ -26,6 +26,7 @@ import com.backendsys.utils.response.PageInfoResult;
 import com.backendsys.utils.v2.PageUtils;
 import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.beans.factory.annotation.Value;
 import org.springframework.stereotype.Service;
 import reactor.core.publisher.Mono;
 
@@ -38,6 +39,9 @@ import java.util.stream.Collectors;
 @Service
 public class CrtGenerateServiceImpl implements CrtGenerateService {
 
+    @Value("${comfyui.is-save}")
+    private Boolean IS_SAVE;
+
     @Autowired
     private ComfyuiService comfyUIService;
     @Autowired
@@ -145,7 +149,7 @@ public class CrtGenerateServiceImpl implements CrtGenerateService {
         Map<String, Object> params = new LinkedHashMap<>();
         params.put("drama_project_storyboard_id", drama_project_storyboard_id);
         // - is_save: 是否转存到 cos/tos
-        comfyUISocketService.connectToSse(client_id, 8000, false, params).subscribe();
+        comfyUISocketService.connectToSse(client_id, 8000, IS_SAVE, params).subscribe();
 
         // -- [ComfyUI] 生图参数 ----------------------------------------
         ComfyuiText2Image comfyuiText2Image = new ComfyuiText2Image();

+ 2 - 1
src/main/resources/application-dev.yml

@@ -198,4 +198,5 @@ klingai:
 
 comfyui:
   host: 127.0.0.1
-  token: $2b$12$.MR4qGaFetN1FPQzbfyIrehsyjnPJ12xAZhR/l7KZpLkUPQTCG4gy
+  token: $2b$12$.MR4qGaFetN1FPQzbfyIrehsyjnPJ12xAZhR/l7KZpLkUPQTCG4gy
+  is-save: false

+ 2 - 1
src/main/resources/application-local.yml

@@ -211,4 +211,5 @@ klingai:
 
 comfyui:
   host: 43.128.1.201
-  token: $2b$12$.MR4qGaFetN1FPQzbfyIrehsyjnPJ12xAZhR/l7KZpLkUPQTCG4gy
+  token: $2b$12$.MR4qGaFetN1FPQzbfyIrehsyjnPJ12xAZhR/l7KZpLkUPQTCG4gy
+  is-save: false

+ 2 - 1
src/main/resources/application-prod.yml

@@ -199,4 +199,5 @@ klingai:
 
 comfyui:
   host: 127.0.0.1
-  token: $2b$12$.MR4qGaFetN1FPQzbfyIrehsyjnPJ12xAZhR/l7KZpLkUPQTCG4gy
+  token: $2b$12$.MR4qGaFetN1FPQzbfyIrehsyjnPJ12xAZhR/l7KZpLkUPQTCG4gy
+  is-save: false