com.baomidou.framework.controller.SuperController Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of spring-wind Show documentation
spring-wind extension of spring framework.
The newest version!
/**
 * Copyright (c) 2011-2014, hubin ([email protected]).
 *
 * Licensed under the Apache License, Version 2.0 (the "License"); you may not
 * use this file except in compliance with the License. You may obtain a copy of
 * the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
 * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
 * License for the specific language governing permissions and limitations under
 * the License.
 */
package com.baomidou.framework.controller;

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

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;

import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.serializer.SerializerFeature;
import com.baomidou.framework.exception.WebException;
import com.baomidou.kisso.SSOHelper;
import com.baomidou.kisso.SSOToken;
import com.baomidou.kisso.common.util.HttpUtil;
import com.baomidou.mybatisplus.plugins.Page;

/**
 * 
 * Controller 基础类（支持 kisso）
 * 
 * 
 * @author hubin
 * @Date 2016-03-15
 */
public class SuperController {

	protected Logger logger = LoggerFactory.getLogger(this.getClass());

	@Autowired
	protected HttpServletRequest request;

	@Autowired
	protected HttpServletResponse response;

	@Autowired
	protected HttpSession session;

	@Autowired
	protected ServletContext application;

	/**
	 * 用户ID
	 */
	protected Long getCurrentUserId() {
		return getSSOToken().getId();
	}

	/**
	 * 返回登录 Token
	 */
	protected SSOToken getSSOToken() {
		SSOToken tk = SSOHelper.attrToken(request);
		if (tk == null) {
			throw new WebException("-1", "The user does not exist, please relogin.");
		}
		return tk;
	}

	/**
	 * 是否为 post 请求
	 */
	protected boolean isPost() {
		return HttpUtil.isPost(request);
	}

	/**
	 * 是否为 get 请求
	 */
	protected boolean isGet() {
		return HttpUtil.isGet(request);
	}

	/**
	 * 
	 * 获取分页对象
	 * 
	 */
	protected  Page getPage() {
		return getPage(10);
	}

	/**
	 * 
	 * 获取分页对象
	 * 
	 * 
	 * @param size
	 *            每页显示数量
	 * @return
	 */
	protected  Page getPage(int size) {
		int _size = size, _index = 1;
		if (request.getParameter("_size") != null) {
			_size = Integer.parseInt(request.getParameter("_size"));
		}
		if (request.getParameter("_index") != null) {
			_index = Integer.parseInt(request.getParameter("_index"));
		}
		return new Page(_index, _size);
	}

	/**
	 * 重定向至地址 url
	 * 
	 * @param url
	 *            请求地址
	 * @return
	 */
	protected String redirectTo(String url) {
		StringBuffer rto = new StringBuffer("redirect:");
		rto.append(url);
		return rto.toString();
	}

	/**
	 * 
	 * 返回 JSON 格式对象
	 * 
	 * @param object
	 *            转换对象
	 * @return
	 */
	protected String toJson(Object object) {
		return JSON.toJSONString(object, SerializerFeature.BrowserCompatible);
	}

	/**
	 * 
	 * 返回 JSON 格式对象
	 * 
	 * @param object
	 *            转换对象
	 * @param features
	 *            序列化特点
	 * @return
	 */
	protected String toJson(Object object, String format) {
		if (format == null) {
			return toJson(object);
		}
		return JSON.toJSONStringWithDateFormat(object, format, SerializerFeature.WriteDateUseDateFormat);
	}

	/**
	 * 
	 * 自动判定是否有跨域操作,转成字符串并返回
	 * 
	 *
	 * @param object
	 * @return 跨域或不跨域的字符串
	 */
	protected String callback(AjaxResult object) {
		return callback(object, null);
	}

	protected String callback(AjaxResult object, String format) {
		String callback = request.getParameter("callback");
		if (callback == null) {
			/**
			 * 非 JSONP 请求
			 */
			return toJson(object, format);
		}
		StringBuffer json = new StringBuffer();
		json.append(callback);
		json.append("(").append(toJson(object, format)).append(")");
		return json.toString();
	}

	protected String callbackSuccess(Object obj) {
		return callback(new AjaxResult(obj));
	}

	protected String callbackFail(String message) {
		return callback(new AjaxResult(false, message));
	}

}