top.jfunc.http.component.AbstractHeaderExtractor 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.MultiValueMap;
import top.jfunc.common.utils.ObjectUtil;

import java.io.IOException;

/**
 * 默认提供cookie的支持
 * @author xiongshiyan at 2020/1/6 , contact me with email [email protected] or phone 15208384257
 */
public abstract class AbstractHeaderExtractor implements HeaderExtractor {
    /**
     * @param s 请求响应
     * @param httpRequest HttpRequest
     * @return 可能返回null
     * @throws IOException IOException
     */
    @Override
    public MultiValueMap extract(S s, HttpRequest httpRequest) throws IOException {
        Config config = httpRequest.getConfig();

        boolean retainResponseHeaders = ObjectUtil.defaultIfNull(httpRequest.retainResponseHeaders() , config.retainResponseHeaders());
        boolean followRedirects = ObjectUtil.defaultIfNull(httpRequest.followRedirects() , config.followRedirects());

        ///1.如果支持重定向，必须读取header
        if(followRedirects){
            retainResponseHeaders = Config.RETAIN_RESPONSE_HEADERS;
        }

        //要求不需要解析header
        if(!retainResponseHeaders){
            return null;
        }

        //2.从响应中获取headers
        return doExtractHeaders(s , httpRequest);
    }
    /**
     * 真实的实现获取header
     * @param s S
     * @param httpRequest HttpRequest
     * @return headers
     * @throws IOException IOException
     */
    protected abstract MultiValueMap doExtractHeaders(S s , HttpRequest httpRequest) throws IOException;
}