io.inugami.commons.threads.WaittingFutureData Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of inugami_commons Show documentation
There is a newer version: 3.3.5
/* --------------------------------------------------------------------
 *  Inugami
 * --------------------------------------------------------------------
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, version 3.
 *
 * This program is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
 * General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program. If not, see .
 */
package io.inugami.commons.threads;

import io.inugami.api.providers.concurrent.FutureData;

import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.CompletableFuture;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.TimeoutException;
import java.util.function.BiConsumer;

/**
 * WaittingFutureData
 *
 * @author patrick_guillerm
 * @since 29 sept. 2017
 */
@SuppressWarnings({"java:S2142"})
public class WaittingFutureData {

    // =========================================================================
    // METHODS
    // =========================================================================
    public List wait(final List> futures) {
        return wait(futures, null, null);
    }

    public List wait(final List> futures, final BiConsumer> onDone) {
        return wait(futures, onDone, null);
    }

    public List wait(final List> futures, final BiConsumer> onDone,
                        final BiConsumer> onError) {
        final List result = new ArrayList<>();
        //@formatter:off
        final BiConsumer>           functionOnDone  = onDone  != null ? onDone  : (data,task) -> {};
        final BiConsumer>  functionOnError = onError != null ? onError : (error,task) -> {};
        //@formatter:on

        final List> completableFutures = new ArrayList<>();
        for (final FutureData futureData : futures) {
            completableFutures.add(buildFuture(futureData, functionOnDone, functionOnError));
        }

        final CompletableFuture[] type = {};
        CompletableFuture.allOf(completableFutures.toArray(type));

        return result;
    }

    // =========================================================================
    // BUILDER
    // =========================================================================
    private CompletableFuture buildFuture(final FutureData futureData, final BiConsumer> onDone,
                                             final BiConsumer> onError) {
        return CompletableFuture.supplyAsync(() -> {
            T returnResult = null;

            try {
                final T futureResult = futureData.getFuture().get(futureData.getTimeout(), futureData.getTimeUnit());
                returnResult = futureResult;
                onDone.accept(futureResult, futureData);
                if (futureData.onDone() != null) {
                    futureData.onDone().forEach(func -> func.onDone(futureResult, futureData.getEvent(),
                                                                    futureData.getChannel()));
                }

            } catch (InterruptedException | ExecutionException | TimeoutException e) {
                onError.accept(e, futureData);
                if (futureData.onError() != null) {
                    futureData.onError().forEach(func -> func.onError(futureData.getEvent(), futureData.getChannel(),
                                                                      futureData.getTask(), e));
                }
            }

            return returnResult;
        });
    }

}