com.liferay.portal.kernel.concurrent.test.MarkerBlockingJob Maven / Gradle / Ivy
/**
* Copyright (c) 2000-present Liferay, Inc. All rights reserved.
*
* This library is free software; you can redistribute it and/or modify it under
* the terms of the GNU Lesser General Public License as published by the Free
* Software Foundation; either version 2.1 of the License, or (at your option)
* any later version.
*
* This library 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 Lesser General Public License for more
* details.
*/
package com.liferay.portal.kernel.concurrent.test;
import java.util.concurrent.CountDownLatch;
/**
* @author Shuyang Zhou
*/
public class MarkerBlockingJob implements Runnable {
public MarkerBlockingJob() {
this(false);
}
public MarkerBlockingJob(boolean blocking) {
this(blocking, false);
}
public MarkerBlockingJob(boolean blocking, boolean throwException) {
_blocking = blocking;
_throwException = throwException;
}
public Thread getRunThread() {
return _runThread;
}
public boolean isEnded() {
return _ended;
}
public boolean isInterrupted() {
return _interrupted;
}
public boolean isStarted() {
return _started;
}
@Override
public void run() {
_runThread = Thread.currentThread();
if (_started) {
throw new IllegalStateException("Job already started");
}
_started = true;
if (_blocking) {
_waitBlockingLatch.countDown();
try {
_blockingLatch.await();
}
catch (InterruptedException ie) {
_interrupted = true;
}
}
if (_throwException) {
throw new RuntimeException();
}
_ended = true;
_endedLatch.countDown();
}
public void unBlock() {
_blockingLatch.countDown();
}
public void waitUntilBlock() throws InterruptedException {
if (!_blocking) {
throw new IllegalStateException("Blocking is not enabled");
}
_waitBlockingLatch.await();
}
public void waitUntilEnded() throws InterruptedException {
_endedLatch.await();
}
private final boolean _blocking;
private final CountDownLatch _blockingLatch = new CountDownLatch(1);
private volatile boolean _ended;
private final CountDownLatch _endedLatch = new CountDownLatch(1);
private volatile boolean _interrupted;
private volatile Thread _runThread;
private volatile boolean _started;
private final boolean _throwException;
private final CountDownLatch _waitBlockingLatch = new CountDownLatch(1);
}