org.nutz.web.ajax.AjaxView Maven / Gradle / Ivy

Go to download
package org.nutz.web.ajax;

import java.io.IOException;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import org.nutz.json.JsonFormat;
import org.nutz.lang.Strings;
import org.nutz.mvc.Mvcs;
import org.nutz.mvc.View;
import org.nutz.web.WebException;
import org.nutz.web.Webs;

/**
 * 将返回渲染成标准 AjaxReturn
 * 
 * 可以在配置文件中设置js的格式，适用开发与生产环境
 * 
 * @author zozoh([email protected])
 * @author pw
 */
public class AjaxView implements View {

    private boolean uc;

    public AjaxView() {
        uc = false;
    }

    public AjaxView(boolean uc) {
        this.uc = uc;
    }

    public AjaxView(String useCompact) {
        uc = Strings.isBlank(useCompact) ? false : Boolean.parseBoolean(useCompact);
    }

    public JsonFormat getJsonFormat() {
        if (uc) {
            // 紧凑模式(生产)
            return JsonFormat.compact();
        } else {
            // 一般模式(开发)
            return JsonFormat.nice();
        }
    }

    @Override
    public void render(HttpServletRequest req, HttpServletResponse resp, Object obj)
            throws IOException {
        AjaxReturn re;
        // 空
        if (null == obj) {
            re = Ajax.ok();
        }
        // 异常
        else if (obj instanceof Throwable) {
            WebException err = Webs.Err.wrap((Throwable) obj);
            String msg = err.getKey();
            try {
                msg = Mvcs.getMessage(req, err.getKey());
            }
            catch (Exception e) {}
            re = Ajax.fail().setErrCode(err.getKey()).setMsg(msg).setData(err.getReason());
        }
        // AjaxReturn
        else if (obj instanceof AjaxReturn) {
            re = (AjaxReturn) obj;
        }
        // 数据对象
        else {
            re = Ajax.ok().setData(obj);
        }

        // 写入返回
        Mvcs.write(resp, re, getJsonFormat());
    }

}