style: 添加 EnjoyTemplate 注释。

This commit is contained in:
Suomm 2023-06-21 22:17:26 +08:00
parent ab2c5b706a
commit 933e4dfb48

View File

@ -1,17 +1,17 @@
/** /*
* Copyright (c) 2022-2023, Mybatis-Flex (fuhai999@gmail.com). * Copyright (c) 2022-2023, Mybatis-Flex (fuhai999@gmail.com).
* <p> * <p>
* Licensed under the Apache License, Version 2.0 (the "License"); * Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License. * you may not use this file except in compliance with the License.
* You may obtain a copy of the License at * You may obtain a copy of the License at
* <p> * <p>
* http://www.apache.org/licenses/LICENSE-2.0 * http://www.apache.org/licenses/LICENSE-2.0
* <p> * <p>
* Unless required by applicable law or agreed to in writing, software * Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS, * distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and * See the License for the specific language governing permissions and
* limitations under the License. * limitations under the License.
*/ */
package com.mybatisflex.codegen.template.impl; package com.mybatisflex.codegen.template.impl;
@ -28,6 +28,9 @@ import java.io.File;
import java.io.FileOutputStream; import java.io.FileOutputStream;
import java.util.Map; import java.util.Map;
/**
* <a href=https://jfinal.com/doc/6-1>JFinal Enjoy</a> 模板引擎实现
*/
public class EnjoyTemplate implements ITemplate { public class EnjoyTemplate implements ITemplate {
private final Engine engine; private final Engine engine;
@ -43,17 +46,22 @@ public class EnjoyTemplate implements ITemplate {
@Override @Override
public void generate(Map<String, Object> params, String templateFilePath, File generateFile) { public void generate(Map<String, Object> params, String templateFilePath, File generateFile) {
if (!generateFile.getParentFile().exists() && !generateFile.getParentFile().mkdirs()){ if (!generateFile.getParentFile().exists() && !generateFile.getParentFile().mkdirs()) {
throw new IllegalStateException("Can not mkdirs by dir: " + generateFile.getParentFile()); throw new IllegalStateException("Can not mkdirs by dir: " + generateFile.getParentFile());
} }
// 开始生成文件
try(FileOutputStream fileOutputStream = new FileOutputStream(generateFile)) { try (FileOutputStream fileOutputStream = new FileOutputStream(generateFile)) {
engine.getTemplate(templateFilePath).render(params, fileOutputStream); engine.getTemplate(templateFilePath).render(params, fileOutputStream);
} catch (Exception e) { } catch (Exception e) {
e.printStackTrace(); e.printStackTrace();
} }
} }
/**
* 从文件或者类路径读取模板
*
* @author 王帅
*/
public static class FileAndClassPathSourceFactory implements ISourceFactory { public static class FileAndClassPathSourceFactory implements ISourceFactory {
@Override @Override