Merge pull request #220 from haumacher/no-multiple-initializations

Superfast baking with Thymeleaf
jbake-org · Jun 24, 2016 · 6e4a165 · 6e4a165
2 parents 809c197 + f5838ab
commit 6e4a165
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/src/main/java/org/jbake/template/ThymeleafTemplateEngine.java b/src/main/java/org/jbake/template/ThymeleafTemplateEngine.java
@@ -40,15 +40,21 @@ public class ThymeleafTemplateEngine extends AbstractTemplateEngine {
     private TemplateEngine templateEngine;
     private FileTemplateResolver templateResolver;
 
+	private String templateMode;
+
     public ThymeleafTemplateEngine(final CompositeConfiguration config, final ContentStore db, final File destination, final File templatesPath) {
         super(config, db, destination, templatesPath);
-        initializeTemplateEngine();
     }
 
-    private void initializeTemplateEngine() {
+    private void initializeTemplateEngine(String mode) {
+    	if (mode.equals(templateMode)) {
+    		return;
+    	}
+        templateMode = mode;
         templateResolver = new FileTemplateResolver();
         templateResolver.setPrefix(templatesPath.getAbsolutePath() + File.separatorChar);
         templateResolver.setCharacterEncoding(config.getString(Keys.TEMPLATE_ENCODING));
+        templateResolver.setTemplateMode(mode);
         templateEngine = new TemplateEngine();
         templateEngine.setTemplateResolver(templateResolver);
         try {
@@ -66,7 +72,6 @@ public void renderDocument(final Map<String, Object> model, final String templat
         Context context = new Context(locale, wrap(model));
         lock.lock();
         try {
-        	initializeTemplateEngine();
             @SuppressWarnings("unchecked")
             Map<String, Object> config = (Map<String, Object>) model.get("config");
             @SuppressWarnings("unchecked")
@@ -79,7 +84,7 @@ public void renderDocument(final Map<String, Object> model, final String templat
                     mode = configMode;
                 }
             }
-            templateResolver.setTemplateMode(mode);
+            initializeTemplateEngine(mode);
             templateEngine.process(templateName, context, writer);
         } finally {
             lock.unlock();