This commit is contained in:
litianxiang
2026-03-24 15:15:59 +08:00
parent 1c5a3a12b9
commit c970ebe691

View File

@@ -292,7 +292,6 @@ public class GenerateServiceImpl extends ServiceImpl<GenerateMapper, Generate> i
log.info("============ProcessGenerateResult listening==========");
log.debug("taskId: " + taskId);
String status = null;
try {
// 1、处理模型返回的数据
GenerateDetail generateDetail = new GenerateDetail();
GenerateCollectionItemVO generateCollectionItemVO = new GenerateCollectionItemVO();
@@ -309,18 +308,18 @@ public class GenerateServiceImpl extends ServiceImpl<GenerateMapper, Generate> i
}
}
// Generate generate = selectByUniqueId(taskId);
String md5 = MD5Utils.encryptFile(minioUtil.getPreSignedUrl(url, 24 * 60), Boolean.FALSE);
// 通过MD5值和level1Type,判断不同level1Type下相同的图片是否被like过
List<Map<String, Long>> libraryIdList = generateDetailMapper.getLibraryIdThroughMD5(md5, generate.getLevel1Type());
if (!libraryIdList.isEmpty()) {
generateDetail.setIsLike((byte) 1);
generateDetail.setLibraryId(libraryIdList.get(0).get("library_id"));
generateCollectionItemVO.setIsLiked(Boolean.TRUE);
}
// String md5 = MD5Utils.encryptFile(minioUtil.getPreSignedUrl(url, 24 * 60), Boolean.FALSE);
// // 通过MD5值和level1Type,判断不同level1Type下相同的图片是否被like过
// List<Map<String, Long>> libraryIdList = generateDetailMapper.getLibraryIdThroughMD5(md5, generate.getLevel1Type());
// if (!libraryIdList.isEmpty()) {
// generateDetail.setIsLike((byte) 1);
// generateDetail.setLibraryId(libraryIdList.get(0).get("library_id"));
// generateCollectionItemVO.setIsLiked(Boolean.TRUE);
// }
generateDetail.setUrl(url);
generateDetail.setGenerateId(generate.getId());
generateDetail.setCreateDate(LocalDateTime.now());
generateDetail.setMd5(md5);
generateDetail.setMd5("");
// 将相应的url保存到数据库
generateDetailMapper.insert(generateDetail);
log.debug("generateDetail: " + generateDetail.toString());
@@ -337,10 +336,7 @@ public class GenerateServiceImpl extends ServiceImpl<GenerateMapper, Generate> i
// 更新redis
redisUtil.addToString(key, new Gson().toJson(generateResultVO), CommonConstant.GENERATE_RESULT_EXPIRE_TIME);
log.debug("generateResultVO: " + generateResultVO.toString());
} catch (Exception e) {
e.printStackTrace();
log.error(e.getMessage());
}
// 执行积分扣除
// ** 注:如果生成的图片都是空白 则不扣积分