|
|
@@ -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}")
|