com.khubla.pragmatach.plugin.velocity.VelocityResponse Maven / Gradle / Ivy

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

import java.io.ByteArrayInputStream;
import java.io.Reader;
import java.io.StringReader;
import java.io.StringWriter;
import java.io.Writer;
import java.util.Map;

import javax.servlet.http.HttpServletResponse;

import org.apache.commons.io.IOUtils;
import org.apache.velocity.VelocityContext;
import org.apache.velocity.app.VelocityEngine;

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

/**
 * @author tome
 */
public class VelocityResponse extends AbstractResponse {
   /**
    * template
    */
   private final String template;
   /**
    * templatename
    */
   private final String templateName;
   /**
    * context
    */
   private final Map context;
   /**
    * velocity
    */
   private final VelocityEngine velocityEngine = new VelocityEngine();

   public VelocityResponse(Map cacheHeaders, String templateName, String template, Map context) {
      super(cacheHeaders);
      this.template = template;
      this.context = context;
      this.templateName = templateName;
   }

   @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 VelocityContext velocityContext = new VelocityContext(context);
         final Reader reader = new StringReader(template);
         final Writer writer = new StringWriter();
         velocityEngine.evaluate(velocityContext, writer, templateName, reader);
         final ByteArrayInputStream bais = new ByteArrayInputStream(writer.toString().getBytes());
         IOUtils.copy(bais, httpServletResponse.getOutputStream());
      } catch (final Exception e) {
         throw new PragmatachException("Exception in render", e);
      }
   }
}