【java后端】restTemplate异常处理

叁歲伎倆 2022-04-17 03:51 544阅读 0赞

*  记录一下今天遇到的问题，和尝试解决过程中的经历 -

在与其他微服务接口进行联调的过程中，因为我这边依赖的接口在正常和异常两种情况下，返回的字段不一样，导致我这边通过restTemplate发送请求时，封装的返回对象不能够灵活适配接口返回对象的字段，于是在服务异常返回时，我这边处理不了。  
于是查询解决这个问题的方法，发现可以自己定义一个ResponseErrorHandler的实现类，对异常情况进行处理。

**依赖接口返回参数示例**

正常情况：

{
        "library_id": "36"
    }

异常情况：

{
        "error_code": 8008,
        "error_msg": "",
        "error_type": "library_name is not string"
    }

由于依赖接口正常和异常情况下http状态码都是200，导致不能通过状态码来判断哪种情况是异常情况，只能通过取body中的值来校验，写了如下FaceResponseErrorHandler类。  
ResponseErrorHandler的实现类：

public class FaceResponseErrorHandler implements ResponseErrorHandler {
    
        /** 对response进行判断，如果是异常情况，返回true */
        @Override
        public boolean hasError(ClientHttpResponse response) throws IOException {
            Map map = JSON.parseObject(convertStreamToString(response.getBody()));
            return map.containsKey("error_code");
        }
    
        /** 异常情况时的处理方法 */
        @Override
        public void handleError(ClientHttpResponse response) {
            throw new LibraryException(FaceErrorCode.IA_LIBRARY_ERROR, FaceResponseErrorHandler.class);
        }
    
        /**
         * InputStream转成String
         * @param is InputStream
         * @return String
         */
        private String convertStreamToString(InputStream is) {
    
            BufferedReader reader = new BufferedReader(new InputStreamReader(is));
            StringBuilder sb = new StringBuilder();
    
            String line;
            try {
                while ((line = reader.readLine()) != null) {
                    sb.append(line);
                }
            } catch (IOException e) {
                throw BaseException.of(FaceErrorCode.IA_LIBRARY_ERROR.of());
            } finally {
                is.close();
            }
    
            return sb.toString();
        }
    }

FaceResponseErrorHandler调用：

restTemplate.setErrorHandler(new FaceResponseErrorHandler());
        Object res = restTemplate.postForObject(host + IAConstant.LIBRARY_MANAGER, addReqDO, Object.class);

测试后发现，这种方式存在异常报错或获取到的res为Null的问题，原因是在FaceResponseErrorHandler类中convertStreamToString()方法里读取数据并转化之后，在finally里面将流关闭，导致后续restTemplate原生方法在处理返回值的时候报流已关闭的错误，即使将finally代码去除，由于在convertStreamToString()方法读取了InputStream对象里面的内容，而InputStream只能读取一次，在InputStream读取之后response的body就空了，后面在获取结果时得到是null，于是这种方法行不通，除非还是通过http状态码来进行判断，但目前状态码条件不满足，最后还是弃用了FaceResponseErrorHandler，又将restTemplate的返回对象定义为Object类型。在debug调试过程中意外发现返回的object对象其实是LinkedHashMap类型的，于是采用了一个比较low的方法来解决这个问题，即将res进行了强转成LinkedHashMap来获取值，然后对数据进行处理：

/** 强转数据并处理 */
    private IALibraryRes getIALibraryRes(Object res) {
        IALibraryRes iaLibraryRes = new IALibraryRes();
        if (res != null) {
            if (res instanceof LinkedHashMap) {
                LinkedHashMap resMap = (LinkedHashMap) res;
                if (resMap.containsKey("error_code")) {
                    for (Object key : resMap.keySet()) {
                        LOGGER.info(String.valueOf(key) + ": " + String.valueOf(resMap.get(key)));
                    }
                    throw new LibraryException(FaceErrorCode.IA_LIBRARY_ERROR, IALibraryServiceImpl.class);
                } else {
                    iaLibraryRes.setLibraryId(String.valueOf(resMap.get("library_id")));
                }
            } else {
                throw new LibraryException(FaceErrorCode.IA_LIBRARY_ERROR, IALibraryServiceImpl.class);
            }
        } else {
            throw new LibraryException(FaceErrorCode.IA_LIBRARY_ERROR, IALibraryServiceImpl.class);
        }
        return iaLibraryRes;
    }

**参考网址：**  
[https://blog.csdn.net/achang07/article/details/80549741][https_blog.csdn.net_achang07_article_details_80549741]  
[https://my.oschina.net/ChenGuop/blog/1581759?tdsourcetag=s\_pctim\_aiomsg][https_my.oschina.net_ChenGuop_blog_1581759_tdsourcetag_s_pctim_aiomsg]  
[https://www.cnblogs.com/fzll/p/3400558.html][https_www.cnblogs.com_fzll_p_3400558.html]

[https_blog.csdn.net_achang07_article_details_80549741]: https://blog.csdn.net/achang07/article/details/80549741
[https_my.oschina.net_ChenGuop_blog_1581759_tdsourcetag_s_pctim_aiomsg]: https://my.oschina.net/ChenGuop/blog/1581759?tdsourcetag=s_pctim_aiomsg
[https_www.cnblogs.com_fzll_p_3400558.html]: https://www.cnblogs.com/fzll/p/3400558.html