自定义获取当前登录用户的注解@CurrentUser

 java自定义一个获取当前登录用户的注解，如@CurrentUser，这样就不用每次request.getAttribute("xxx")这样获取了，我准备自定义一个参数注解，放在参数中使用，自定义注解代码如下。

import java.lang.annotation.Documented;
import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;
import cn.com.xf.common.Constants;
//参数类型注解
@Target({ ElementType.PARAMETER })
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface CurrentUser {
 String value() default Constants.CURRENT_USER;
}

自定义一个类，然后实现实现方法参数解析器HandlerMethodArgumentResolver，将request获取到的用户信息返回给我们的@CurrentUser注解，代码如下。

import org.springframework.core.MethodParameter;
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.com.xf.web.bind.annotation.CurrentUser;
public class CurrentUserMethodArgumentResolver implements HandlerMethodArgumentResolver {
 @Override
 public Object resolveArgument(MethodParameter parameter, ModelAndViewContainer container, NativeWebRequest request,
   WebDataBinderFactory binderFactory) throws Exception {
  CurrentUser currentUserAnnotation = parameter.getParameterAnnotation(CurrentUser.class);
  return request.getAttribute(currentUserAnnotation.value(), NativeWebRequest.SCOPE_REQUEST);
 }
 @Override
 public boolean supportsParameter(MethodParameter parameter) {
  if (parameter.hasParameterAnnotation(CurrentUser.class)) {
   return true;
  }
  return false;
 }
}

最后在spring-mvc.xml中添加一个参数解析器bean来初始化我们自定义的类，代码如下。

<mvc:annotation-driven>
 <mvc:argument-resolvers>
  <bean class="cn.com.baidu.web.bind.method.CurrentUserMethodArgumentResolver"/>
 </mvc:argument-resolvers>
</mvc:annotation-driven>

于是我们就可以在我们的controller中使用我们获取当前用户的注解@CurrentUser了，如下。

 @RequestMapping("/app/add")
 public String addApp(@CurrentUser TsUser user, Model model) {     
  model.addAttribute("user", user);
  return "bar/app-add";

 }

注意：要记得举一反三，本套教程不是仅仅是教你获取当前登录用户才能用到，只要是参数注解赋值都是可以的。

完毕！