Spring boot - 将参数从拦截器传递给控制器​​中的方法

Question

出于学习目的，我制作了一个自定义身份验证系统，我通过Authorization 标头将令牌从客户端传递到服务器。

在服务器端，我想知道是否可以在拦截器中，在请求到达控制器中的方法之前，创建一个带有来自令牌的电子邮件作为属性的用户对象，然后传递这个用户反对我需要的每个请求。

这是我想要得到的，例如：

@RestController
public class HelloController {

    @RequestMapping("/")
    public String index(final User user) {
        return user.getEmail();
    }

}

public class User {
    private String email;
}

其中 user 是我在预拦截器中使用请求 Authorization 标头创建的对象，然后我可以传递或不传递给 RestController 中的任何方法。

这可能吗？

Answer 1

推荐解决方案

我将创建一个带有@Scope 请求的@Bean 请求，该请求将持有用户，然后将适当的实体放入该持有人，然后在方法内从该持有人获取。

@Component
@Scope("request")
public class CurrentUser {
    private User currentUser;

    public User getCurrentUser() {
        return currentUser;
    }

    public void setCurrentUser(User currentUser) {
        this.currentUser = currentUser;
    }
}

然后

@Component
public class MyInterceptor implements HandlerInterceptor {

   private CurrentUser currentUser;

   @Autowired
   MyInterceptor(CurrentUser currentUser) {
       this.currentUser = currentUser;
   }

   @Override
   public boolean preHandle(
      HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
      this.currentUser.setCurrentUser(new User("whatever"));
      return true;
   }
}

在控制器中

@RestController
public class HelloController {

    private CurrentUser currentUser;

    @Autowired
    HelloController(CurrentUser currentUser) {
        this.currentUser = currentUser;
    }

    @RequestMapping("/")
    public String index() {
        return currentUser.getCurrentUser().getEmail();
    }

}

替代解决方案

如果您想要的对象只包含一个字段，您可以作弊并将该字段添加到HttpServletRequest 参数中，然后就可以看到奇迹发生了。

@Component
public class MyInterceptor implements HandlerInterceptor {
   @Override
   public boolean preHandle(
      HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
      //TRY ONE AT THE TIME: email OR user
      //BOTH SHOULD WORK BUT SEPARATELY OF COURSE
      request.setAttribute("email", "login@domain.com");
      request.setAttribute("user", new User("login@domain.com"));
      return true;
   }
}