top.jfunc.http.component.AbstractStreamExtractor Maven / Gradle / Ivy

Go to download
package top.jfunc.http.component;

import top.jfunc.http.config.Config;
import top.jfunc.http.request.HttpRequest;
import top.jfunc.common.utils.IoUtil;
import top.jfunc.common.utils.ObjectUtil;

import java.io.IOException;
import java.io.InputStream;

/**
 * 从请求响应中获取stream
 * @author xiongshiyan at 2020/1/6 , contact me with email [email protected] or phone 15208384257
 */
public abstract class AbstractStreamExtractor implements StreamExtractor {
    /**
     * 忽略body的话，就返回空stream
     * @param s 请求响应
     * @param httpRequest HttpRequest
     * @return InputStream
     * @throws IOException IOException
     */
    @Override
    public InputStream extract(S s, HttpRequest httpRequest) throws IOException {
        Config config = httpRequest.getConfig();
        boolean ignoreResponseBody = ObjectUtil.defaultIfNull(httpRequest.ignoreResponseBody(), config.ignoreResponseBody());
        if(ignoreResponseBody){
            return IoUtil.emptyStream();
        }
        InputStream inputStream = doExtract(s, httpRequest);
        return null != inputStream ? inputStream : IoUtil.emptyStream();
    }

    /**
     * 真正的获取响应中的InputStream
     * @param s S
     * @param httpRequest HttpRequest
     * @return InputStream
     * @throws IOException IOException
     */
    protected abstract InputStream doExtract(S s, HttpRequest httpRequest) throws IOException;
}