如何从请求 header 中获取 user_id 而不是将其作为请求参数传递？ 然后通过header发回

Question

对于各种 REST api 端点，user_id 将到达后端，需要进一步处理，然后作为响应发送回前端。

我有一种感觉，我可以通过 header 来做到这一点，而不是每次都将它作为路径参数传递，只是我似乎还找不到相关信息。

目前我将响应作为 ResponseEntity 发送。 如果可能的话，我想保留这个选项。

我正在使用 Java 和 Spring 启动。

Answer 1

示例基于

https://docs.spring.io/spring-framework/docs/current/javadoc-api/org/springframework/http/ResponseEntity.html

编辑以根据请求添加 readign header

@RequestMapping("/handle")
public ResponseEntity<String> handle(HttpServletRequest httpRequest) {
  String userId= httpRequest.getHeader("user_id");
  HttpHeaders responseHeaders = new HttpHeaders();
  responseHeaders.set("user_id", userId);
  return new ResponseEntity<String>("Hello World", responseHeaders, HttpStatus.CREATED);
}

Answer 2

我已经决定，对于我的场景，我只需要获取用户 ID 然后用它进行响应的最佳方法是使用 @RequestHeader("userId") Long userId 注释。

让我们看看我最初是如何配置 enpoint 的：

@PostMapping(path = "/add-follower/{userIdForFollowing}/{currentUserId}")
public ResponseEntity<String> addFollower(@PathVariable ("userIdForFollowing") Long userIdForFollowing, @PathVariable Long currentUserId)
{
    Follow newFollow = followService.returnNewFollow(userIdForFollowing, currentUserId);
    
    newFollow = followService.saveFollowToDb(newFollow);

    return new ResponseEntity<>("Follow saved successfully", HttpStatus.OK);
}

现在，让我们看看我如何重构端点以从 header 获取 ID 并在响应中返回它们：

@PostMapping(path = "/add-follower")
public ResponseEntity<String> addFollower(@RequestHeader("userIdForFollowing") Long userIdForFollowing, @RequestHeader("currentUserId") Long currentUserId)
{

    Follow newFollow = followService.returnNewFollow(userIdForFollowing, currentUserId);
    newFollow = followService.saveFollowToDb(newFollow);

    //here I will add more code which should replace the String in the ResponseEntity.
    return new ResponseEntity<>("Follow saved successfully", HttpStatus.OK);
}