play.template2.GTTemplateInstanceFactoryLive Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of fastergt Show documentation
RePlay is a fork of the Play1 framework, made and maintained by Codeborne.
The newest version!
package play.template2;


import play.template2.compile.GTCompiler;
import play.template2.compile.GTJavaCompileToClass;
import play.template2.exceptions.GTException;

import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.net.URL;
import java.util.*;

public class GTTemplateInstanceFactoryLive extends GTTemplateInstanceFactory {

    private final Class templateClass;

    public static class CL extends ClassLoader {

        private final Set classNames;
        private final Map resource2bytes;
        private final ClassLoader parent = Thread.currentThread().getContextClassLoader();

        public CL(GTJavaCompileToClass.CompiledClass[] compiledClasses) {
            this.classNames = new HashSet<>(compiledClasses.length);
            this.resource2bytes = new HashMap<>(compiledClasses.length);

            for (GTJavaCompileToClass.CompiledClass cp : compiledClasses) {
                classNames.add(cp.classname);
                defineClass(cp.classname, cp.bytes, 0, cp.bytes.length);
                String resourceName = cp.classname.replace(".", "/") + ".class";
                resource2bytes.put(resourceName, cp.bytes);
            }
        }

        @Override
        public InputStream getResourceAsStream(String s) {
            if (resource2bytes.containsKey(s)) {
                return new ByteArrayInputStream(resource2bytes.get(s));
            } else {
                return parent.getResourceAsStream(s);
            }
        }

        @Override
        public Class loadClass(String s) throws ClassNotFoundException {
            if (!classNames.contains(s)) return parent.loadClass(s);
            return super.loadClass(s);
        }

        @Override
        public URL getResource(String s) {
            if ( !classNames.contains(s)) return parent.getResource(s);
            return super.getResource(s);
        }

        @Override
        public Enumeration getResources(String s) throws IOException {
            if ( !classNames.contains(s)) return parent.getResources(s);
            return super.getResources(s);
        }
    }

    public GTTemplateInstanceFactoryLive(GTCompiler.CompiledTemplate compiledTemplate) {
        CL cl = new CL(compiledTemplate.compiledJavaClasses);
        try {
            this.templateClass = (Class) cl.loadClass(compiledTemplate.templateClassName);
        } catch (Exception e) {
            throw new GTException("Error creating template class instance", e);
        }
    }

    @Override
    public Class getTemplateClass() {
        return templateClass;
    }
}