com.khubla.pragmatach.plugin.thymeleaf.ThymeleafResponse Maven / Gradle / Ivy

Go to download
package com.khubla.pragmatach.plugin.thymeleaf;

import java.io.OutputStreamWriter;
import java.io.Writer;
import java.util.Map;

import javax.servlet.http.HttpServletResponse;

import org.thymeleaf.TemplateEngine;
import org.thymeleaf.context.Context;
import org.thymeleaf.templateresolver.ITemplateResolver;

import com.khubla.pragmatach.framework.api.PragmatachException;
import com.khubla.pragmatach.framework.controller.impl.AbstractResponse;

/**
 * @author tome
 */
public class ThymeleafResponse extends AbstractResponse {
   /**
    * template
    */
   private final String templateName;
   /**
    * context
    */
   private final Map context;
   /**
    * resolver
    */
   private final ITemplateResolver templateResolver;

   public ThymeleafResponse(Map cacheHeaders, String templateName, Map context, ITemplateResolver templateResolver) {
      super(cacheHeaders);
      this.templateName = templateName;
      this.context = context;
      this.templateResolver = templateResolver;
   }

   @Override
   public String getContentType() throws PragmatachException {
      return CONTENT_TYPE_HTML;
   }

   @Override
   public Map getHeaders() throws PragmatachException {
      return super.getCacheHeaders();
   }

   @Override
   public void render(HttpServletResponse httpServletResponse) throws PragmatachException {
      try {
         final Writer writer = new OutputStreamWriter(httpServletResponse.getOutputStream());
         final TemplateEngine templateEngine = new TemplateEngine();
         templateEngine.setTemplateResolver(templateResolver);
         final Context ctx = new Context();
         ctx.setVariables(context);
         templateEngine.process(templateName, ctx, writer);
         writer.flush();
      } catch (final Exception e) {
         throw new PragmatachException("Exception in render", e);
      }
   }
}