カスタム@CurrentUserパラメータ解析現在のログインユーザーを取得

9784 ワード

HandlerMethodArgumentResolver概要HandlerMethodArgumentResolverインタフェースには、次の2つの方法があります.
	//            
	boolean supportsParameter(MethodParameter parameter);
	
	//          
	@Nullable
	Object resolveArgument(MethodParameter parameter, @Nullable ModelAndViewContainer mavContainer,
			NativeWebRequest webRequest, @Nullable WebDataBinderFactory binderFactory) throws Exception;

現在のユーザー注釈のカスタマイズ
/**
 *       
 */
@Retention(RetentionPolicy.RUNTIME)
@Target(value = ElementType.PARAMETER)
public @interface CurrentUser {

}

カスタムパラメータ解析
/**
 *     MethodArgumentResolver
 */
public class CurrentUserMethodArgumentResolver implements HandlerMethodArgumentResolver {

	@Inject
	private UserService userService;

	/**
	 *       
	 */
	@Override
	public boolean supportsParameter(MethodParameter methodParameter) {
		//          @CurrentUser        User    
		return methodParameter.hasParameterAnnotation(CurrentUser.class) && User.class.isAssignableFrom(methodParameter.getParameterType());
	}

	/**
	 *     
	 */
	@SuppressWarnings("unchecked")
	@Override
	public Object resolveArgument(MethodParameter methodParameter, ModelAndViewContainer modelAndViewContainer, NativeWebRequest nativeWebRequest, WebDataBinderFactory webDataBinderFactory) throws Exception {
		//      
		return userService.getCurrent();
	}

}

Springbootプロジェクト登録カスタム解析器
@Configuration
public class MvcConfig implements WebMvcConfigurer {
	@Bean
	public CurrentUserMethodArgumentResolver currentUserMethodArgumentResolver() {
		return new CurrentUserMethodArgumentResolver();
	}

	@Override
	public void addArgumentResolvers(List<HandlerMethodArgumentResolver> argumentResolvers) {
		argumentResolvers.add(currentUserMethodArgumentResolver());
	}
	
	......
}

従来のプロジェクト登録カスタム解析器
<mvc:annotation-driven>
   <mvc:argument-resolvers>
      <bean class="xx.xx.xx.currentUserMethodArgumentResolver" />
   </mvc:argument-resolvers>
</mvc:annotation-driven>

@CurrentUser注記を使用して現在のユーザーを取得
@Controller
@RequestMapping("/test")
public class TestController {

	@GetMapping("/currentUser")
	public String test(@CurrentUser User currentUser) {
		......
	}

}