com.allways.utils.CookieUtil Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of allways-base Show documentation
allways-base
There is a newer version: 1.0.4.RELEASE
package com.allways.utils;

import com.allways.constants.Constants;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;

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

/**
 * Cookie.Util
 *
 */
public class CookieUtil {

	// 记住我默认7天
	private static final int COOKIE_MAX_AGE = 7*24*60*60;
	// 保存路径,根路径
	private static final String COOKIE_PATH = "/";

	/**
	 * 保存
	 *
	 * @param response 响应
	 * @param key cookie的key
	 * @param value cookie的value
	 * @param ifRemember 是否记住密码
	 */
	public static void set(HttpServletResponse response, String key, String value, boolean ifRemember) {
		int age = ifRemember?COOKIE_MAX_AGE:-1;
		set(response, key, value, null, COOKIE_PATH, age, true);
	}

	/**
	 * 保存
	 *
	 * @param response 响应
	 * @param key cookie的key
	 * @param value cookie的value
	 * @param maxAge cookie有效时间
	 */
	private static void set(HttpServletResponse response, String key, String value, String domain, String path, int maxAge, boolean isHttpOnly) {
		Cookie cookie = new Cookie(key, value);
		if (domain != null) {
			cookie.setDomain(domain);
		}
		cookie.setPath(path);
		cookie.setMaxAge(maxAge);
		cookie.setHttpOnly(isHttpOnly);
		response.addCookie(cookie);
	}

	/**
	 * 查询value
	 *
	 * @param request 请求
	 * @param key cookie的key
	 * @return 获取Cookie值
	 */
	public static String getValue(HttpServletRequest request, String key) {
		Cookie cookie = get(request, key);
		if (cookie != null) {
			return cookie.getValue();
		}
		return null;
	}

	/**
	 * 查询Cookie
	 *
	 * @param request 请求
	 * @param key cookie的key
	 */
	private static Cookie get(HttpServletRequest request, String key) {
		Cookie[] arr_cookie = request.getCookies();
		if (arr_cookie != null && arr_cookie.length > 0) {
			for (Cookie cookie : arr_cookie) {
				if (cookie.getName().equals(key)) {
					return cookie;
				}
			}
		}
		return null;
	}

	/**
	 * 删除Cookie
	 *
	 * @param request 请求
	 * @param response 响应
	 * @param key cookie的key
	 */
	public static void remove(HttpServletRequest request, HttpServletResponse response, String key) {
		Cookie cookie = get(request, key);
		if (cookie != null) {
			set(response, key, "", null, COOKIE_PATH, 0, true);
		}
	}

	public static HttpServletRequest getRequest() {
		return ((ServletRequestAttributes)RequestContextHolder.getRequestAttributes()).getRequest();
	}

	/**
	 * @return 获取登录用户
	 */
	public static String getLoginUser(){
		HttpServletRequest request = getRequest();
		String username = request.getHeader(Constants.USER_NAME);
		return username;
	}

}