zwhui il y a 1 an
Parent
commit
5dcc7b77b0

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

@@ -1457,7 +1457,9 @@ public class MidjourneyServiceImpl implements MidjourneyService {
                 .stream().map(item -> JSONUtil.toBean((String) item, MidjourneyUserQueue.class)).collect(Collectors.toList());
         list.forEach(item -> {
             try {
-                item.setBase64Array(JSONUtil.toJsonStr(List.of(encodeImageToBase64("/mnt/picture/"+item.getQueueId()+"."+item.getFileExt(),item.getFileExt()))));
+                if (item.getFileExt() != null){
+                    item.setBase64Array(JSONUtil.toJsonStr(List.of(encodeImageToBase64("/mnt/picture/"+item.getQueueId()+"."+item.getFileExt(),item.getFileExt()))));
+                }
                 Long rank = redisService.zRank(RedisService.key.MIDJOURNEY_QUEUE_LIMIT.getName() + userId, item.getQueueId());
                 item.setPosition((rank != null) ? rank + 1 : 1);
             } catch (IOException e) {
@@ -1558,6 +1560,7 @@ public class MidjourneyServiceImpl implements MidjourneyService {
      * base64图片存本地
      */
     public static String base64ToPic(Long queueId,List<String> base64Array) throws IOException {
+        if (base64Array == null) return null;
         for (String base64 : base64Array) {
             String fileExt = "jpeg";
             if (base64.contains(";")){