com.fastchar.out.FastOutVelocity Maven / Gradle / Ivy

package com.fastchar.out;

import com.fastchar.core.FastAction;
import com.fastchar.core.FastChar;
import org.apache.velocity.Template;
import org.apache.velocity.VelocityContext;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.PrintWriter;
import java.util.Enumeration;
import java.util.Map;

public class FastOutVelocity extends FastOut {

    public FastOutVelocity() {
        this.contentType = "text/html";
    }

    /**
     * 响应数据
     * @param action
     */
    @Override
    public void response(FastAction action) throws Exception {
        if (String.valueOf(data).toLowerCase().endsWith(".xml")) {
            this.contentType = "text/xml";
        }

        HttpServletResponse response = action.getResponse();
        HttpServletRequest request = action.getRequest();
        response.setContentType(toContentType());
        response.setCharacterEncoding(getCharset());

        Template template = FastChar.getTemplates().getVelocity().getTemplate(String.valueOf(data));

        VelocityContext context = new VelocityContext();
        Map finalContext = FastChar.getTemplates().getFinalContext();
        for (String key : finalContext.keySet()) {
            context.put(key, finalContext.get(key));
        }

        for (Enumeration attrs = request.getAttributeNames(); attrs.hasMoreElements();) {
            String attrName = attrs.nextElement();
            context.put(attrName, request.getAttribute(attrName));
        }
        for (Enumeration attrs = request.getSession().getAttributeNames(); attrs.hasMoreElements();) {
            String attrName = attrs.nextElement();
            context.put(attrName, request.getSession().getAttribute(attrName));
        }
        PrintWriter writer = response.getWriter();
        template.merge(context, writer);
        writer.flush();
    }
}