结合redis编写User自定义参数解析器UserArgumentResolver
import javax.servlet.http.Cookie;
import javax.servlet.http.HttpServletRequest;
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.xingaohbd.seckil.model.User;
import cn.xingaohbd.seckil.service.UserService; @Service
public class UserArgumentResolver implements HandlerMethodArgumentResolver { @Autowired
public UserService userService; public String getParameterCokies(HttpServletRequest request,String tokenName){
Cookie[] cookies = request.getCookies();
for (Cookie ck : cookies) {
if(ck.getName().equals(tokenName)){
return ck.getValue();
}
}
return null;
} @Override
public boolean supportsParameter(MethodParameter methodParameter) {
Class<?> p_class = methodParameter.getParameterType();
return p_class == User.class;
} @Override
public Object resolveArgument(MethodParameter parameter, ModelAndViewContainer mavContainer,
NativeWebRequest nativeWebRequest, WebDataBinderFactory binderFactory) throws Exception {
HttpServletRequest request = nativeWebRequest.getNativeRequest(HttpServletRequest.class);
String requestParameter_token = request.getParameter("token");
String cokies_token = getParameterCokies(request, "token");
if(requestParameter_token == null && cokies_token == null){
return null;
}
return userService.getUserFromRedisByToken(requestParameter_token != null ? requestParameter_token : cokies_token);
} }
UserService
@Service
@Transactional
public class UserServiceImpl implements UserService {
@Autowired
private UserRedis userRedis; @Override
public Object getUserFromRedisByToken(String token) {
return userRedis.get(token);
}
}
在WebConfig中注册自定义参数解析器UserArgumentResolver
import java.util.List;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Configuration;
import org.springframework.util.ResourceUtils;
import org.springframework.web.method.support.HandlerMethodArgumentResolver;
import org.springframework.web.servlet.config.annotation.ResourceHandlerRegistry;
import org.springframework.web.servlet.config.annotation.WebMvcConfigurationSupport; @Configuration
public class WebConfig extends WebMvcConfigurationSupport{ @Autowired
UserArgumentResolver userArgumentResolver; @Override
protected void addArgumentResolvers(List<HandlerMethodArgumentResolver> argumentResolvers) {
argumentResolvers.add(userArgumentResolver);
} public void addResourceHandlers(ResourceHandlerRegistry registry){
registry.addResourceHandler("/**").addResourceLocations(ResourceUtils.CLASSPATH_URL_PREFIX+"/static/");
super.addResourceHandlers(registry);
} }