From b3a0e377e5616951e8804e54d173fd1c90e29e21 Mon Sep 17 00:00:00 2001 From: zzhuan <1260310633@qq.com> Date: Wed, 14 Jan 2026 17:47:36 +0800 Subject: [PATCH] =?UTF-8?q?=E6=A8=A1=E6=9D=BF=E7=AE=A1=E7=90=86?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../controller/TemplateController.java | 70 +++++++++++ .../xinghe/codegen/entity/TemplateEntity.java | 60 ++++++++++ .../entity/enums/TemplateTypeEnum.java | 56 +++++++++ .../entity/req/menu/MenuBindFunReq.java | 3 +- .../entity/req/template/TemplateAddReq.java | 43 +++++++ .../entity/req/template/TemplateEditReq.java | 45 ++++++++ .../entity/req/template/TemplateQueryReq.java | 32 +++++ .../entity/res/template/TemplateQueryRes.java | 40 +++++++ .../xinghe/codegen/mapper/TemplateMapper.java | 16 +++ .../codegen/mapper/def/TemplateDef.java | 72 ++++++++++++ .../codegen/service/TemplateService.java | 45 ++++++++ .../codegen/service/impl/MenuServiceImpl.java | 74 ++++++------ .../service/impl/TemplateServiceImpl.java | 109 ++++++++++++++++++ 13 files changed, 628 insertions(+), 37 deletions(-) create mode 100644 src/main/java/com/cczsa/xinghe/codegen/controller/TemplateController.java create mode 100644 src/main/java/com/cczsa/xinghe/codegen/entity/TemplateEntity.java create mode 100644 src/main/java/com/cczsa/xinghe/codegen/entity/enums/TemplateTypeEnum.java create mode 100644 src/main/java/com/cczsa/xinghe/codegen/entity/req/template/TemplateAddReq.java create mode 100644 src/main/java/com/cczsa/xinghe/codegen/entity/req/template/TemplateEditReq.java create mode 100644 src/main/java/com/cczsa/xinghe/codegen/entity/req/template/TemplateQueryReq.java create mode 100644 src/main/java/com/cczsa/xinghe/codegen/entity/res/template/TemplateQueryRes.java create mode 100644 src/main/java/com/cczsa/xinghe/codegen/mapper/TemplateMapper.java create mode 100644 src/main/java/com/cczsa/xinghe/codegen/mapper/def/TemplateDef.java create mode 100644 src/main/java/com/cczsa/xinghe/codegen/service/TemplateService.java create mode 100644 src/main/java/com/cczsa/xinghe/codegen/service/impl/TemplateServiceImpl.java diff --git a/src/main/java/com/cczsa/xinghe/codegen/controller/TemplateController.java b/src/main/java/com/cczsa/xinghe/codegen/controller/TemplateController.java new file mode 100644 index 0000000..e8b8ea7 --- /dev/null +++ b/src/main/java/com/cczsa/xinghe/codegen/controller/TemplateController.java @@ -0,0 +1,70 @@ +package com.cczsa.xinghe.codegen.controller; + +import com.cczsa.xinghe.codegen.entity.req.template.TemplateAddReq; +import com.cczsa.xinghe.codegen.entity.req.template.TemplateEditReq; +import com.cczsa.xinghe.codegen.entity.req.template.TemplateQueryReq; +import com.cczsa.xinghe.codegen.entity.res.template.TemplateQueryRes; +import com.cczsa.xinghe.codegen.service.TemplateService; +import com.cczsa.xinghe.codegen.util.XResult; +import io.swagger.v3.oas.annotations.Operation; +import io.swagger.v3.oas.annotations.tags.Tag; +import jakarta.validation.Valid; +import lombok.RequiredArgsConstructor; +import org.springframework.web.bind.annotation.DeleteMapping; +import org.springframework.web.bind.annotation.GetMapping; +import org.springframework.web.bind.annotation.PathVariable; +import org.springframework.web.bind.annotation.PostMapping; +import org.springframework.web.bind.annotation.PutMapping; +import org.springframework.web.bind.annotation.RequestBody; +import org.springframework.web.bind.annotation.RequestMapping; +import org.springframework.web.bind.annotation.RestController; + +import java.util.List; +import java.util.Map; + +/** + * 模板管理 控制层。 + * + * @author xia + * @version 0.0.1 + */ +@Tag(name = "模板管理") +@RequiredArgsConstructor +@RestController +@RequestMapping("/template") +public class TemplateController { + + private final TemplateService templateService; + + @Operation(summary = "获取模板列表", description = "获取模板列表") + @PostMapping("/query") + public XResult> query(@RequestBody @Valid TemplateQueryReq req) { + return templateService.query(req); + } + + @Operation(summary = "创建模板", description = "创建模板") + @PutMapping("/add") + public XResult add(@RequestBody @Valid TemplateAddReq req) { + return templateService.add(req); + } + + @Operation(summary = "修改模板", description = "修改模板") + @PostMapping("/edit") + public XResult edit(@RequestBody @Valid TemplateEditReq req) { + return templateService.edit(req); + } + + @Operation(summary = "删除模板", description = "删除模板") + @DeleteMapping("/delete/{id}") + public XResult delete(@PathVariable Long id) { + return templateService.delete(id); + } + + @Operation(summary = "模板类型列表", description = "模板类型列表") + @GetMapping("/templateType") + public XResult>> getTemplateTypeList() { + return templateService.getTemplateTypeList(); + } + + +} \ No newline at end of file diff --git a/src/main/java/com/cczsa/xinghe/codegen/entity/TemplateEntity.java b/src/main/java/com/cczsa/xinghe/codegen/entity/TemplateEntity.java new file mode 100644 index 0000000..750a1e0 --- /dev/null +++ b/src/main/java/com/cczsa/xinghe/codegen/entity/TemplateEntity.java @@ -0,0 +1,60 @@ +package com.cczsa.xinghe.codegen.entity; + +import com.cczsa.xinghe.codegen.entity.enums.TemplateTypeEnum; +import com.mybatisflex.annotation.Id; +import com.mybatisflex.annotation.KeyType; +import com.mybatisflex.annotation.Table; +import java.io.Serializable; + +import java.io.Serial; + +import lombok.AllArgsConstructor; +import lombok.Builder; +import lombok.Data; +import lombok.NoArgsConstructor; +import lombok.EqualsAndHashCode; + +/** + * 实体类。 + * + * @author My + * @since 0.0.1 + */ +@Data +@Builder +@NoArgsConstructor +@AllArgsConstructor +@EqualsAndHashCode(callSuper = true) +@Table("cg_template") +public class TemplateEntity extends BaseEntity implements Serializable { + + @Serial + private static final long serialVersionUID = 1L; + + /** + * 主键ID + */ + @Id(keyType = KeyType.Auto) + private Long id; + + /** + * 是否使用 + */ + private Boolean isUse; + + /** + * 模板名称 + */ + private String templateName; + + /** + * 模板类型 + */ + private TemplateTypeEnum templateType; + + /** + * 正文 + */ + private String content; + +} diff --git a/src/main/java/com/cczsa/xinghe/codegen/entity/enums/TemplateTypeEnum.java b/src/main/java/com/cczsa/xinghe/codegen/entity/enums/TemplateTypeEnum.java new file mode 100644 index 0000000..5299044 --- /dev/null +++ b/src/main/java/com/cczsa/xinghe/codegen/entity/enums/TemplateTypeEnum.java @@ -0,0 +1,56 @@ +package com.cczsa.xinghe.codegen.entity.enums; + +import com.fasterxml.jackson.annotation.JsonValue; +import com.mybatisflex.annotation.EnumValue; +import io.swagger.v3.oas.annotations.media.Schema; +import lombok.Getter; + +import java.util.Arrays; +import java.util.HashMap; +import java.util.List; +import java.util.Map; +import java.util.stream.Collectors; + +/** + * 模板类型枚举 + * + * @author My + */ +@Schema(description = "模板类型", + example = "0", + allowableValues = {"0: controller", "1: service", "2: mapper"}) +@Getter +public enum TemplateTypeEnum { + + CONTROLLER(0, "controller"), + SERVICE(1, "service"), + MAPPER(2, "mapper"); + + private final int code; + private final String desc; + + TemplateTypeEnum(int code, String desc) { + this.code = code; + this.desc = desc; + } + + @JsonValue + @EnumValue + public int getCode() { + return code; + } + + /** + * 获取枚举值列表 + */ + public static List> getEnumList() { + return Arrays.stream(values()) + .map(enumValue -> { + Map map = new HashMap<>(); + map.put(enumValue.getCode(), enumValue.getDesc()); + return map; + }) + .collect(Collectors.toList()); + } + +} \ No newline at end of file diff --git a/src/main/java/com/cczsa/xinghe/codegen/entity/req/menu/MenuBindFunReq.java b/src/main/java/com/cczsa/xinghe/codegen/entity/req/menu/MenuBindFunReq.java index b739300..18b4f6c 100644 --- a/src/main/java/com/cczsa/xinghe/codegen/entity/req/menu/MenuBindFunReq.java +++ b/src/main/java/com/cczsa/xinghe/codegen/entity/req/menu/MenuBindFunReq.java @@ -7,6 +7,7 @@ import lombok.Setter; import java.io.Serial; import java.io.Serializable; +import java.util.List; /** * 菜单绑定权限 请求参数 @@ -28,6 +29,6 @@ public class MenuBindFunReq implements Serializable { @NotNull(message = "功能ID不能为空") @Schema(description = "功能ID") - private Long funId; + private List funIdList; } diff --git a/src/main/java/com/cczsa/xinghe/codegen/entity/req/template/TemplateAddReq.java b/src/main/java/com/cczsa/xinghe/codegen/entity/req/template/TemplateAddReq.java new file mode 100644 index 0000000..df08b1e --- /dev/null +++ b/src/main/java/com/cczsa/xinghe/codegen/entity/req/template/TemplateAddReq.java @@ -0,0 +1,43 @@ +package com.cczsa.xinghe.codegen.entity.req.template; + +import com.cczsa.xinghe.codegen.entity.enums.TemplateTypeEnum; +import io.swagger.v3.oas.annotations.media.Schema; +import jakarta.validation.constraints.NotBlank; +import jakarta.validation.constraints.NotNull; +import lombok.Getter; +import lombok.Setter; + +import java.io.Serial; +import java.io.Serializable; + +/** + * 创建模板 请求参数 + * + * @author xia + * @version 0.0.1 + */ +@Getter +@Setter +@Schema(description = "创建模板-参数") +public class TemplateAddReq implements Serializable { + + @Serial + private static final long serialVersionUID = 1L; + + @NotNull(message = "是否使用不能为空") + @Schema(description = "是否使用") + private Boolean isUse; + + @NotBlank(message = "模板名称不能为空") + @Schema(description = "模板名称") + private String templateName; + + @NotNull(message = "模板类型不能为空") + @Schema(description = "模板类型") + private TemplateTypeEnum templateType; + + @NotBlank(message = "正文不能为空") + @Schema(description = "正文") + private String content; + +} diff --git a/src/main/java/com/cczsa/xinghe/codegen/entity/req/template/TemplateEditReq.java b/src/main/java/com/cczsa/xinghe/codegen/entity/req/template/TemplateEditReq.java new file mode 100644 index 0000000..4dcfd23 --- /dev/null +++ b/src/main/java/com/cczsa/xinghe/codegen/entity/req/template/TemplateEditReq.java @@ -0,0 +1,45 @@ +package com.cczsa.xinghe.codegen.entity.req.template; + +import com.cczsa.xinghe.codegen.entity.enums.TemplateTypeEnum; +import io.swagger.v3.oas.annotations.media.Schema; +import jakarta.validation.constraints.NotBlank; +import jakarta.validation.constraints.NotNull; +import lombok.Getter; +import lombok.Setter; +import java.io.Serial; +import java.io.Serializable; + +/** + * 修改模板 请求参数 + * @author xia + * @version 0.0.1 + */ +@Getter +@Setter +@Schema(description = "修改模板-参数") +public class TemplateEditReq implements Serializable { + + @Serial + private static final long serialVersionUID = 1L; + + @NotNull(message = "id不能为空") + @Schema(description = "id") + private Long id; + + @NotNull(message = "是否使用不能为空") + @Schema(description = "是否使用") + private Boolean isUse; + + @NotBlank(message = "模板名称不能为空") + @Schema(description = "模板名称") + private String templateName; + + @NotNull(message = "模板类型不能为空") + @Schema(description = "模板类型") + private TemplateTypeEnum templateType; + + @NotBlank(message = "正文不能为空") + @Schema(description = "正文") + private String content; + +} diff --git a/src/main/java/com/cczsa/xinghe/codegen/entity/req/template/TemplateQueryReq.java b/src/main/java/com/cczsa/xinghe/codegen/entity/req/template/TemplateQueryReq.java new file mode 100644 index 0000000..fd301b5 --- /dev/null +++ b/src/main/java/com/cczsa/xinghe/codegen/entity/req/template/TemplateQueryReq.java @@ -0,0 +1,32 @@ +package com.cczsa.xinghe.codegen.entity.req.template; + +import com.cczsa.xinghe.codegen.entity.enums.TemplateTypeEnum; +import io.swagger.v3.oas.annotations.media.Schema; +import lombok.Getter; +import lombok.Setter; +import java.io.Serial; +import java.io.Serializable; + +/** + * 获取模板列表 请求参数 + * @author xia + * @version 0.0.1 + */ +@Getter +@Setter +@Schema(description = "获取模板列表-参数") +public class TemplateQueryReq implements Serializable { + + @Serial + private static final long serialVersionUID = 1L; + + @Schema(description = "是否使用") + private Boolean isUse; + + @Schema(description = "模板名称") + private String templateName; + + @Schema(description = "模板类型") + private TemplateTypeEnum templateType; + +} diff --git a/src/main/java/com/cczsa/xinghe/codegen/entity/res/template/TemplateQueryRes.java b/src/main/java/com/cczsa/xinghe/codegen/entity/res/template/TemplateQueryRes.java new file mode 100644 index 0000000..1de6e5e --- /dev/null +++ b/src/main/java/com/cczsa/xinghe/codegen/entity/res/template/TemplateQueryRes.java @@ -0,0 +1,40 @@ +package com.cczsa.xinghe.codegen.entity.res.template; + +import com.cczsa.xinghe.codegen.entity.enums.TemplateTypeEnum; +import io.swagger.v3.oas.annotations.media.Schema; +import jakarta.validation.constraints.NotBlank; +import jakarta.validation.constraints.NotNull; +import lombok.Getter; +import lombok.Setter; +import java.io.Serial; +import java.io.Serializable; + +/** + * 获取模板列表 响应参数 + * @author xia + * @version 0.0.1 + */ +@Getter +@Setter +@Schema(description = "获取模板列表-响应") +public class TemplateQueryRes implements Serializable { + + @Serial + private static final long serialVersionUID = 1L; + + @Schema(description = "id") + private Long id; + + @Schema(description = "是否使用") + private Boolean isUse; + + @Schema(description = "模板名称") + private String templateName; + + @Schema(description = "模板类型") + private TemplateTypeEnum templateType; + + @Schema(description = "正文") + private String content; + +} diff --git a/src/main/java/com/cczsa/xinghe/codegen/mapper/TemplateMapper.java b/src/main/java/com/cczsa/xinghe/codegen/mapper/TemplateMapper.java new file mode 100644 index 0000000..19ba289 --- /dev/null +++ b/src/main/java/com/cczsa/xinghe/codegen/mapper/TemplateMapper.java @@ -0,0 +1,16 @@ +package com.cczsa.xinghe.codegen.mapper; + +import org.apache.ibatis.annotations.Mapper; +import com.mybatisflex.core.BaseMapper; +import com.cczsa.xinghe.codegen.entity.TemplateEntity; + +/** + * 映射层。 + * + * @author My + * @since 0.0.1 + */ +@Mapper +public interface TemplateMapper extends BaseMapper { + +} diff --git a/src/main/java/com/cczsa/xinghe/codegen/mapper/def/TemplateDef.java b/src/main/java/com/cczsa/xinghe/codegen/mapper/def/TemplateDef.java new file mode 100644 index 0000000..9e57fd3 --- /dev/null +++ b/src/main/java/com/cczsa/xinghe/codegen/mapper/def/TemplateDef.java @@ -0,0 +1,72 @@ +package com.cczsa.xinghe.codegen.mapper.def; + +import com.mybatisflex.core.query.QueryColumn; +import com.mybatisflex.core.table.TableDef; + +import java.io.Serial; + +/** + * 表定义层。 + * + * @author My + * @since 0.0.1 + */ +public class TemplateDef extends TableDef { + + @Serial + private static final long serialVersionUID = 1L; + + /** + * + */ + public static final TemplateDef TEMPLATE_ENTITY = new TemplateDef(); + + /** + * 主键ID + */ + public final QueryColumn ID = new QueryColumn(this, "id"); + + /** + * 是否使用 + */ + public final QueryColumn IS_USE = new QueryColumn(this, "is_use"); + + /** + * 正文 + */ + public final QueryColumn CONTENT = new QueryColumn(this, "content"); + + /** + * 模板名称 + */ + public final QueryColumn TEMPLATE_NAME = new QueryColumn(this, "template_name"); + + /** + * 模板类型 + */ + public final QueryColumn TEMPLATE_TYPE = new QueryColumn(this, "template_type"); + + /** + * 所有字段。 + */ + public final QueryColumn ALL_COLUMNS = new QueryColumn(this, "*"); + + /** + * 默认字段，不包含逻辑删除或者 large 等字段。 + */ + public final QueryColumn[] DEFAULT_COLUMNS = new QueryColumn[]{ID, IS_USE, TEMPLATE_NAME, TEMPLATE_TYPE, CONTENT}; + + public TemplateDef() { + super("", "cg_template"); + } + + private TemplateDef(String schema, String name, String alisa) { + super(schema, name, alisa); + } + + public TemplateDef as(String alias) { + String key = getNameWithSchema() + "." + alias; + return getCache(key, k -> new TemplateDef("", "cg_template", alias)); + } + +} diff --git a/src/main/java/com/cczsa/xinghe/codegen/service/TemplateService.java b/src/main/java/com/cczsa/xinghe/codegen/service/TemplateService.java new file mode 100644 index 0000000..159ae15 --- /dev/null +++ b/src/main/java/com/cczsa/xinghe/codegen/service/TemplateService.java @@ -0,0 +1,45 @@ +package com.cczsa.xinghe.codegen.service; + +import com.cczsa.xinghe.codegen.entity.req.template.TemplateAddReq; +import com.cczsa.xinghe.codegen.entity.req.template.TemplateEditReq; +import com.cczsa.xinghe.codegen.entity.req.template.TemplateQueryReq; +import com.cczsa.xinghe.codegen.entity.res.template.TemplateQueryRes; +import com.cczsa.xinghe.codegen.util.XResult; + +import java.util.List; +import java.util.Map; + +/** + * 模板管理 服务层接口。 + * + * @author xia + * @version 0.0.1 + */ +public interface TemplateService { + + /** + * 获取模板列表 + */ + XResult> query(TemplateQueryReq req); + + /** + * 创建模板 + */ + XResult add(TemplateAddReq req); + + /** + * 修改模板 + */ + XResult edit(TemplateEditReq req); + + /** + * 删除模板 + */ + XResult delete(Long id); + + /** + * 获取模板类型列表 + */ + XResult>> getTemplateTypeList(); + +} \ No newline at end of file diff --git a/src/main/java/com/cczsa/xinghe/codegen/service/impl/MenuServiceImpl.java b/src/main/java/com/cczsa/xinghe/codegen/service/impl/MenuServiceImpl.java index 908980f..80bdfa5 100644 --- a/src/main/java/com/cczsa/xinghe/codegen/service/impl/MenuServiceImpl.java +++ b/src/main/java/com/cczsa/xinghe/codegen/service/impl/MenuServiceImpl.java @@ -194,47 +194,49 @@ public class MenuServiceImpl implements MenuService { return XResult.failed("菜单不存在"); } // 获取操作 - QueryWrapper queryFun = new QueryWrapper(); - queryFun.eq(FunOperationEntity::getId, req.getFunId()); - FunOperationEntity funOperationEntity = funOperationMapper.selectOneByQuery(queryFun); - if (funOperationEntity == null) { - return XResult.failed("权限不存在"); - } - // 获取功能 - QueryWrapper queryItem = new QueryWrapper(); - queryItem.eq(FunItemEntity::getId, funOperationEntity.getItemId()); - FunItemEntity funItemEntity = funItemMapper.selectOneByQuery(queryItem); - if (funItemEntity == null) { - return XResult.failed("功能不存在"); - } - Boolean menuIsTenant = menuEntity.getIsTenant(); - Boolean itemIsTenant = funItemEntity.getIsTenant(); - if (!menuIsTenant.equals(itemIsTenant)) { - return XResult.failed("菜单和权限是否租户不一致"); - } + List funIdList = req.getFunIdList(); // 删除旧绑定 QueryWrapper deleteMenuBindFun = new QueryWrapper(); deleteMenuBindFun.eq(MenuBindFunEntity::getMenuId, req.getMenuId()); - deleteMenuBindFun.eq(MenuBindFunEntity::getFunId, req.getFunId()); menuBindFunMapper.deleteByQuery(deleteMenuBindFun); - // 获取最大ID - MenuBindFunDef menuBindFunDef = MenuBindFunDef.MENU_BIND_FUN_ENTITY; - QueryWrapper queryId = new QueryWrapper(); - queryId.select(menuBindFunDef.ID) - .from(menuBindFunDef) - .orderBy(MenuBindFunEntity::getId, false); - Long maxId = menuBindFunMapper.selectOneByQueryAs(queryId, Long.class); - if (maxId == null) { - maxId = 10000L; - } else { - maxId++; + for (Long funId : funIdList) { + QueryWrapper queryFun = new QueryWrapper(); + queryFun.eq(FunOperationEntity::getId, funId); + FunOperationEntity funOperationEntity = funOperationMapper.selectOneByQuery(queryFun); + if (funOperationEntity == null) { + return XResult.failed("权限不存在:" + funId); + } + // 获取功能 + QueryWrapper queryItem = new QueryWrapper(); + queryItem.eq(FunItemEntity::getId, funOperationEntity.getItemId()); + FunItemEntity funItemEntity = funItemMapper.selectOneByQuery(queryItem); + if (funItemEntity == null) { + return XResult.failed("功能不存在:" + funId); + } + Boolean menuIsTenant = menuEntity.getIsTenant(); + Boolean itemIsTenant = funItemEntity.getIsTenant(); + if (!menuIsTenant.equals(itemIsTenant)) { + return XResult.failed("菜单和权限是否租户不一致:" + funId); + } + // 获取最大ID + MenuBindFunDef menuBindFunDef = MenuBindFunDef.MENU_BIND_FUN_ENTITY; + QueryWrapper queryId = new QueryWrapper(); + queryId.select(menuBindFunDef.ID) + .from(menuBindFunDef) + .orderBy(MenuBindFunEntity::getId, false); + Long maxId = menuBindFunMapper.selectOneByQueryAs(queryId, Long.class); + if (maxId == null) { + maxId = 10000L; + } else { + maxId++; + } + // 新增绑定 + MenuBindFunEntity menuBindFunEntity = new MenuBindFunEntity(); + menuBindFunEntity.setId(maxId); + menuBindFunEntity.setMenuId(req.getMenuId()); + menuBindFunEntity.setFunId(funId); + menuBindFunMapper.insertSelective(menuBindFunEntity); } - // 新增绑定 - MenuBindFunEntity menuBindFunEntity = new MenuBindFunEntity(); - menuBindFunEntity.setId(maxId); - menuBindFunEntity.setMenuId(req.getMenuId()); - menuBindFunEntity.setFunId(req.getFunId()); - menuBindFunMapper.insertSelective(menuBindFunEntity); return XResult.ok(); } diff --git a/src/main/java/com/cczsa/xinghe/codegen/service/impl/TemplateServiceImpl.java b/src/main/java/com/cczsa/xinghe/codegen/service/impl/TemplateServiceImpl.java new file mode 100644 index 0000000..2d3c951 --- /dev/null +++ b/src/main/java/com/cczsa/xinghe/codegen/service/impl/TemplateServiceImpl.java @@ -0,0 +1,109 @@ +package com.cczsa.xinghe.codegen.service.impl; + +import com.cczsa.xinghe.codegen.entity.TemplateEntity; +import com.cczsa.xinghe.codegen.entity.enums.TemplateTypeEnum; +import com.cczsa.xinghe.codegen.entity.req.template.TemplateAddReq; +import com.cczsa.xinghe.codegen.entity.req.template.TemplateEditReq; +import com.cczsa.xinghe.codegen.entity.req.template.TemplateQueryReq; +import com.cczsa.xinghe.codegen.entity.res.template.TemplateQueryRes; +import com.cczsa.xinghe.codegen.mapper.TemplateMapper; +import com.cczsa.xinghe.codegen.mapper.def.TemplateDef; +import com.cczsa.xinghe.codegen.service.TemplateService; +import com.cczsa.xinghe.codegen.util.XResult; +import com.github.xiaoymin.knife4j.core.util.StrUtil; +import com.mybatisflex.core.query.QueryWrapper; +import lombok.RequiredArgsConstructor; +import org.springframework.beans.BeanUtils; +import org.springframework.stereotype.Service; + +import java.util.List; +import java.util.Map; + +/** + * 模板管理 服务层实现。 + * + * @author xia + * @version 0.0.1 + */ +@Service +@RequiredArgsConstructor +public class TemplateServiceImpl implements TemplateService { + + private final TemplateMapper templateMapper; + + + /** + * 获取模板列表 + */ + @Override + public XResult> query(TemplateQueryReq req) { + TemplateDef templateDef = TemplateDef.TEMPLATE_ENTITY; + QueryWrapper queryWrapper = new QueryWrapper(); + queryWrapper.select(templateDef.ALL_COLUMNS) + .from(templateDef) + .like(TemplateEntity::getTemplateName, req.getTemplateName(), StrUtil.isNotBlank(req.getTemplateName())) + .eq(TemplateEntity::getTemplateType, req.getTemplateType(), req.getTemplateType() != null) + .eq(TemplateEntity::getIsUse, req.getIsUse(), req.getIsUse() != null); + List templateQueryRes = templateMapper.selectListByQueryAs(queryWrapper, TemplateQueryRes.class); + return XResult.ok(templateQueryRes); + } + + /** + * 创建模板 + */ + @Override + public XResult add(TemplateAddReq req) { + // 模板名称是否存在 + QueryWrapper queryName = new QueryWrapper(); + queryName.eq(TemplateEntity::getTemplateName, req.getTemplateName()); + long count = templateMapper.selectCountByQuery(queryName); + if (count > 0) { + return XResult.failed("模板名称已存在"); + } + TemplateEntity templateEntity = new TemplateEntity(); + BeanUtils.copyProperties(req, templateEntity); + templateMapper.insert(templateEntity); + return XResult.ok(); + } + + /** + * 修改模板 + */ + @Override + public XResult edit(TemplateEditReq req) { + // 获取模板 + TemplateEntity templateEntity = templateMapper.selectOneById(req.getId()); + if (templateEntity == null) { + return XResult.failed("模板不存在"); + } + // 模板名称是否存在 + QueryWrapper queryName = new QueryWrapper(); + queryName.eq(TemplateEntity::getTemplateName, req.getTemplateName()); + queryName.ne(TemplateEntity::getId, req.getId()); + long count = templateMapper.selectCountByQuery(queryName); + if (count > 0) { + return XResult.failed("模板名称已存在"); + } + BeanUtils.copyProperties(req, templateEntity); + templateMapper.update(templateEntity); + return XResult.ok(); + } + + /** + * 删除模板 + */ + @Override + public XResult delete(Long id) { + templateMapper.deleteById(id); + return XResult.ok(); + } + + /** + * 获取模板类型列表 + */ + @Override + public XResult>> getTemplateTypeList() { + return XResult.ok(TemplateTypeEnum.getEnumList()); + } + +} \ No newline at end of file