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 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;

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

/**
 * WaittingFutureData
 * 
 * @author patrick_guillerm
 * @since 29 sept. 2017
 */
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) {
        final CompletableFuture result = 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;
        });
        
        return result;
    }
    
}