cn.icanci.loopstack.rec.engine.script.client.AbstractRetryClient Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of rec-engine-script Show documentation

The newest version!

package cn.icanci.loopstack.rec.engine.script.client;

import java.util.concurrent.*;

/**
 * 自动重试客户端
 *
 * @author icanci
 * @since 1.0 Created in 2022/11/14 22:16
 */
public abstract class AbstractRetryClient implements Client {

    protected static final ThreadPoolExecutor HTTP_POOL = new ThreadPoolExecutor(40, //
        120, //
        60L, //
        TimeUnit.SECONDS, //
        new LinkedBlockingQueue<>(2000), //
        runnable -> new Thread(runnable, "AbstractRetryClient Pool-" + runnable.hashCode()), //
        (r, executor) -> {
            throw new RuntimeException("AbstractRetryClient Pool is EXHAUSTED!");
        });

    @Override
    public  V call(RpcRequest request, Class clazz) throws RemoteException {
        return retry(request, clazz, 0, request.getRetry());
    }

    private  V retry(RpcRequest request, Class clazz, int retryCount, int retry) throws RemoteException {
        try {
            return doExecute(request, clazz);
        } catch (RemoteException | ExecutionException | InterruptedException e) {
            throw new RemoteException(e);
        } catch (TimeoutException e) {
            while (retryCount < retry) {
                retryCount++;
                return retry(request, clazz, retry, retryCount);
            }
            throw new RemoteException(e.getMessage(), e);
        }
    }

    protected abstract  V doExecute(RpcRequest request, Class clazz) throws ExecutionException, InterruptedException, TimeoutException;

}