Polishing
This commit is contained in:
parent
737d77a739
commit
6825287360
|
|
@ -1,5 +1,5 @@
|
|||
/*
|
||||
* Copyright 2002-2019 the original author or authors.
|
||||
* Copyright 2002-2020 the original author or authors.
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
|
|
@ -42,8 +42,8 @@ public class MapMethodProcessor implements HandlerMethodArgumentResolver, Handle
|
|||
|
||||
@Override
|
||||
public boolean supportsParameter(MethodParameter parameter) {
|
||||
return Map.class.isAssignableFrom(parameter.getParameterType()) &&
|
||||
parameter.getParameterAnnotations().length == 0;
|
||||
return (Map.class.isAssignableFrom(parameter.getParameterType()) &&
|
||||
parameter.getParameterAnnotations().length == 0);
|
||||
}
|
||||
|
||||
@Override
|
||||
|
|
@ -70,8 +70,8 @@ public class MapMethodProcessor implements HandlerMethodArgumentResolver, Handle
|
|||
}
|
||||
else if (returnValue != null) {
|
||||
// should not happen
|
||||
throw new UnsupportedOperationException("Unexpected return type: " +
|
||||
returnType.getParameterType().getName() + " in method: " + returnType.getMethod());
|
||||
throw new UnsupportedOperationException("Unexpected return type [" +
|
||||
returnType.getParameterType().getName() + "] in method: " + returnType.getMethod());
|
||||
}
|
||||
}
|
||||
|
||||
|
|
|
|||
|
|
@ -1,5 +1,5 @@
|
|||
/*
|
||||
* Copyright 2002-2017 the original author or authors.
|
||||
* Copyright 2002-2020 the original author or authors.
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
|
|
@ -70,8 +70,8 @@ public class ModelMethodProcessor implements HandlerMethodArgumentResolver, Hand
|
|||
}
|
||||
else {
|
||||
// should not happen
|
||||
throw new UnsupportedOperationException("Unexpected return type: " +
|
||||
returnType.getParameterType().getName() + " in method: " + returnType.getMethod());
|
||||
throw new UnsupportedOperationException("Unexpected return type [" +
|
||||
returnType.getParameterType().getName() + "] in method: " + returnType.getMethod());
|
||||
}
|
||||
}
|
||||
|
||||
|
|
|
|||
|
|
@ -38,29 +38,24 @@ import org.springframework.web.method.support.ModelAndViewContainer;
|
|||
*/
|
||||
public class PrincipalMethodArgumentResolver implements HandlerMethodArgumentResolver {
|
||||
|
||||
|
||||
@Override
|
||||
public boolean supportsParameter(MethodParameter parameter) {
|
||||
Class<?> paramType = parameter.getParameterType();
|
||||
return Principal.class.isAssignableFrom(paramType);
|
||||
return Principal.class.isAssignableFrom(parameter.getParameterType());
|
||||
}
|
||||
|
||||
@Override
|
||||
public Object resolveArgument(MethodParameter parameter, @Nullable ModelAndViewContainer mavContainer,
|
||||
NativeWebRequest webRequest, @Nullable WebDataBinderFactory binderFactory) throws Exception {
|
||||
|
||||
Class<?> paramType = parameter.getParameterType();
|
||||
|
||||
HttpServletRequest request = webRequest.getNativeRequest(HttpServletRequest.class);
|
||||
if (request == null) {
|
||||
throw new IllegalStateException(
|
||||
"Current request is not of type [HttpServletRequest]: " + webRequest);
|
||||
throw new IllegalStateException("Current request is not of type HttpServletRequest: " + webRequest);
|
||||
}
|
||||
|
||||
Principal principal = request.getUserPrincipal();
|
||||
if (principal != null && !paramType.isInstance(principal)) {
|
||||
throw new IllegalStateException(
|
||||
"Current user principal is not of type [" + paramType.getName() + "]: " + principal);
|
||||
if (principal != null && !parameter.getParameterType().isInstance(principal)) {
|
||||
throw new IllegalStateException("Current user principal is not of type [" +
|
||||
parameter.getParameterType().getName() + "]: " + principal);
|
||||
}
|
||||
|
||||
return principal;
|
||||
|
|
|
|||
Loading…
Reference in New Issue