org.yamcs.activities.OngoingActivity Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of yamcs-core Show documentation

There is a newer version: 5.10.7

package org.yamcs.activities;

import java.util.concurrent.CompletableFuture;
import java.util.concurrent.Future;

import org.yamcs.security.User;

public class OngoingActivity {

    private final Activity activity;

    // Underlying work. This can be cancelled, after which the resultFuture triggers
    Future workFuture;

    // Future that updates state following completion of the work
    CompletableFuture resultFuture;

    private User stopRequester;

    public OngoingActivity(Activity activity) {
        this.activity = activity;
    }

    public Activity getActivity() {
        return activity;
    }

    public CompletableFuture onResult() {
        return resultFuture;
    }

    /**
     * If a stop has been requested, this returns the first user that asked to cancel.
     */
    public User getStopRequester() {
        return stopRequester;
    }

    /**
     * Request a stop of this activity. This method does not block.
     */
    public void cancel(User user) {
        activity.setStatus(ActivityStatus.CANCELLED);
        if (stopRequester == null) {
            stopRequester = user;
        }
        workFuture.cancel(true);
    }

    public void complete(User user) {
        verifyManualActivity();
        stopRequester = user;
        ((CompletableFuture) workFuture).complete(null);
    }

    public void completeExceptionally(String failureReason, User user) {
        verifyManualActivity();
        stopRequester = user;
        ((CompletableFuture) workFuture).completeExceptionally(new ManualFailureException(failureReason));
    }

    private void verifyManualActivity() {
        if (!ActivityService.ACTIVITY_TYPE_MANUAL.equals(activity.getType())) {
            throw new UnsupportedOperationException("Cannot complete a non-manual activity");
        }
    }
}