diff --git a/agileboot-system/wol-codegenerator/src/main/java/com/agileboot/codegen/mapper/UserTemplateRepositoryMapper.java b/agileboot-system/wol-codegenerator/src/main/java/com/agileboot/codegen/mapper/UserTemplateRepositoryMapper.java index 738445c..3449d90 100644 --- a/agileboot-system/wol-codegenerator/src/main/java/com/agileboot/codegen/mapper/UserTemplateRepositoryMapper.java +++ b/agileboot-system/wol-codegenerator/src/main/java/com/agileboot/codegen/mapper/UserTemplateRepositoryMapper.java @@ -25,9 +25,11 @@ public interface UserTemplateRepositoryMapper extends BaseMapper selectEnabledByUserId(@Param("userId") Long userId); /** @@ -36,9 +38,11 @@ public interface UserTemplateRepositoryMapper extends BaseMapper selectAllByUserId(@Param("userId") Long userId); /** diff --git a/agileboot-system/wol-codegenerator/src/main/java/com/agileboot/codegen/service/impl/UserTemplateServiceImpl.java b/agileboot-system/wol-codegenerator/src/main/java/com/agileboot/codegen/service/impl/UserTemplateServiceImpl.java index 968146b..9dd5eb4 100644 --- a/agileboot-system/wol-codegenerator/src/main/java/com/agileboot/codegen/service/impl/UserTemplateServiceImpl.java +++ b/agileboot-system/wol-codegenerator/src/main/java/com/agileboot/codegen/service/impl/UserTemplateServiceImpl.java @@ -10,6 +10,7 @@ import com.agileboot.common.satoken.utils.LoginHelper; import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; import com.baomidou.mybatisplus.extension.plugins.pagination.Page; import java.util.Arrays; +import com.agileboot.codegen.mapper.UserTemplateRepositoryMapper; import freemarker.template.Configuration; import freemarker.template.Template; import lombok.RequiredArgsConstructor; @@ -36,6 +37,7 @@ import java.util.stream.Collectors; public class UserTemplateServiceImpl implements IUserTemplateService { private final UserTemplateMapper userTemplateMapper; + private final UserTemplateRepositoryMapper repositoryMapper; @Override public Page selectUserTemplateList(Page page, Long userId, String templateGroup, Integer isPublic) { @@ -149,6 +151,9 @@ public class UserTemplateServiceImpl implements IUserTemplateService { throw new RuntimeException("删除操作失败"); } + // 清理模板仓库中的相关记录 + repositoryMapper.deleteByTemplate(currentUserId, "user", id.toString()); + return true; } @@ -169,7 +174,14 @@ public class UserTemplateServiceImpl implements IUserTemplateService { } } - return userTemplateMapper.deleteByIds(Arrays.asList(ids)) > 0; + boolean deleteResult = userTemplateMapper.deleteByIds(Arrays.asList(ids)) > 0; + + // 清理模板仓库中的相关记录 + for (Long id : ids) { + repositoryMapper.deleteByTemplate(currentUserId, "user", id.toString()); + } + + return deleteResult; } @Override