com.diboot.framework.utils.BaseHelper Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of diboot-framework Show documentation
Diboot Framework
The newest version!
package com.diboot.framework.utils;

import com.diboot.framework.model.BaseUser;
import com.diboot.framework.model.JsonResult;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.apache.shiro.SecurityUtils;
import org.apache.shiro.subject.Subject;
import org.springframework.http.HttpStatus;
import org.springframework.http.MediaType;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.servlet.http.HttpSession;
import java.io.IOException;
import java.io.PrintWriter;
import java.util.*;

/***
 * 应用通用帮助类
 * @author [email protected]
 * @version 20161107
 *
 */
public class BaseHelper {
	private static final Logger logger = LoggerFactory.getLogger(BaseHelper.class);
	
	/**
	 * 得到当前登录的用户名
	 * @return 
	 */
	public static T getCurrentUser(){
		try{
			Subject subject = SecurityUtils.getSubject();
			if(subject != null && subject.isAuthenticated()){
				return (T)subject.getPrincipal();
			}
		}
		catch (Exception e){
			logger.warn("获取用户信息异常", e);
		}
		return null;
	}

	/**
	 * 得到当前登录的用户id
	 * @return 
	 */
	public static Long getCurrentUserId(){
		BaseUser user = getCurrentUser();
		if(user != null){
			return (Long)user.getId();
		}
		if(logger.isDebugEnabled()){
			logger.warn("无法获取当前用户Id!");
		}
		return null;
	}
	
	/**
	 * 是否来自微信
	 * @param request
	 * @return
	 */
	public static boolean isFromWechat(HttpServletRequest request){
		String userAgent = request.getHeader("user-agent").toLowerCase();
		return V.notEmpty(userAgent) && userAgent.contains("micromessenger");
	}

	/***
	 * 是否为AJAX请求
	 * @param request
	 * @return
	 */
	public static boolean isAjaxRequest(HttpServletRequest request){
		String header = request.getHeader("X-Requested-With");
		if (V.notEmpty(header) && "XMLHttpRequest".equals(header)){
			return true;
		}
		return false;
	}

	/**
	 * 获取Session中的对象
	 * @param request
	 * @param key
	 * @return
	 */
	public static Object getSessionObj(HttpServletRequest request, String key){
		HttpSession session = request.getSession(false);
		if(session != null && session.getAttribute(key) != null){
			return session.getAttribute(key);
		}
		return null;
	}

	/***
	 * 获取请求URI (去除contextPath)
	 * @param request
	 * @return
	 */
	public static String getRequestMappingURI(HttpServletRequest request){
		String contextPath = request.getContextPath();
		if(V.notEmpty(contextPath)){
			return S.replace(request.getRequestURI(), contextPath, "");
		}
		return request.getRequestURI();
	}

	/***
	 * 返回json格式错误信息
	 * @param response
	 * @param jsonResult
	 */
	public static void responseJson(HttpServletResponse response, JsonResult jsonResult){
		// 处理异步请求
		PrintWriter pw = null;
		try {
			response.setStatus(HttpStatus.OK.value());
			response.setContentType(MediaType.APPLICATION_JSON_UTF8_VALUE);
			pw = response.getWriter();
			pw.write(JSON.stringify(jsonResult));
			pw.flush();
		}
		catch (IOException e) {
			logger.error("处理异步请求异常", e);
		}
		finally {
			if (pw != null) {
				pw.close();
			}
		}
	}

	/***
	 * 转换key-value对的List为Map
	 * @param mapList
	 * @return
	 */
	public static Map convert2VKMap(List> mapList) {
		Map map = new HashMap(mapList.size());
		if(V.notEmpty(mapList)) {
			for(Map kvmap : mapList){
				map.put(kvmap.get("v"), String.valueOf(kvmap.get("k")));
			}
		}
		return map;
	}


	/***
	 * 打印所有参数信息
	 * @param request
	 */
	public static void dumpParams(HttpServletRequest request){
		Map params = request.getParameterMap();
		if(params != null && !params.isEmpty()){
			for(Map.Entry entry : params.entrySet()){
				String[] values = entry.getValue();
				if(values != null && values.length > 0){
					logger.debug(entry.getKey() + "=" + S.join(values));
				}
			}
		}
	}

	/***
	 * 将请求参数值转换为Map
	 * @param request
	 * @return
	 */
	public static Map convertParams2Map(HttpServletRequest request){
		Map result = new HashMap(16);
		if(request == null){
			return result;
		}
		Enumeration paramNames = request.getParameterNames();
		while (paramNames.hasMoreElements()){
			String paramName = (String) paramNames.nextElement();
			String[] values = request.getParameterValues(paramName);
			if(V.notEmpty(values)){
				if(values.length == 1){
					result.put(paramName, values[0]);
				}
				else{
					// 多个值需传递到后台SQL的in语句
					result.put(paramName, values);
				}
			}
		}
		return result;
	}

	/***
	 * 格式化查询统计的结果，便于chart图表显示
	 * @param dataList
	 * @return
	 */
	public static Map formatQueryResult4Chart(List> dataList){
		Map map = new HashMap<>();
		if(V.isEmpty(dataList)){
			return map;
		}
		List> rowList = new ArrayList<>();
		// 获取标题行
		List headers = extractHeadersFromQueryResult(dataList);
		// 标题行
		map.put("labelList", headers);
		for(String name : headers){
			Map objMap = new HashMap<>();
			objMap.put("name", name);
			List