com.clickntap.smart.SmartView Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of Stripecube Show documentation
Stripecube is an open source Java framework for Web Applications
There is a newer version: 1.30
package com.clickntap.smart;

import com.clickntap.tool.bean.BeanUtils;
import com.clickntap.tool.script.ScriptEngine;
import com.clickntap.utils.ConstUtils;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.springframework.web.servlet.View;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.ByteArrayOutputStream;
import java.io.OutputStream;
import java.lang.reflect.Method;
import java.util.Map;

public class SmartView implements View {
    public final static String CACHE_CONTROL = "Cache-control";
    public final static String PRAGMA = "Pragma";
    public final static String NO_CACHE = "no-cache";
    private static Log log = LogFactory.getLog(SmartView.class);
    private String contentType;
    private Mode mode;
    private ScriptEngine viewEngine;
    private String binaryKey;

    public void setMode(Mode mode) {
        this.mode = mode;
    }

    public void setBinaryKey(String binaryKey) {
        this.binaryKey = binaryKey;
    }

    public void render(Map model, HttpServletRequest request, HttpServletResponse response) throws Exception {
        Object self = model.get(ConstUtils.THIS);
        SmartContext ctx = null;
        if (self instanceof SmartContext)
            ctx = (SmartContext) model.get(ConstUtils.THIS);
        else {
            ctx = new SmartContext(request, response);
            ctx.put(ConstUtils.THIS, self);
        }
        if (SmartCache.isCached(ctx)) {
            SmartCache.handleResponse(ctx);
            return;
        }
        OutputStream out = response.getOutputStream();
        if (SmartCache.cacheControl(ctx))
            out = new ByteArrayOutputStream();
        try {
            switch (mode) {
                case TEXT:
                    response.setHeader(CACHE_CONTROL, NO_CACHE);
                    response.setHeader(PRAGMA, NO_CACHE);
                    response.setContentType(getContentType());
                    response.setCharacterEncoding(ConstUtils.UTF_8);
                    if (ctx.getException() != null)
                        throw ctx.getException();
                    viewEngine.eval(ctx, ctx.getRef(), out);
                    break;
                case BINARY:
                    Object outputBean = ctx.get(binaryKey);
                    String contentType = null;
                    try {
                        contentType = BeanUtils.getValue(outputBean, "contentType").toString();
                    } catch (Exception e) {
                        contentType = null;
                    }
                    if (contentType == null)
                        response.setContentType(getContentType());
                    else {
                        if (contentType.contains("UTF-8")) {
                            response.setCharacterEncoding(ConstUtils.UTF_8);
                        }
                        response.setContentType(contentType);
                    }
                    if (ctx.getException() != null)
                        throw ctx.getException();
                    if (outputBean != null) {
                        Method method = outputBean.getClass().getMethod("copyTo", OutputStream.class);
                        if (method != null)
                            method.invoke(outputBean, out);
                    }
                    break;
            }
        } catch (Exception e) {
            if (log.isErrorEnabled()) {
                ctx.getSmartApp().addError(e);
                log.error("template error", e);
            }
            if (ctx.getException() == null)
                ctx.setException(e);
            ctx.put(ConstUtils.THIS, ctx);
            viewEngine.eval(ctx, ctx.getController().getErrorRef(), out);
        }
        if (out instanceof ByteArrayOutputStream)
            SmartCache.handleResponse(ctx, (ByteArrayOutputStream) out);
    }

    public void setViewEngine(ScriptEngine viewEngine) {
        this.viewEngine = viewEngine;
    }

    public String getContentType() {
        return contentType;
    }

    public void setContentType(String contentType) {
        this.contentType = contentType;
    }

    public enum Mode {
        TEXT, BINARY
    }
}