diff --git a/src/main/java/com/ai/da/service/impl/LibraryServiceImpl.java b/src/main/java/com/ai/da/service/impl/LibraryServiceImpl.java index 6679b842..39e99876 100644 --- a/src/main/java/com/ai/da/service/impl/LibraryServiceImpl.java +++ b/src/main/java/com/ai/da/service/impl/LibraryServiceImpl.java @@ -848,10 +848,13 @@ public class LibraryServiceImpl extends ServiceImpl impl if (Objects.isNull(sysFile)) { throw new BusinessException("model.not.found", ResultEnum.PROMPT.getCode()); } + SysFileExtra sysFileExtra = sysFileExtraMapper.selectOne(new QueryWrapper().eq("sys_id", sysFile.getId())); + // 添加模特信息到library Library library = CopyUtil.copyObject(sysFile, Library.class); library.setId(null); library.setAccountId(accountId); + library.setLevel3Type(sysFileExtra.getAgeGroup()); library.setName(DateUtil.dateToStr(new Date(), DateUtil.YYYY_MM_DD)); List imagesWidthAndHeight = minioUtil.getImagesWidthAndHeight(sysFile.getUrl()); library.setWidth(imagesWidthAndHeight.get(0)); @@ -860,6 +863,14 @@ public class LibraryServiceImpl extends ServiceImpl impl library.setUpdateDate(null); libraryMapper.insert(library); + // 添加点位信息到library + LibraryModelPoint sysModelPoint = libraryModelPointService.getByRelationId(sysModelId, "System"); + sysModelPoint.setId(null); + sysModelPoint.setModelType("Library"); + sysModelPoint.setRelationId(library.getId()); + sysModelPoint.setCreateDate(new Date()); + libraryModelPointService.save(sysModelPoint); + return library.getId(); } }