it.unibo.alchemist.boundary.grid.cluster.WorkerSetImpl Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of alchemist-grid Show documentation

Alchemist Distributed Execution for grid systems

There is a newer version: 35.0.0

/*
 * Copyright (C) 2010-2023, Danilo Pianini and contributors
 * listed, for each module, in the respective subproject's build.gradle.kts file.
 *
 * This file is part of Alchemist, and is distributed under the terms of the
 * GNU General Public License, with a linking exception,
 * as described in the file LICENSE in the Alchemist distribution's top directory.
 */
package it.unibo.alchemist.boundary.grid.cluster;

import edu.umd.cs.findbugs.annotations.SuppressFBWarnings;
import it.unibo.alchemist.boundary.grid.config.RemoteGeneralSimulationConfig;
import it.unibo.alchemist.boundary.grid.simulation.RemoteSimulationImpl;
import it.unibo.alchemist.boundary.grid.simulation.SimulationSet;
import it.unibo.alchemist.boundary.grid.simulation.RemoteResult;
import it.unibo.alchemist.boundary.grid.simulation.RemoteSimulation;
import org.apache.ignite.Ignite;
import org.apache.ignite.IgniteCompute;
import org.apache.ignite.cluster.ClusterGroup;

import java.util.HashSet;
import java.util.List;
import java.util.Objects;
import java.util.Set;
import java.util.stream.Collectors;

/**
 * Implementation of {@link WorkerSet} which uses Apache Ignite.
 *
 */
public final class WorkerSetImpl implements WorkerSet {

    private final ClusterGroup grp;
    private final Ignite ignite;

    /**
     * 
     * @param ignite Ignite instance
     * @param grp workers' group
     */
    @SuppressFBWarnings(value = "EI_EXPOSE_REP2", justification = "This is intentional")
    public WorkerSetImpl(final Ignite ignite, final ClusterGroup grp) {
        this.grp = Objects.requireNonNull(grp);
        this.ignite = Objects.requireNonNull(ignite);
    }

    @Override
    public Set distributeSimulations(final SimulationSet simulationsSet) {
        final IgniteCompute compute = this.ignite.compute(this.grp);
        try (RemoteGeneralSimulationConfig gc = new RemoteGeneralSimulationConfig(
                simulationsSet.getGeneralSimulationConfig(),
                this.ignite
        )) {
            final List> jobs = simulationsSet.getSimulationConfigs().stream()
                    .map(e -> new RemoteSimulationImpl<>(gc, e, ignite.cluster().localNode().id()))
                    .collect(Collectors.toList());
            return new HashSet<>(compute.call(jobs));
        }
    }

}