Преглед на файлове

Merge branch 'mj' into dev

zwhui преди 2 години
родител
ревизия
55f09a7bf5
променени са 1 файла, в които са добавени 6 реда и са изтрити 1 реда
  1. 6 1
      netflix-service/src/main/java/com/cyksj/service/midjourney/impl/MidjourneyServiceImpl.java

+ 6 - 1
netflix-service/src/main/java/com/cyksj/service/midjourney/impl/MidjourneyServiceImpl.java

@@ -287,8 +287,13 @@ public class MidjourneyServiceImpl implements MidjourneyService {
     private static List<String> uploadBase64Pic(List<String> base64Array) throws IOException {
         List<String> list = new ArrayList<>();
         for (String base64 : base64Array) {
+            String fileExt = "jpeg";
+            if (base64.contains(";")){
+                fileExt = base64.split(";")[0].split("/")[1];
+                base64 = base64.split(",")[1];
+            }
             byte[] body = Base64.getDecoder().decode(base64);
-            Path tempFile = Files.createTempFile(UUID.randomUUID().toString(), ".jpeg");
+            Path tempFile = Files.createTempFile(UUID.randomUUID().toString(), "." + fileExt);
             try (FileImageOutputStream imageOutput = new FileImageOutputStream(tempFile.toFile())) {
                 imageOutput.write(body, 0, body.length);
             }