From 328086cec1d2c7346f2664bf6c81f2ceb84c9c05 Mon Sep 17 00:00:00 2001 From: Suomm <1474983351@qq.com> Date: Wed, 17 May 2023 20:15:11 +0800 Subject: [PATCH] =?UTF-8?q?feat:=20=E6=96=B0=E5=A2=9E=E8=AE=BE=E7=BD=AE?= =?UTF-8?q?=E6=A8=A1=E6=9D=BF=E6=96=87=E4=BB=B6=E4=BD=8D=E7=BD=AE=EF=BC=8C?= =?UTF-8?q?=E4=BD=BF=E7=94=A8=E6=8C=87=E5=AE=9A=E7=9A=84=E6=A8=A1=E6=9D=BF?= =?UTF-8?q?=E7=94=9F=E6=88=90=E6=96=87=E4=BB=B6=E3=80=82?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../codegen/config/GlobalConfig.java | 6 + .../codegen/config/StrategyConfig.java | 2 +- .../codegen/config/TemplateConfig.java | 157 ++++++++++++++++++ .../codegen/constant/GenTypeConst.java | 36 ++++ .../codegen/generator/GeneratorFactory.java | 19 ++- .../codegen/generator/IGenerator.java | 28 ++++ .../generator/impl/ControllerGenerator.java | 13 +- .../generator/impl/EntityGenerator.java | 13 +- .../generator/impl/MapperGenerator.java | 13 +- .../generator/impl/MapperXmlGenerator.java | 13 +- .../generator/impl/ServiceGenerator.java | 13 +- .../generator/impl/ServiceImplGenerator.java | 13 +- .../generator/impl/TableDefGenerator.java | 13 +- .../template/{ => impl}/EnjoyTemplate.java | 23 ++- 14 files changed, 338 insertions(+), 24 deletions(-) create mode 100644 mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/config/TemplateConfig.java create mode 100644 mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/constant/GenTypeConst.java rename mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/template/{ => impl}/EnjoyTemplate.java (67%) diff --git a/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/config/GlobalConfig.java b/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/config/GlobalConfig.java index c6b482d5..2b1de937 100644 --- a/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/config/GlobalConfig.java +++ b/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/config/GlobalConfig.java @@ -32,6 +32,7 @@ public class GlobalConfig { private final JavadocConfig javadocConfig; private final PackageConfig packageConfig; private final StrategyConfig strategyConfig; + private final TemplateConfig templateConfig; // === 可选配置 === @@ -61,6 +62,7 @@ public class GlobalConfig { this.javadocConfig = new JavadocConfig(); this.packageConfig = new PackageConfig(); this.strategyConfig = new StrategyConfig(); + this.templateConfig = new TemplateConfig(); } public JavadocConfig getJavadocConfig() { @@ -75,6 +77,10 @@ public class GlobalConfig { return strategyConfig; } + public TemplateConfig getTemplateConfig() { + return templateConfig; + } + public EntityConfig getEntityConfig() { if (entityConfig == null) { entityConfig = new EntityConfig(); diff --git a/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/config/StrategyConfig.java b/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/config/StrategyConfig.java index 0b1c53ed..d538e0be 100644 --- a/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/config/StrategyConfig.java +++ b/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/config/StrategyConfig.java @@ -1,7 +1,7 @@ package com.mybatisflex.codegen.config; -import com.mybatisflex.codegen.template.EnjoyTemplate; import com.mybatisflex.codegen.template.ITemplate; +import com.mybatisflex.codegen.template.impl.EnjoyTemplate; import java.util.HashMap; import java.util.HashSet; diff --git a/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/config/TemplateConfig.java b/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/config/TemplateConfig.java new file mode 100644 index 00000000..b0e0af8f --- /dev/null +++ b/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/config/TemplateConfig.java @@ -0,0 +1,157 @@ +/** + * Copyright (c) 2022-2023, Mybatis-Flex (fuhai999@gmail.com). + *

+ * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + *

+ * http://www.apache.org/licenses/LICENSE-2.0 + *

+ * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ +package com.mybatisflex.codegen.config; + +import com.mybatisflex.codegen.constant.GenTypeConst; +import com.mybatisflex.codegen.generator.GeneratorFactory; +import com.mybatisflex.codegen.template.ITemplate; +import com.mybatisflex.codegen.template.impl.EnjoyTemplate; + +/** + * 模板配置。 + * + * @author 王帅 + * @since 2023-05-17 + */ +@SuppressWarnings("unused") +public class TemplateConfig { + + /** + * 生成代码的模板引擎。 + */ + private ITemplate template = new EnjoyTemplate(); + + /** + * 获取模板引擎。 + */ + public ITemplate getTemplate() { + return template; + } + + /** + * 设置模板引擎。 + */ + public TemplateConfig setTemplate(ITemplate template) { + this.template = template; + return this; + } + + /** + * 获取生成 Entity 模板文件的位置。 + */ + public String getEntity() { + return GeneratorFactory.getGenerator(GenTypeConst.ENTITY).getTemplatePath(); + } + + /** + * 设置生成 Entity 模板文件的位置。 + */ + public TemplateConfig setEntity(String entity) { + GeneratorFactory.getGenerator(GenTypeConst.ENTITY).setTemplatePath(entity); + return this; + } + + /** + * 获取生成 Mapper 模板文件的位置。 + */ + public String getMapper() { + return GeneratorFactory.getGenerator(GenTypeConst.MAPPER).getTemplatePath(); + } + + /** + * 设置生成 Mapper 模板文件的位置。 + */ + public TemplateConfig setMapper(String mapper) { + GeneratorFactory.getGenerator(GenTypeConst.MAPPER).setTemplatePath(mapper); + return this; + } + + /** + * 获取生成 Service 模板文件的位置。 + */ + public String getService() { + return GeneratorFactory.getGenerator(GenTypeConst.SERVICE).getTemplatePath(); + } + + /** + * 设置生成 Service 模板文件的位置。 + */ + public TemplateConfig setService(String service) { + GeneratorFactory.getGenerator(GenTypeConst.SERVICE).setTemplatePath(service); + return this; + } + + /** + * 获取生成 ServiceImpl 模板文件的位置。 + */ + public String getServiceImpl() { + return GeneratorFactory.getGenerator(GenTypeConst.SERVICE_IMPL).getTemplatePath(); + } + + /** + * 设置生成 ServiceImpl 模板文件的位置。 + */ + public TemplateConfig setServiceImpl(String serviceImpl) { + GeneratorFactory.getGenerator(GenTypeConst.SERVICE_IMPL).setTemplatePath(serviceImpl); + return this; + } + + /** + * 获取生成 Controller 模板文件的位置。 + */ + public String getController() { + return GeneratorFactory.getGenerator(GenTypeConst.CONTROLLER).getTemplatePath(); + } + + /** + * 设置生成 Controller 模板文件的位置。 + */ + public TemplateConfig setController(String controller) { + GeneratorFactory.getGenerator(GenTypeConst.CONTROLLER).setTemplatePath(controller); + return this; + } + + /** + * 获取生成 TableDef 模板文件的位置。 + */ + public String getTableDef() { + return GeneratorFactory.getGenerator(GenTypeConst.TABLE_DEF).getTemplatePath(); + } + + /** + * 设置生成 TableDef 模板文件的位置。 + */ + public TemplateConfig setTableDef(String tableDef) { + GeneratorFactory.getGenerator(GenTypeConst.TABLE_DEF).setTemplatePath(tableDef); + return this; + } + + /** + * 获取生成 MapperXml 模板文件的位置。 + */ + public String getMapperXml() { + return GeneratorFactory.getGenerator(GenTypeConst.MAPPER_XML).getTemplatePath(); + } + + /** + * 设置生成 MapperXml 模板文件的位置。 + */ + public TemplateConfig setMapperXml(String mapperXml) { + GeneratorFactory.getGenerator(GenTypeConst.MAPPER_XML).setTemplatePath(mapperXml); + return this; + } + +} \ No newline at end of file diff --git a/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/constant/GenTypeConst.java b/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/constant/GenTypeConst.java new file mode 100644 index 00000000..915b1550 --- /dev/null +++ b/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/constant/GenTypeConst.java @@ -0,0 +1,36 @@ +/** + * Copyright (c) 2022-2023, Mybatis-Flex (fuhai999@gmail.com). + *

+ * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + *

+ * http://www.apache.org/licenses/LICENSE-2.0 + *

+ * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ +package com.mybatisflex.codegen.constant; + +/** + * 生成类型常量。 + * + * @author 王帅 + * @since 2023-05-17 + */ +public class GenTypeConst { + + public static final String ENTITY = "entity"; + public static final String MAPPER = "mapper"; + public static final String SERVICE = "service"; + public static final String SERVICE_IMPL = "serviceImpl"; + public static final String CONTROLLER = "controller"; + public static final String TABLE_DEF = "tableDef"; + public static final String MAPPER_XML = "mapperXml"; + private GenTypeConst() { + } + +} \ No newline at end of file diff --git a/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/GeneratorFactory.java b/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/GeneratorFactory.java index e44fcd6e..ce5b4e85 100644 --- a/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/GeneratorFactory.java +++ b/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/GeneratorFactory.java @@ -15,6 +15,7 @@ */ package com.mybatisflex.codegen.generator; +import com.mybatisflex.codegen.constant.GenTypeConst; import com.mybatisflex.codegen.generator.impl.*; import java.util.Collection; @@ -26,13 +27,17 @@ public class GeneratorFactory { private static final Map generators = new HashMap<>(); static { - registerGenerator("entity", new EntityGenerator()); - registerGenerator("mapper", new MapperGenerator()); - registerGenerator("service", new ServiceGenerator()); - registerGenerator("serviceImpl", new ServiceImplGenerator()); - registerGenerator("controller", new ControllerGenerator()); - registerGenerator("tableDef", new TableDefGenerator()); - registerGenerator("mapperXml", new MapperXmlGenerator()); + registerGenerator(GenTypeConst.ENTITY, new EntityGenerator()); + registerGenerator(GenTypeConst.MAPPER, new MapperGenerator()); + registerGenerator(GenTypeConst.SERVICE, new ServiceGenerator()); + registerGenerator(GenTypeConst.SERVICE_IMPL, new ServiceImplGenerator()); + registerGenerator(GenTypeConst.CONTROLLER, new ControllerGenerator()); + registerGenerator(GenTypeConst.TABLE_DEF, new TableDefGenerator()); + registerGenerator(GenTypeConst.MAPPER_XML, new MapperXmlGenerator()); + } + + public static IGenerator getGenerator(String genType) { + return generators.get(genType); } diff --git a/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/IGenerator.java b/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/IGenerator.java index 5f9e57fe..4d653e85 100644 --- a/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/IGenerator.java +++ b/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/IGenerator.java @@ -18,6 +18,34 @@ package com.mybatisflex.codegen.generator; import com.mybatisflex.codegen.config.GlobalConfig; import com.mybatisflex.codegen.entity.Table; +/** + * 文件生成器接口。 + * + * @author Michael Yang + * @author 王帅 + */ public interface IGenerator { + + /** + * 获取模板文件位置。 + * + * @return 路径 + */ + String getTemplatePath(); + + /** + * 设置模板文件位置。 + * + * @param templatePath + */ + void setTemplatePath(String templatePath); + + /** + * 根据模板生成文件。 + * + * @param table 表内容 + * @param globalConfig 全局配置 + */ void generate(Table table, GlobalConfig globalConfig); + } diff --git a/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/ControllerGenerator.java b/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/ControllerGenerator.java index cc03dc06..885cc130 100644 --- a/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/ControllerGenerator.java +++ b/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/ControllerGenerator.java @@ -34,7 +34,7 @@ import java.util.Map; */ public class ControllerGenerator implements IGenerator { - private final String templatePath; + private String templatePath; public ControllerGenerator() { this(TemplateConst.CONTROLLER); @@ -70,6 +70,15 @@ public class ControllerGenerator implements IGenerator { params.put("javadocConfig", globalConfig.getJavadocConfig()); params.put("controllerConfig", globalConfig.getControllerConfig()); - strategyConfig.getTemplateEngine().generate(params, templatePath, controllerJavaFile); + globalConfig.getTemplateConfig().getTemplate().generate(params, templatePath, controllerJavaFile); } + + public String getTemplatePath() { + return templatePath; + } + + public void setTemplatePath(String templatePath) { + this.templatePath = templatePath; + } + } diff --git a/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/EntityGenerator.java b/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/EntityGenerator.java index 5e0c56a6..e2a11003 100644 --- a/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/EntityGenerator.java +++ b/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/EntityGenerator.java @@ -34,7 +34,7 @@ import java.util.Map; */ public class EntityGenerator implements IGenerator { - private final String templatePath; + private String templatePath; public EntityGenerator() { this(TemplateConst.ENTITY); @@ -70,6 +70,15 @@ public class EntityGenerator implements IGenerator { params.put("entityConfig", globalConfig.getEntityConfig()); params.put("javadocConfig", globalConfig.getJavadocConfig()); - strategyConfig.getTemplateEngine().generate(params, templatePath, entityJavaFile); + globalConfig.getTemplateConfig().getTemplate().generate(params, templatePath, entityJavaFile); } + + public String getTemplatePath() { + return templatePath; + } + + public void setTemplatePath(String templatePath) { + this.templatePath = templatePath; + } + } diff --git a/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/MapperGenerator.java b/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/MapperGenerator.java index 2511e3bc..6720a538 100644 --- a/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/MapperGenerator.java +++ b/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/MapperGenerator.java @@ -34,7 +34,7 @@ import java.util.Map; */ public class MapperGenerator implements IGenerator { - private final String templatePath; + private String templatePath; public MapperGenerator() { this(TemplateConst.MAPPER); @@ -70,6 +70,15 @@ public class MapperGenerator implements IGenerator { params.put("mapperConfig", globalConfig.getMapperConfig()); params.put("javadocConfig", globalConfig.getJavadocConfig()); - strategyConfig.getTemplateEngine().generate(params, templatePath, mapperJavaFile); + globalConfig.getTemplateConfig().getTemplate().generate(params, templatePath, mapperJavaFile); } + + public String getTemplatePath() { + return templatePath; + } + + public void setTemplatePath(String templatePath) { + this.templatePath = templatePath; + } + } diff --git a/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/MapperXmlGenerator.java b/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/MapperXmlGenerator.java index 2713c721..d309485d 100644 --- a/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/MapperXmlGenerator.java +++ b/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/MapperXmlGenerator.java @@ -34,7 +34,7 @@ import java.util.Map; */ public class MapperXmlGenerator implements IGenerator { - private final String templatePath; + private String templatePath; public MapperXmlGenerator() { this(TemplateConst.MAPPER_XML); @@ -67,6 +67,15 @@ public class MapperXmlGenerator implements IGenerator { params.put("table", table); params.put("packageConfig", packageConfig); - strategyConfig.getTemplateEngine().generate(params, templatePath, mapperXmlFile); + globalConfig.getTemplateConfig().getTemplate().generate(params, templatePath, mapperXmlFile); } + + public String getTemplatePath() { + return templatePath; + } + + public void setTemplatePath(String templatePath) { + this.templatePath = templatePath; + } + } diff --git a/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/ServiceGenerator.java b/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/ServiceGenerator.java index 1a479112..d2242d00 100644 --- a/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/ServiceGenerator.java +++ b/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/ServiceGenerator.java @@ -34,7 +34,7 @@ import java.util.Map; */ public class ServiceGenerator implements IGenerator { - private final String templatePath; + private String templatePath; public ServiceGenerator() { this(TemplateConst.SERVICE); @@ -70,6 +70,15 @@ public class ServiceGenerator implements IGenerator { params.put("serviceConfig", globalConfig.getServiceConfig()); params.put("javadocConfig", globalConfig.getJavadocConfig()); - strategyConfig.getTemplateEngine().generate(params, templatePath, serviceJavaFile); + globalConfig.getTemplateConfig().getTemplate().generate(params, templatePath, serviceJavaFile); } + + public String getTemplatePath() { + return templatePath; + } + + public void setTemplatePath(String templatePath) { + this.templatePath = templatePath; + } + } diff --git a/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/ServiceImplGenerator.java b/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/ServiceImplGenerator.java index 13f921b4..ebe9b2e5 100644 --- a/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/ServiceImplGenerator.java +++ b/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/ServiceImplGenerator.java @@ -34,7 +34,7 @@ import java.util.Map; */ public class ServiceImplGenerator implements IGenerator { - private final String templatePath; + private String templatePath; public ServiceImplGenerator() { this(TemplateConst.SERVICE_IMPL); @@ -70,6 +70,15 @@ public class ServiceImplGenerator implements IGenerator { params.put("javadocConfig", globalConfig.getJavadocConfig()); params.put("serviceImplConfig", globalConfig.getServiceImplConfig()); - strategyConfig.getTemplateEngine().generate(params, templatePath, serviceImplJavaFile); + globalConfig.getTemplateConfig().getTemplate().generate(params, templatePath, serviceImplJavaFile); } + + public String getTemplatePath() { + return templatePath; + } + + public void setTemplatePath(String templatePath) { + this.templatePath = templatePath; + } + } diff --git a/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/TableDefGenerator.java b/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/TableDefGenerator.java index ebb8dc70..f35e1402 100644 --- a/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/TableDefGenerator.java +++ b/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/TableDefGenerator.java @@ -34,7 +34,7 @@ import java.util.Map; */ public class TableDefGenerator implements IGenerator { - private final String templatePath; + private String templatePath; public TableDefGenerator() { this(TemplateConst.TABLE_DEF); @@ -70,6 +70,15 @@ public class TableDefGenerator implements IGenerator { params.put("javadocConfig", globalConfig.getJavadocConfig()); params.put("tableDefConfig", globalConfig.getTableDefConfig()); - strategyConfig.getTemplateEngine().generate(params, templatePath, tableDefJavaFile); + globalConfig.getTemplateConfig().getTemplate().generate(params, templatePath, tableDefJavaFile); } + + public String getTemplatePath() { + return templatePath; + } + + public void setTemplatePath(String templatePath) { + this.templatePath = templatePath; + } + } diff --git a/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/template/EnjoyTemplate.java b/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/template/impl/EnjoyTemplate.java similarity index 67% rename from mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/template/EnjoyTemplate.java rename to mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/template/impl/EnjoyTemplate.java index d1e92323..3ea8fbe4 100644 --- a/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/template/EnjoyTemplate.java +++ b/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/template/impl/EnjoyTemplate.java @@ -13,10 +13,15 @@ * See the License for the specific language governing permissions and * limitations under the License. */ -package com.mybatisflex.codegen.template; +package com.mybatisflex.codegen.template.impl; import com.jfinal.template.Engine; import com.jfinal.template.expr.ast.FieldGetters; +import com.jfinal.template.source.ClassPathSource; +import com.jfinal.template.source.FileSource; +import com.jfinal.template.source.ISource; +import com.jfinal.template.source.ISourceFactory; +import com.mybatisflex.codegen.template.ITemplate; import com.mybatisflex.core.util.StringUtil; import java.io.File; @@ -29,8 +34,8 @@ public class EnjoyTemplate implements ITemplate { public EnjoyTemplate() { engine = Engine.create("mybatis-flex", engine -> { - engine.setToClassPathSourceFactory(); engine.addSharedMethod(StringUtil.class); + engine.setSourceFactory(new FileAndClassPathSourceFactory()); }); // 以下配置将支持 user.girl 表达式去调用 user 对象的 boolean isGirl() 方法 Engine.addFieldGetterToFirst(new FieldGetters.IsMethodFieldGetter()); @@ -48,4 +53,18 @@ public class EnjoyTemplate implements ITemplate { e.printStackTrace(); } } + + public static class FileAndClassPathSourceFactory implements ISourceFactory { + + @Override + public ISource getSource(String baseTemplatePath, String fileName, String encoding) { + // 先从文件寻找资源，找不到再从类路径寻找资源 + if (new File(fileName).exists()) { + return new FileSource(baseTemplatePath, fileName, encoding); + } + return new ClassPathSource(baseTemplatePath, fileName, encoding); + } + + } + }