@Override
public final ListenableFuture<State> stop() {
lock.lock();
try {
if (state == State.NEW) {
state = State.TERMINATED;
terminated(State.NEW);
startup.set(State.TERMINATED);
shutdown.set(State.TERMINATED);
} else if (state == State.STARTING) {
shutdownWhenStartupFinishes = true;
startup.set(State.STOPPING);
} else if (state == State.RUNNING) {
state = State.STOPPING;
stopping(State.RUNNING);
doStop();
}
} catch (Throwable shutdownFailure) {
// put the exception in the future, the user can get it via Future.get()
notifyFailed(shutdownFailure);
} finally {
lock.unlock();
}
return shutdown;
}
public void testNotifyFailedWhenNotStarted() {
AbstractService service = new DefaultService();
try {
service.notifyFailed(new Exception());
fail();
} catch (IllegalStateException expected) {
}
}
/**
* Implementing classes should invoke this method once their service has stopped. It will cause
* the service to transition from {@link State#STOPPING} to {@link State#TERMINATED}.
*
* @throws IllegalStateException if the service is neither {@link State#STOPPING} nor {@link
* State#RUNNING}.
*/
protected final void notifyStopped() {
lock.lock();
try {
if (state != State.STOPPING && state != State.RUNNING) {
IllegalStateException failure =
new IllegalStateException("Cannot notifyStopped() when the service is " + state);
notifyFailed(failure);
throw failure;
}
terminated(state);
shutdown.set(State.TERMINATED);
} finally {
lock.unlock();
}
}
@Override
public final ListenableFuture<State> start() {
lock.lock();
try {
if (state == State.NEW) {
starting();
doStart();
}
} catch (Throwable startupFailure) {
// put the exception in the future, the user can get it via Future.get()
notifyFailed(startupFailure);
} finally {
lock.unlock();
}
return startup;
}
/**
* Implementing classes should invoke this method once their service has started. It will cause
* the service to transition from {@link State#STARTING} to {@link State#RUNNING}.
*
* @throws IllegalStateException if the service is not {@link State#STARTING}.
*/
protected final void notifyStarted() {
lock.lock();
try {
if (state != State.STARTING) {
IllegalStateException failure =
new IllegalStateException("Cannot notifyStarted() when the service is " + state);
notifyFailed(failure);
throw failure;
}
running();
if (shutdownWhenStartupFinishes) {
stop();
} else {
startup.set(State.RUNNING);
}
} finally {
lock.unlock();
}
}