com.yammer.dropwizard.views.mustache.MustacheViewRenderer Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of dropwizard-views Show documentation

There is a newer version: 0.6.2

package com.yammer.dropwizard.views.mustache;

import com.github.mustachejava.Mustache;
import com.github.mustachejava.MustacheFactory;
import com.google.common.base.Charsets;
import com.google.common.cache.CacheBuilder;
import com.google.common.cache.CacheLoader;
import com.google.common.cache.LoadingCache;
import com.yammer.dropwizard.views.View;
import com.yammer.dropwizard.views.ViewRenderer;

import javax.ws.rs.WebApplicationException;
import java.io.IOException;
import java.io.OutputStream;
import java.io.OutputStreamWriter;
import java.util.Locale;

public class MustacheViewRenderer implements ViewRenderer {
    private final LoadingCache, MustacheFactory> factories;

    public MustacheViewRenderer() {
        this.factories = CacheBuilder.newBuilder()
                                     .build(new CacheLoader, MustacheFactory>() {
                                         @Override
                                         public MustacheFactory load(Class key) throws Exception {
                                             return new CachingMustacheFactory(key);
                                         }
                                     });
    }

    @Override
    public boolean isRenderable(View view) {
        return view.getTemplateName().endsWith(".mustache");
    }

    @Override
    public void render(View view, Locale locale, OutputStream output) throws IOException, WebApplicationException {
        final OutputStreamWriter writer = new OutputStreamWriter(output, Charsets.UTF_8);
        try {
            final Mustache template = factories.getUnchecked(view.getClass())
                                               .compile(view.getTemplateName());
            template.execute(writer, view);
        } finally {
            writer.close();
        }
    }
}