com.github.cosycode.ext.web.http.MyHttpClient Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of extend-mod Show documentation
扩展模块, 用于存放一些非常用的工具或模块的扩展类, 例如在poi基础上扩展的excel的导入模块, 模拟按键模块
The newest version!
package com.github.cosycode.ext.web.http;

import com.github.cosycode.common.ext.hub.LazySingleton;
import com.github.cosycode.ext.io.cache.AbstractKeyCacheHandler;
import lombok.Getter;
import lombok.Setter;
import lombok.experimental.Accessors;
import lombok.extern.slf4j.Slf4j;
import org.apache.hc.client5.http.impl.classic.CloseableHttpClient;
import org.apache.hc.core5.http.io.HttpClientResponseHandler;

import java.io.IOException;
import java.util.function.BiConsumer;
import java.util.function.Consumer;

/**
 * Description :  recommend that create a single client for per different http service.
 * 
 * created in  2023/5/14
 * 
 *
 * @author CPF
 **/
@Slf4j
@Setter
@Getter
@Accessors(chain = true)
public class MyHttpClient {

    public static final LazySingleton DEFAULT_INSTANCE = LazySingleton.of(() -> {
        MyHttpClient myHttpClient = new MyHttpClient(Http5ClientConfig.getCloseableHttpClient(), MyHttpResponse.DEFAULT_HANDLER);
        myHttpClient.setWebCacheHandler(null);
        myHttpClient.setPreProcess(req -> log.info("[{}] {}", req.method(), req.requestUrl()));
        myHttpClient.setPostProcess((req, resp) -> {
            if (resp.isCode(404)) {
                log.warn("[{}] {} ==> {}", req.method(), req.requestUrl(), resp.data());
            }
        });
        return myHttpClient;
    });

    final CloseableHttpClient closeableHttpClient;
    /**
     * used for parsing responses
     */
    final HttpClientResponseHandler httpClientResponseHandler;
    /**
     * used for cache, whether to use cache, it can be null if you don't need caches
     */
    AbstractKeyCacheHandler webCacheHandler;
    /**
     * pre-process: before sending the request, you can do some processing
     */
    Consumer preProcess;
    /**
     * post-process: after receiving the response, you can do some processing
     */
    BiConsumer postProcess;

    public MyHttpClient(CloseableHttpClient closeableHttpClient, HttpClientResponseHandler httpClientResponseHandler) {
        this.closeableHttpClient = closeableHttpClient;
        this.httpClientResponseHandler = httpClientResponseHandler;
    }

    /**
     * 发送之前, 经过 preProcess 进行处理, 之后将消息转给 send 方法
     */
    public static MyHttpResponse send(MyHttpRequest myHttpRequest, CloseableHttpClient closeableHttpClient,
                                      AbstractKeyCacheHandler webCacheHandler,
                                      HttpClientResponseHandler httpClientResponseHandler,
                                      Consumer preProcess,
                                      BiConsumer postProcess) throws IOException {
        // pre-processing
        if (preProcess != null) {
            preProcess.accept(myHttpRequest);
        }
        // send request
        final MyHttpResponse myHttpResponse;
        if (webCacheHandler == null) {
            myHttpResponse = MyHttpClient.doSend(myHttpRequest, closeableHttpClient, httpClientResponseHandler);
        } else {
            myHttpResponse = webCacheHandler.computeIfAbsent(myHttpRequest, () -> MyHttpClient.doSend(myHttpRequest, closeableHttpClient, httpClientResponseHandler), r -> !r.isCode(429));
        }
        // post-processing
        if (postProcess != null) {
            postProcess.accept(myHttpRequest, myHttpResponse);
        }
        return myHttpResponse;
    }

    private static  R doSend(MyHttpRequest request, CloseableHttpClient closeableHttpClient, HttpClientResponseHandler httpClientResponseHandler) throws IOException {
        return HttpUtils.http(closeableHttpClient, request.method(), request.requestUrl(),
                request.headers(), request.params(), request.jsonBody(), httpClientResponseHandler);
    }

    public MyHttpResponse send(MyHttpRequest myHttpRequest) throws IOException {
        return send(myHttpRequest, closeableHttpClient, webCacheHandler, httpClientResponseHandler, preProcess, postProcess);
    }

    public MyHttpResponse download(MyHttpRequest myHttpRequest, String savePath) throws IOException {
        MyHttpDownloadResponseHandler responseHandler = new MyHttpDownloadResponseHandler(savePath);
        return send(myHttpRequest, closeableHttpClient, null, responseHandler, preProcess, null);
    }

}