top.jfunc.http.cookie.CookieInterceptor Maven / Gradle / Ivy

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

import top.jfunc.http.config.Config;
import top.jfunc.http.interceptor.Interceptor;
import top.jfunc.http.request.HttpRequest;
import top.jfunc.http.response.ClientHttpResponse;

import java.io.IOException;
import java.util.Objects;

/**
 * 用于处理{@link Cookie}的拦截器
 * @author xiongshiyan at 2020/12/8 , contact me with email [email protected] or phone 15208384257
 */
public class CookieInterceptor implements Interceptor {
    private CookieAccessor cookieAccessor;

    public CookieInterceptor(CookieAccessor cookieAccessor) {
        this.cookieAccessor = Objects.requireNonNull(cookieAccessor);
    }

    @Override
    public HttpRequest onBefore(HttpRequest httpRequest) throws IOException {
            getCookieAccessor().addCookieIfNecessary(httpRequest);
        return httpRequest;
    }

    @Override
    public ClientHttpResponse onBeforeReturn(HttpRequest httpRequest, ClientHttpResponse clientHttpResponse) throws IOException {
        //必须要获取header
        httpRequest.retainResponseHeaders(Config.RETAIN_RESPONSE_HEADERS);
        getCookieAccessor().saveCookieIfNecessary(httpRequest, clientHttpResponse.getHeaders());
        return clientHttpResponse;
    }

    public CookieAccessor getCookieAccessor() {
        return cookieAccessor;
    }
}