/*
* JBoss, Home of Professional Open Source.
* Copyright 2012, Red Hat, Inc., and individual contributors
* as indicated by the @author tags. See the copyright.txt file in the
* distribution for a full listing of individual contributors.
*
* This 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 software 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.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this software; if not, write to the Free
* Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
* 02110-1301 USA, or see the FSF site: http://www.fsf.org.
*/
package org.jboss.as.domain.controller.plan;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.concurrent.TimeUnit;
import javax.security.auth.Subject;
import org.jboss.as.controller.BlockingTimeout;
import org.jboss.as.controller.remote.TransactionalProtocolClient;
import org.jboss.as.domain.controller.ServerIdentity;
import org.jboss.as.domain.controller.logging.DomainControllerLogger;
/**
* @author Emanuel Muckenhuber
*/
class ConcurrentServerGroupUpdateTask extends AbstractServerGroupRolloutTask implements Runnable {
public ConcurrentServerGroupUpdateTask(List<ServerUpdateTask> tasks, ServerUpdatePolicy updatePolicy,
ServerTaskExecutor executor, Subject subject, BlockingTimeout blockingTimeout) {
super(tasks, updatePolicy, executor, subject, blockingTimeout);
}
@Override
public void execute() {
final Map<ServerIdentity, ServerUpdateTask> outstanding = new HashMap<>();
final ServerTaskExecutor.ServerOperationListener listener = new ServerTaskExecutor.ServerOperationListener();
int preparedTimeout = 0;
for(final ServerUpdateTask task : tasks) {
final ServerIdentity identity = task.getServerIdentity();
if (updatePolicy.canUpdateServer(identity) && !Thread.currentThread().isInterrupted()) {
// Execute the task
int serverTimeout = executor.executeTask(listener, task);
if (serverTimeout > -1) {
outstanding.put(task.getServerIdentity(), task);
if (serverTimeout > preparedTimeout) {
preparedTimeout = serverTimeout;
}
}
} else {
DomainControllerLogger.HOST_CONTROLLER_LOGGER.tracef("Skipping server update task for %s", identity);
}
}
boolean interrupted = false;
long deadline = System.currentTimeMillis() + preparedTimeout;
long remaining = preparedTimeout;
while (!interrupted && !outstanding.isEmpty() && remaining > 0) {
try {
// Wait for all prepared results
final TransactionalProtocolClient.PreparedOperation<ServerTaskExecutor.ServerOperation> prepared = listener.retrievePreparedOperation(remaining, TimeUnit.MILLISECONDS);
if (prepared == null) {
// timed out
break;
}
final ServerIdentity identity = prepared.getOperation().getIdentity();
recordPreparedOperation(identity, prepared);
outstanding.remove(identity);
} catch (InterruptedException e) {
interrupted = true;
}
remaining = deadline - System.currentTimeMillis();
}
if (!outstanding.isEmpty()) {
if (interrupted) {
DomainControllerLogger.HOST_CONTROLLER_LOGGER.interruptedAwaitingPreparedResponse(getClass().getSimpleName(), outstanding.keySet());
} else {
DomainControllerLogger.HOST_CONTROLLER_LOGGER.timedOutAwaitingPreparedResponse(getClass().getSimpleName(), preparedTimeout, outstanding.keySet());
}
for (Map.Entry<ServerIdentity, ServerUpdateTask> entry : outstanding.entrySet()) {
ServerIdentity identity = entry.getKey();
executor.cancelTask(identity);
if (!interrupted) {
handlePreparePhaseTimeout(identity, entry.getValue(), preparedTimeout);
}
}
}
if(interrupted) {
Thread.currentThread().interrupt();
}
}
}