mirror of
				https://github.com/continew-org/continew-admin.git
				synced 2025-10-31 22:57:17 +08:00 
			
		
		
		
	chore(generator): 优化部分代码生成代码
This commit is contained in:
		| @@ -16,6 +16,7 @@ | ||||
|  | ||||
| package top.continew.admin.generator.model.entity; | ||||
|  | ||||
| import cn.hutool.core.util.StrUtil; | ||||
| import com.baomidou.mybatisplus.annotation.*; | ||||
| import io.swagger.v3.oas.annotations.media.Schema; | ||||
| import jakarta.validation.constraints.NotBlank; | ||||
| @@ -25,6 +26,7 @@ import lombok.Data; | ||||
| import lombok.NoArgsConstructor; | ||||
| import org.hibernate.validator.constraints.Length; | ||||
| import top.continew.admin.common.constant.RegexConstants; | ||||
| import top.continew.starter.core.constant.StringConstants; | ||||
|  | ||||
| import java.io.Serial; | ||||
| import java.io.Serializable; | ||||
| @@ -114,6 +116,15 @@ public class GenConfigDO implements Serializable { | ||||
|     private LocalDateTime updateTime; | ||||
|  | ||||
|     public GenConfigDO(String tableName) { | ||||
|         this.setTableName(tableName); | ||||
|     } | ||||
|  | ||||
|     public void setTableName(String tableName) { | ||||
|         this.tableName = tableName; | ||||
|         // 默认表前缀(sys_user -> sys_) | ||||
|         int underLineIndex = StrUtil.indexOf(tableName, StringConstants.C_UNDERLINE); | ||||
|         if (-1 != underLineIndex) { | ||||
|             this.tablePrefix = StrUtil.subPre(tableName, underLineIndex + 1); | ||||
|         } | ||||
|     } | ||||
| } | ||||
|   | ||||
| @@ -125,11 +125,6 @@ public class GeneratorServiceImpl implements GeneratorService { | ||||
|             if (null != lastGenConfig) { | ||||
|                 genConfig.setAuthor(lastGenConfig.getAuthor()); | ||||
|             } | ||||
|             // 默认表前缀(sys_user -> sys_) | ||||
|             int underLineIndex = StrUtil.indexOf(tableName, StringConstants.C_UNDERLINE); | ||||
|             if (-1 != underLineIndex) { | ||||
|                 genConfig.setTablePrefix(StrUtil.subPre(tableName, underLineIndex + 1)); | ||||
|             } | ||||
|         } | ||||
|         return genConfig; | ||||
|     } | ||||
| @@ -245,19 +240,16 @@ public class GeneratorServiceImpl implements GeneratorService { | ||||
|             GeneratePreviewResp generatePreview = new GeneratePreviewResp(); | ||||
|             generatePreview.setBackend(isBackend); | ||||
|             generatePreviewList.add(generatePreview); | ||||
|             if (isBackend) { | ||||
|                 generatePreview.setFileName(className + extension); | ||||
|                 generatePreview.setContent(TemplateUtils.render(templateConfig.getTemplatePath(), BeanUtil | ||||
|                     .beanToMap(innerGenConfig))); | ||||
|             } else { | ||||
|                 generatePreview.setFileName(".vue".equals(extension) && "index".equals(classNameSuffix) | ||||
|             String fileName = className + extension; | ||||
|             if (!isBackend) { | ||||
|                 fileName = ".vue".equals(extension) && "index".equals(classNameSuffix) | ||||
|                     ? "index.vue" | ||||
|                     : this.getFrontendFileName(classNamePrefix, className, extension)); | ||||
|                 innerGenConfig.setFieldConfigs(fieldConfigList); | ||||
|                     : this.getFrontendFileName(classNamePrefix, className, extension); | ||||
|             } | ||||
|             generatePreview.setFileName(fileName); | ||||
|             generatePreview.setContent(TemplateUtils.render(templateConfig.getTemplatePath(), BeanUtil | ||||
|                 .beanToMap(innerGenConfig))); | ||||
|             } | ||||
|             setPreviewPath(generatePreview, innerGenConfig, templateConfig); | ||||
|             this.setPreviewPath(generatePreview, innerGenConfig, templateConfig); | ||||
|         } | ||||
|         return generatePreviewList; | ||||
|     } | ||||
|   | ||||
		Reference in New Issue
	
	Block a user