com.fastchar.extend.thymeleaf.FastThymeleafEngine Maven / Gradle / Ivy

Go to download
package com.fastchar.extend.thymeleaf;

import com.fastchar.core.FastChar;
import com.fastchar.interfaces.IFastTemplate;
import com.fastchar.utils.FastStringUtils;
import org.thymeleaf.TemplateEngine;
import org.thymeleaf.context.Context;
import org.thymeleaf.templateresolver.StringTemplateResolver;

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

public class FastThymeleafEngine extends TemplateEngine implements IFastTemplate {

    private FastTemplateResolver fastTemplateResolver;

    public FastThymeleafEngine() {
        fastTemplateResolver = new FastTemplateResolver(FastChar.getServletContext());
        setTemplateResolver(fastTemplateResolver);
        addTemplateResolver(new StringTemplateResolver());
    }

    public FastTemplateResolver getFastTemplateResolver() {
        return fastTemplateResolver;
    }


    public String run(Context context, String filePath) {
        try {
            if (context == null || FastStringUtils.isEmpty(filePath)) {
                return null;
            }
            Map finalContext = FastChar.getTemplates().getFinalContext();
            for (String key : finalContext.keySet()) {
                context.setVariable(key, finalContext.get(key));
            }
            StringWriter writer = new StringWriter();
            process(filePath, context, writer);
            String data = writer.toString();
            writer.close();
            return data;
        } catch (Exception e) {
            e.printStackTrace();
        }
        return null;
    }

    @Override
    public String run(Map params, String template) {
        try {
            if (params == null || FastStringUtils.isEmpty(template)) {
                return null;
            }
            Context context = new Context();
            params.putAll(FastChar.getTemplates().getFinalContext());
            for (String key : params.keySet()) {
                context.setVariable(key, params.get(key));
            }
            StringWriter writer = new StringWriter();
            process(template, context, writer);
            String data = writer.toString();
            writer.close();
            return data;
        } catch (Exception e) {
            e.printStackTrace();
        }
        return null;
    }

    @Override
    public String run(Map params, File template) {
        if (params == null || template == null || !template.exists()) {
            return null;
        }
        Context context = new Context();
        params.putAll(FastChar.getTemplates().getFinalContext());
        for (String key : params.keySet()) {
            context.setVariable(key, params.get(key));
        }
        return run(context, template.getAbsolutePath());
    }

}