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"; }
注意:要记得举一反三,本套教程不是仅仅是教你获取当前登录用户才能用到,只要是参数注解赋值都是可以的。
完毕!