bear.core.BearScriptPhase Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of bear Show documentation
The newest version!
package bear.core;

import bear.console.ConsolesDivider;
import bear.console.GroupDivider;
import bear.main.BearFX;
import bear.main.event.PhaseFinishedEventToUI;
import bear.task.TaskDef;
import bear.task.TaskResult;
import chaschev.util.CatchyCallable;
import com.google.common.base.Preconditions;

import java.util.List;
import java.util.concurrent.Callable;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicInteger;
import java.util.concurrent.atomic.AtomicLong;

/**
* @author Andrey Chaschev [email protected]
*/
public class BearScriptPhase> {
    public final String id = SessionContext.randomId();

    TaskDef taskDef;

    final AtomicInteger partiesArrived = new AtomicInteger();
    public final AtomicInteger partiesOk = new AtomicInteger();

    final AtomicLong minimalOkDuration = new AtomicLong(-1);

    BearFX bearFX;

    final GroupDivider groupDivider;

    protected long startedAtMs;

    int partiesCount = -1;

    public volatile int partiesPending;
    public volatile int partiesFailed = 0;

    public BearScriptPhase(TaskDef taskDef, BearFX bearFX, GroupDivider groupDivider) {
        this.taskDef = taskDef;
        this.bearFX = bearFX;
        this.groupDivider = groupDivider;
    }

    public void init(List $s){
        groupDivider.init($s);
        this.partiesCount = $s.size();
        startedAtMs = System.currentTimeMillis(); //possibly wrong

    }

    public String getName() {
        return taskDef.getName();
    }

    public String getDisplayName() {
        return taskDef.getDisplayName();
    }

    public void addArrival(final SessionContext $, final long duration, TaskResult result) {
        Preconditions.checkArgument(partiesCount != -1, "BearScriptPhase is not initialized");

        groupDivider.addArrival($);

        if (result.ok()) {
            partiesOk.incrementAndGet();

            if (minimalOkDuration.compareAndSet(-1, duration)) {
                $.getGlobal().scheduler.schedule(new CatchyCallable(new Callable() {
                    @Override
                    public Void call() throws Exception {
                        boolean haveHangUpJobs;
                        boolean alreadyFinished;

                        if (partiesArrived.compareAndSet(partiesCount, -1)) {
                            alreadyFinished = false;
                            haveHangUpJobs = false;
                        } else {
                            if (partiesArrived.compareAndSet(-1, -1)) {
                                haveHangUpJobs = false;
                                alreadyFinished = true;
                            } else {
                                alreadyFinished = false;
                                haveHangUpJobs = true;
                            }
                        }

                        if (!alreadyFinished) {
                            sendAllFinishedResults($, duration);
                        }
                        return null;
                    }
                }), duration * 3, TimeUnit.MILLISECONDS);
            }
        }


        partiesArrived.incrementAndGet();

        partiesPending = partiesCount - partiesArrived.get();
        partiesFailed = partiesArrived.get() - partiesOk.get();

        if (partiesArrived.compareAndSet(partiesCount, -1)) {
            sendAllFinishedResults($, duration);
        }
    }

    private void sendAllFinishedResults(SessionContext $, long duration) {
        List groups = groupDivider.divideIntoGroups();

        SessionContext.ui.info(
            new PhaseFinishedEventToUI(duration, groups, getName())
                .setParentId(id));
    }

    @Override
    public String toString() {
        final StringBuilder sb = new StringBuilder("BearScriptPhase{");
        sb.append("id='").append(id).append('\'');
        sb.append(", taskDef=").append(taskDef);
        sb.append(", partiesPending=").append(partiesPending);
        sb.append('}');
        return sb.toString();
    }
}