springboot设置响应状态码

心已赠人 2021-12-16 07:35 739阅读 0赞

业务需要自定义http的响应状态,图中响应状态码是200，需要改成401.

使用HttpServletResponse对象直接设置状态局限比较大，可以改用以下方式，通过ResponseEntity可以返回我们需要的数据还可以自定义状态码。

@RequestMapping("/oauth/token")  
    public ResponseEntity<Object> token(@RequestParam(value="grant\_type",required=false)String grant\_type,  
            @RequestParam(value="client\_id",required=false)String client\_id,  
            @RequestParam(value="client\_secret",required=false)String client\_secret)\{  
        OauthError error=new OauthError();  
        AccessTokenResponse tokenResponse=new AccessTokenResponse();  
        if(grant\_type==null||client\_id==null||client\_secret==null)\{  
            error.setError(ResponseErrorEnum.INVALID\_REQUEST.getError());  
            error.setError\_description(ResponseErrorEnum.INVALID\_REQUEST.getError\_description());  
            return new ResponseEntity<Object>(error,HttpStatus.OK);  
        \}  
        if(GrantTypeEnum.getGrantTypeEnumByCode(grant\_type)==null)\{  
            error.setError(ResponseErrorEnum.UNSUPPORTED\_GRANT\_TYPE.getError());  
            error.setError\_description(ResponseErrorEnum.UNSUPPORTED\_GRANT\_TYPE.getError\_description());  
            return new ResponseEntity<Object>(error,HttpStatus.OK);  
        \}  
        if(grant\_type.equals(GrantTypeEnum.CLIENT\_CREDENTIALS.getCode()))\{  
            tokenResponse=oauthService.clientCredentials(client\_id, client\_secret);  
            if(tokenResponse==null)\{  
                error.setError(ResponseErrorEnum.INVALID\_CLIENT.getError());  
                error.setError\_description(ResponseErrorEnum.INVALID\_CLIENT.getError\_description());  
                return new ResponseEntity<Object>(error,HttpStatus.UNAUTHORIZED);  
            \}  
              
        \}  
        return new ResponseEntity<Object>(tokenResponse,HttpStatus.OK);  
    \}  
运行测试，401