JavaÃëɱϵͳ£¨¾Å£©·Ö²¼Ê½session

´´½¨UUID

Éú³Écookie

//ÉèÖÃÓÐЧÆÚ¡£

//²ÎÊýУÑé

@Configuration
public class WebConfig  extends WebMvcConfigurerAdapter{
 
 @Autowired
 UserArgumentResolver userArgumentResolver;
 
 @Override
 public void addArgumentResolvers(List<HandlerMethodArgumentResolver> argumentResolvers) {
  argumentResolvers.add(userArgumentResolver);
 }
 
 //»Øµ÷·½·¨
}

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

import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.core.MethodParameter;
import org.springframework.stereotype.Service;
import org.springframework.web.bind.support.WebDataBinderFactory;
import org.springframework.web.context.request.NativeWebRequest;
import org.springframework.web.method.support.HandlerMethodArgumentResolver;
import org.springframework.web.method.support.ModelAndViewContainer;

import cn.tedu.miaosha.domain.MiaoshaUser;
import cn.tedu.miaosha.service.MiaoshaUserService;

@Service
public class UserArgumentResolver implements HandlerMethodArgumentResolver {

 @Autowired
 MiaoshaUserService userService;
 
 public boolean supportsParameter(MethodParameter parameter) {
  Class<?> clazz = parameter.getParameterType();
  return clazz==MiaoshaUser.class;
 }

 public Object resolveArgument(MethodParameter parameter, ModelAndViewContainer mavContainer,
   NativeWebRequest webRequest, WebDataBinderFactory binderFactory) throws Exception {
  HttpServletRequest request = webRequest.getNativeRequest(HttpServletRequest.class);
  HttpServletResponse response = webRequest.getNativeResponse(HttpServletResponse.class);
  
  String paramToken = request.getParameter(MiaoshaUserService.COOKI_NAME_TOKEN);
  String cookieToken = getCookieValue(request, MiaoshaUserService.COOKI_NAME_TOKEN);
  if(StringUtils.isEmpty(cookieToken) && StringUtils.isEmpty(paramToken)) {
   return null;
  }
  String token = StringUtils.isEmpty(paramToken)?cookieToken:paramToken;
  return userService.getByToken(response, token);
 }

 private String getCookieValue(HttpServletRequest request, String cookiName) {
  Cookie[]  cookies = request.getCookies();
  for(Cookie cookie : cookies) {
   if(cookie.getName().equals(cookiName)) {
    return cookie.getValue();
   }
  }
  return null;
 }

}