Prechádzať zdrojové kódy

Merge branch 'mj' into pre

zwhui 2 rokov pred
rodič
commit
c56697843e

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

@@ -2,6 +2,7 @@ package com.cyksj.web.controller.mirror;
 
 import cn.hutool.core.date.DateUtil;
 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;
@@ -64,6 +65,8 @@ public class MirrorController {
 
     private final MidjourneyAccountService midjourneyAccountService;
 
+    private final EnvCommonService envCommonService;
+
     @Value("${midjourney.url}")
     private String midjourneyHost;
 
@@ -313,7 +316,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}")