/**
* Add a request to the scheduler.
*
* <p>Returns true if the caller acquired a job slot and must send the job to execution.
*
* @param request
* @return
*/
private synchronized boolean submit(PrioritizedRequest request) {
if (_jobs.put(request.getId(), request) != null) {
throw new RuntimeException(
"Duplicate mover id detected. Please report to [email protected].");
}
if (_semaphore.tryAcquire()) {
return true;
} else {
_queue.add(request);
return false;
}
}
/**
* Get mover id for given door request. If there is no mover associated with {@code
* doorUniqueueRequest} a new mover will be created by using provided {@code moverSupplier}.
*
* <p>The returned mover id generated with following encoding: | 31- queue id -24|23- job id -0|
*
* @param moverSupplier {@link MoverSupplier} which can create a mover for given requests.
* @param doorUniqueId unique request identifier generated by the door.
* @param priority
* @return mover id
*/
public int getOrCreateMover(MoverSupplier moverSupplier, String doorUniqueId, IoPriority priority)
throws CacheException {
checkState(!_isShutdown);
try {
/* Create the request if it doesn't already exists.
*/
PrioritizedRequest request =
_moverByRequests.computeIfAbsent(
doorUniqueId,
key -> {
try {
return createRequest(moverSupplier, key, priority);
} catch (CacheException e) {
throw new RuntimeException(e);
}
});
/* If not already queued, submit it.
*/
if (request.queue()) {
if (submit(request)) {
/* There was a free slot in the queue so we submit directly to execution.
*/
sendToExecution(request);
} else if (_semaphore.getMaxPermits() <= 0) {
LOGGER.warn(
"A task was added to queue '{}', however the queue is not "
+ "configured to execute any tasks.",
_name);
}
}
return request.getId();
} catch (RuntimeException e) {
Throwables.propagateIfInstanceOf(e.getCause(), CacheException.class);
throw e;
}
}
