com.fastchar.extend.velocity.FastVelocityEngine Maven / Gradle / Ivy

package com.fastchar.extend.velocity;

import com.fastchar.annotation.AFastClassFind;
import com.fastchar.core.FastChar;
import com.fastchar.exception.FastTemplateException;
import com.fastchar.interfaces.IFastTemplate;
import com.fastchar.utils.FastFileUtils;
import com.fastchar.utils.FastStringUtils;
import org.apache.velocity.VelocityContext;
import org.apache.velocity.app.Velocity;
import org.apache.velocity.app.VelocityEngine;

import java.io.File;
import java.io.IOException;
import java.io.StringWriter;
import java.util.Map;

@AFastClassFind(value = {"org.apache.velocity.app.VelocityEngine"}, url = {"https://mvnrepository.com/artifact/org.apache.velocity/velocity-engine-core"})
public class FastVelocityEngine extends VelocityEngine implements IFastTemplate {

    public FastVelocityEngine() {
        setProperty(Velocity.FILE_RESOURCE_LOADER_PATH, FastChar.getPath().getWebRootPath());
        setProperty(Velocity.ENCODING_DEFAULT, FastChar.getConstant().getEncoding());
        setProperty(Velocity.INPUT_ENCODING, FastChar.getConstant().getEncoding());
        setProperty(Velocity.ENCODING_DEFAULT, FastChar.getConstant().getEncoding());
        setProperty(Velocity.VM_LIBRARY_AUTORELOAD, false);

        setProperty(Velocity.RESOURCE_MANAGER_DEFAULTCACHE_SIZE, 0);
        if (FastChar.getConstant().isDebug()) {
            setProperty(Velocity.RESOURCE_LOADER_CACHE, false);
        } else {
            setProperty(Velocity.RESOURCE_LOADER_CACHE, true);
            setProperty(Velocity.RESOURCE_LOADER_CHECK_INTERVAL, 3600);
        }
    }


    @Override
    public String run(Map params, String template) {
        try {
            if (params == null || FastStringUtils.isEmpty(template)) {
                return null;
            }
            VelocityContext context = new VelocityContext();
            params.putAll(FastChar.getTemplates().getFinalContext());
            for (String key : params.keySet()) {
                context.put(key, params.get(key));
            }

            StringWriter writer = new StringWriter();
            evaluate(context, writer, "FastChar", template);
            String data = writer.toString();
            writer.close();
            return data;
        } catch (IOException e) {
            e.printStackTrace();
        }
        return null;
    }

    @Override
    public String run(Map params, File template) {
        if (params == null || template == null || !template.exists()) {
            return null;
        }
        try {
            return run(params, FastFileUtils.readFileToString(template));
        } catch (IOException e) {
            throw new FastTemplateException(e);
        }
    }
}