欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

【java后端】restTemplate异常处理

程序员文章站 2022-06-26 14:36:56
...

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

在与其他微服务接口进行联调的过程中,因为我这边依赖的接口在正常和异常两种情况下,返回的字段不一样,导致我这边通过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://my.oschina.net/ChenGuop/blog/1581759?tdsourcetag=s_pctim_aiomsg
https://www.cnblogs.com/fzll/p/3400558.html