SpringMVC杂记(十七) HandlerMethodArgumentResolver接口应用example

自从spring3.1 开始就有了这个接口，可以为@RequestMapping标注的方法扩展传入的参数。 
以shiro为例，扩展一个标注，@CurrentUser,只要有这个标注，就可以在shiro的安全上下文中取出适当的对象直接从参数传入，request响应函数。 

Java代码  

1. import java.lang.annotation.Documented;  
2. import java.lang.annotation.ElementType;  
3. import java.lang.annotation.Retention;  
4. import java.lang.annotation.RetentionPolicy;  
5. import java.lang.annotation.Target;  
6.   
7. @Documented  
8. @Target({ElementType.PARAMETER})  
9. @Retention(RetentionPolicy.RUNTIME)  
10. public @interface CurrentUser {  
11. }  

Java代码  

1. @RequestMapping(value = "/test1")  
2. public @ResponseBody String test1(@CurrentUser Long userId) {  
3.     return userId.toString();  
4. }  
5.   
6. @RequestMapping(value = "/test2")  
7. public @ResponseBody String test2(@CurrentUser UserDetails userDetails) {  
8.     return userDetails.toString();  
9. }  

Java代码  

1. import java.lang.annotation.Annotation;  
2.   
3. import org.springframework.core.MethodParameter;  
4. import org.springframework.web.bind.support.WebDataBinderFactory;  
5. import org.springframework.web.context.request.NativeWebRequest;  
6. import org.springframework.web.method.support.HandlerMethodArgumentResolver;  
7. import org.springframework.web.method.support.ModelAndViewContainer;  
8.   
9. import com.github.yingzhuo.mycar2.annotation.CurrentUser;  
10. import com.github.yingzhuo.mycar2.security.SecurityUtils;  
11. import com.github.yingzhuo.mycar2.security.UserDetails;  
12.   
13. public class CurrentUserHandlerMethodArgumentResolver implements HandlerMethodArgumentResolver {  
14.   
15.     @Override  
16.     public boolean supportsParameter(MethodParameter parameter) {  
17.         Class<?> klass = parameter.getParameterType();  
18.         if (klass.isAssignableFrom(UserDetails.class) || klass.isAssignableFrom(Long.class)) {  
19.             Annotation[] as = parameter.getParameterAnnotations();  
20.             for (Annotation a : as) {  
21.                 if (a.annotationType() == CurrentUser.class) {  
22.                     return true;  
23.                 }  
24.             }  
25.         }  
26.         return false;  
27.     }  
28.   
29.     @Override  
30.     public Object resolveArgument(MethodParameter parameter, ModelAndViewContainer mavContainer, NativeWebRequest webRequest,  
31.             WebDataBinderFactory binderFactory) throws Exception {  
32.           
33.         if ((SecurityUtils.isAuthenticated() || SecurityUtils.isRemembered()) == false) {  
34.             return null;  
35.         }  
36.           
37.         Class<?> klass = parameter.getParameterType();  
38.           
39.         UserDetails userDetails = SecurityUtils.getUserDetails();  
40.           
41.         if (klass.isAssignableFrom(UserDetails.class)) {  
42.             return SecurityUtils.getUserDetails();  
43.         }  
44.           
45.         if (klass.isAssignableFrom(Long.class)) {  
46.             return userDetails != null ? userDetails.getId() : null;  
47.         }  
48.           
49.         return null;  
50.     }  
51. }  

最后，需要配置一下 

Xml代码  

1. <mvc:annotation-driven>  
2.     <mvc:argument-resolvers>  
3.         <bean class="xxx.yyy.CurrentUserHandlerMethodArgumentResolver" />  
4.     </mvc:argument-resolvers>  
5. </mvc:annotation-driven>