关注微信公众号 太平洋学习网 扫描二维码
最新消息:关注【太平洋学习网】微信公众号,可以获取全套资料,【全套Java基础27天】【JavaEE就业视频4个月】【Android就业视频4个月】

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

Javaweb admin 浏览 评论

 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";

 }

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

完毕!

来源网站:太平洋学习网,转载请注明出处:http://www.tpyyes.com/a/javaweb/2017/0729/172.html