From c5e5b51852399f4d79d9543bcd8fef13801238fb Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=BE=90=E4=BD=A9?= <1779019091@qq.com> Date: Thu, 17 Aug 2023 16:11:26 +0800 Subject: [PATCH] =?UTF-8?q?generateSketch=20=E5=8A=9F=E8=83=BD--=E7=A9=BA?= =?UTF-8?q?=E6=8C=87=E9=92=88=E5=BC=82=E5=B8=B8=20=E4=BF=AE=E6=94=B9?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../da/service/impl/GenerateServiceImpl.java | 18 +++++++++++------- 1 file changed, 11 insertions(+), 7 deletions(-) diff --git a/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java b/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java index cb2dbda5..1cfa5865 100644 --- a/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java +++ b/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java @@ -87,14 +87,18 @@ public class GenerateServiceImpl extends ServiceImpl im // 3、将请求信息落库 // 3.1 sketch在t_collection_element表中的信息是否需要更新 如 level2Type - CollectionElement collectionElement = collectionElementMapper.selectById(sketchId); - if (StringUtil.isNullOrEmpty(collectionElement.getLevel2Type()) || !(collectionElement.getLevel2Type()).equals(generateThroughImageTextDTO.getLevel2Type()) ){ - collectionElement.setLevel2Type(generateThroughImageTextDTO.getLevel2Type()); - QueryWrapper queryWrapper = new QueryWrapper<>(); - queryWrapper.eq("id", sketchId); - collectionElementMapper.update(collectionElement,queryWrapper); + CollectionElement collectionElement = null; + if(!Objects.isNull(sketchId)){ + collectionElement = collectionElementMapper.selectById(sketchId); + if (StringUtil.isNullOrEmpty(collectionElement.getLevel2Type()) || !(collectionElement.getLevel2Type()).equals(generateThroughImageTextDTO.getLevel2Type()) ){ + collectionElement.setLevel2Type(generateThroughImageTextDTO.getLevel2Type()); + QueryWrapper queryWrapper = new QueryWrapper<>(); + queryWrapper.eq("id", sketchId); + collectionElementMapper.update(collectionElement,queryWrapper); + } } + // 3.2 将本次generate的请求信息添加到t_generate表中 save(generate); @@ -106,7 +110,7 @@ public class GenerateServiceImpl extends ServiceImpl im GenerateCollectionVO generateCollectionVO = new GenerateCollectionVO(); List generatedCollectionItems = new ArrayList<>(); generateCollectionVO.setGenerateId(generate.getId()); - generateCollectionVO.setCollectionId(collectionElement.getCollectionId()); + generateCollectionVO.setCollectionId(Objects.isNull(collectionElement) ? null : collectionElement.getCollectionId()); generateCollectionVO.setGeneratedCollectionItems(generatedCollectionItems); // 5、处理模型返回的数据 // 5.1 将相应的url保存到数据库