java - Spring默认的RequestParam值等于方法调用

Question

我正在尝试添加一个具有默认值的请求参数 - 但我希望该默认值是登录用户的名称。

我有一个方法 getUsername()，它返回当前用户的名称，但我无法将注释的值设置为方法调用(或类属性)。这是我的方法:

@RequestMapping(method = RequestMethod.GET)
public @ResponseBody ResponseEntity<WebUser> getUser(
        @RequestParam(value = "username", defaultValue = getUsername()) String username) {
    return ResponseEntity.ok(service.getUser(getUsername(), username.toLowerCase()));
}

我可以使 RequestParam 不再是必需的，并在为 null 时填充它 - 但这感觉不太优雅(或 spring-ish)。还有其他方法可以实现这一点吗？

Answer 1

正如 fateddy 所建议的，最简单的方法是实现 HandlerMethodArgumentResolver。

public class UsernameHandlerMethodArgumentResolver implements HandlerMethodArgumentResolver {
    @Override
    public boolean supportsParameter(MethodParameter methodParameter) {
        return methodParameter.getParameterType().equals(Username.class);
    }
    @Override
    public Object resolveArgument(MethodParameter methodParameter,
                                  ModelAndViewContainer modelAndViewContainer,
                                  NativeWebRequest nativeWebRequest,
                                  WebDataBinderFactory webDataBinderFactory) throws Exception {
        String username = nativeWebRequest.getParameter("username");
        if (username == null && nativeWebRequest.getUserPrincipal() != null) {
            username = nativeWebRequest.getUserPrincipal().getName();
        }
        return new Username(username);
    }
}

这需要一个简单的用户名类:

public class Username {
    private String username;
    public Username(String username) {
        this.username = username;
    }
    public String getValue() {
        return this.username;
    }
}

以及注释

@Target(ElementType.PARAMETER)
@Retention(RetentionPolicy.RUNTIME)
public @interface UserRequest {}

为了正确配置，需要对 WebMvcConfigurerAdapter 进行非常小的更改:

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

就是这样。很好，可以走了。现在我们可以简单地将参数放入 Controller 端点:

@RequestMapping(method = RequestMethod.GET)
public @ResponseBody ResponseEntity<WebUser> getUser(@UserRequest Username username) {
    return ResponseEntity.ok(service.getUser(username, username.toLowerCase()));
}