restTemplate异常处理

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

在与其他微服务接口进行联调的过程中，因为我这边依赖的接口在正常和异常两种情况下，返回的字段不一样，导致我这边通过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

restTemplate异常处理

猜你喜欢