zwhui 2 년 전
부모
커밋
f2cad3a433
1개의 변경된 파일4개의 추가작업 그리고 1개의 파일을 삭제
  1. 4 1
      netflix-web/src/main/java/com/cyksj/web/controller/mirror/MirrorController.java

+ 4 - 1
netflix-web/src/main/java/com/cyksj/web/controller/mirror/MirrorController.java

@@ -1,6 +1,7 @@
 package com.cyksj.web.controller.mirror;
 
 import cn.hutool.json.JSONObject;
+import com.cyksj.common.EnvCommonService;
 import com.cyksj.common.exception.BusinessRuntimeException;
 import com.cyksj.common.task.GlobalThreadPoolTaskExecutor;
 import com.cyksj.common.util.IoKit;
@@ -59,6 +60,8 @@ public class MirrorController {
 
     private final MidjourneyAccountService midjourneyAccountService;
 
+    private final EnvCommonService envCommonService;
+
     @Value("${midjourney.url}")
     private String midjourneyHost;
 
@@ -309,7 +312,7 @@ public class MirrorController {
     public void midjourneyMirror(@PathVariable Long relationId,HttpServletResponse response) throws IOException {
         Long userId = StpUserUtil.getLoginIdAsLong();
         MidjourneyUser midjourneyUser = midjourneyAccountService.getMidjourneyUserToken(userId, relationId);
-        response.sendRedirect( midjourneyHost + "/8081/api/applets/mirror/midjourneyMirrorWithToken/" + midjourneyUser.getUserToken());
+        response.sendRedirect( midjourneyHost + (envCommonService.isPreEnv() ? "8082":"8081") +"/api/applets/mirror/midjourneyMirrorWithToken/" + midjourneyUser.getUserToken());
     }
 
     @GetMapping("/midjourneyMirrorWithToken/{userToken}")