Browse Source

fix 上传图片替换cdn域名

zwhui 2 năm trước cách đây
mục cha
commit
7b1b681f6d

+ 5 - 1
midjourney/src/main/java/com/yhlxj/service/midjourney/impl/MidjourneyServiceImpl.java

@@ -49,6 +49,7 @@ import javax.imageio.stream.ImageInputStream;
 import java.awt.image.BufferedImage;
 import java.io.File;
 import java.io.IOException;
+import java.net.MalformedURLException;
 import java.net.URL;
 import java.nio.file.Files;
 import java.nio.file.Path;
@@ -1114,7 +1115,10 @@ public class MidjourneyServiceImpl implements MidjourneyService {
         conversation.setAction("UPLOAD");
         conversation.setUserId(user.getId());
         conversation.setStatus("SUCCESS");
-        conversation.setImageUrl(jsonObject.getJSONArray("result").getStr(0));
+        String result = jsonObject.getJSONArray("result").getStr(0);
+        URL url = new URL(result);
+        url = new URL(url.getProtocol(), "cdn.mj.liuliangbang.vip", url.getPort(), url.getFile());
+        conversation.setImageUrl(url.toString());
         conversationMapper.insert(conversation);
         return conversation;
     }