com.firefly.template.TemplateFactory Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of firefly-template Show documentation
There is a newer version: 4.0.20
package com.firefly.template;

import com.firefly.template.function.*;
import com.firefly.template.parser.ViewFileReader;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.io.BufferedInputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

public class TemplateFactory {

	private static Logger log = LoggerFactory.getLogger("firefly-system");

	private Config config;
	private Map map = new HashMap();

	public TemplateFactory() {

	}
	
	public TemplateFactory(File file) {
		this.config = new Config();
		config.setViewPath(file.getAbsolutePath());
	}

	public TemplateFactory(String path) {
		this.config = new Config();
		config.setViewPath(path);
	}

	public TemplateFactory(Config config) {
		this.config = config;
	}

	public Config getConfig() {
		return config;
	}

	public void setConfig(Config config) {
		this.config = config;
	}
	
	public TemplateFactory init() {
		if(config == null)
			throw new IllegalArgumentException("template config is null");
		
		long start = System.currentTimeMillis();
		FunctionRegistry.add("dateFormat", new DateFormatFunction(config.getCharset()));
		FunctionRegistry.add("len", new LengthFunction(config.getCharset()));
		FunctionRegistry.add("increment", new AutoincrementFunction(config.getCharset()));
		FunctionRegistry.add("cut", new CutStringFunction(config.getCharset()));
		FunctionRegistry.add("escape", new XmlEscapeFunction(config.getCharset()));
		FunctionRegistry.add("remove", new RemoveAttributeFunction());
		FunctionRegistry.add("modOut", new ModuloOutputFunction(config.getCharset()));
		
		ViewFileReader reader = new ViewFileReader(config);
		List javaFiles = reader.getJavaFiles();
		List templateFiles = reader.getTemplateFiles();
		List classNames = reader.getClassNames();
		
		for (int i = 0; i < javaFiles.size(); i++) {
			String c = javaFiles.get(i);
			final String classFileName = c.substring(0, c.length() - 4) + "class";
			ClassLoader classLoader = new TemplateClassLoader(classFileName, TemplateFactory.class.getClassLoader());
			
			try {
				map.put(templateFiles.get(i), (View)classLoader.loadClass(classNames.get(i)).getConstructor(TemplateFactory.class).newInstance(this));
			} catch (Throwable e) {
				log.error("load class error", e);
			}
		}
		long end = System.currentTimeMillis();
		log.info("template initialization spends time in {} ms", (end - start));
		return this;
	}
	
	public View getView(String name) {
		return map.get(name);
	}
	
	private class TemplateClassLoader extends ClassLoader {
		private String classFileName;
		
		public TemplateClassLoader(String classFileName, ClassLoader classLoader) {
			super(classLoader);
			this.classFileName = classFileName;
		}
		
		@Override
		public Class findClass(String name) {
			BufferedInputStream bis = null;
			byte[] b = null;
			try {
				File file = new File(classFileName);
				b = new byte[(int)file.length()];
				bis = new BufferedInputStream(new FileInputStream(file));
				bis.read(b);
			} catch (Throwable e) {
				log.error("read class file error", e);
			} finally {
				if(bis != null)
					try {
						bis.close();
					} catch (IOException e) {
						log.error("close error", e);
					}
			}
			
			return defineClass(name, b, 0, b.length);
		}
	}

}