/*
* JBoss, Home of Professional Open Source.
* Copyright 2013, 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.wildfly.plugin.deployment.domain;
import java.io.File;
import java.io.IOException;
import java.util.List;
import java.util.Map;
import java.util.UUID;
import java.util.concurrent.ExecutionException;
import org.jboss.as.controller.client.helpers.domain.DeploymentActionResult;
import org.jboss.as.controller.client.helpers.domain.DeploymentActionsCompleteBuilder;
import org.jboss.as.controller.client.helpers.domain.DeploymentPlan;
import org.jboss.as.controller.client.helpers.domain.DeploymentPlanBuilder;
import org.jboss.as.controller.client.helpers.domain.DeploymentPlanResult;
import org.jboss.as.controller.client.helpers.domain.DomainClient;
import org.jboss.as.controller.client.helpers.domain.DomainDeploymentManager;
import org.jboss.as.controller.client.helpers.domain.DuplicateDeploymentNameException;
import org.jboss.as.controller.client.helpers.domain.ServerGroupDeploymentActionResult;
import org.jboss.as.controller.client.helpers.domain.ServerGroupDeploymentPlanBuilder;
import org.jboss.as.controller.client.helpers.domain.ServerIdentity;
import org.jboss.as.controller.client.helpers.domain.ServerStatus;
import org.jboss.as.controller.client.helpers.domain.ServerUpdateResult;
import org.wildfly.plugin.common.DeploymentInspector;
import org.wildfly.plugin.deployment.DeploymentExecutionException;
import org.wildfly.plugin.deployment.DeploymentFailureException;
import org.wildfly.plugin.deployment.MatchPatternStrategy;
/**
* Based on org.wildfly.plugin.deployment.domain.DomainDeployment.
*
* @author <a href="mailto:jperkins@redhat.com">James R. Perkins</a>
* @author <a href="mitrovic.dejan@gmail.com">Dejan Mitrovic</a>
*/
public class DomainDeployment
{
public static enum Type
{
DEPLOY, FORCE_DEPLOY, UNDEPLOY, UNDEPLOY_IGNORE_MISSING, REDEPLOY,
}
public static enum Status
{
SUCCESS, REQUIRES_RESTART
}
private final File content;
private final DomainClient client;
private final Domain domain;
private final String name;
private final Type type;
private final String matchPattern;
private final MatchPatternStrategy matchPatternStrategy;
/**
* Creates a new deployment.
*
* @param client the client for the server
* @param domain the domain information
* @param content the content for the deployment
* @param name the name of the deployment, if {@code null} the name of the content file is used
* @param type the deployment type
* @param matchPattern the pattern for matching multiple artifacts, if {@code null} the name is
* used.
* @param matchPatternStrategy the strategy for handling multiple artifacts.
*/
public DomainDeployment(final DomainClient client, final Domain domain, final File content,
final String name, final Type type, final String matchPattern,
final MatchPatternStrategy matchPatternStrategy)
{
this.content = content;
this.client = client;
this.domain = domain;
this.name = (name == null ? content.getName() : name);
this.type = type;
this.matchPattern = matchPattern;
this.matchPatternStrategy = matchPatternStrategy;
}
/**
* Creates a new deployment.
*
* @param client the client for the server
* @param domain the domain information
* @param content the content for the deployment
* @param name the name of the deployment, if {@code null} the name of the content file is used
* @param type the deployment type
* @param matchPattern the pattern for matching multiple artifacts, if {@code null} the name is
* used.
* @param matchPatternStrategy the strategy for handling multiple artifacts.
*
* @return the new deployment
*/
public static DomainDeployment create(final DomainClient client, final Domain domain,
final File content, final String name, final Type type, final String matchPattern,
final MatchPatternStrategy matchPatternStrategy)
{
return new DomainDeployment(client, domain, content, name, type, matchPattern,
matchPatternStrategy);
}
private DeploymentPlan createPlan(final DeploymentPlanBuilder builder) throws IOException,
DuplicateDeploymentNameException, DeploymentFailureException
{
DeploymentActionsCompleteBuilder completeBuilder = null;
List<String> existingDeployments = DeploymentInspector.getDeployments(client, name, matchPattern);
switch (type)
{
case DEPLOY:
{
completeBuilder = builder.add(name, content).andDeploy();
break;
}
case FORCE_DEPLOY:
{
if (existingDeployments.contains(name))
{
completeBuilder = builder.replace(name, content);
} else
{
completeBuilder = builder.add(name, content).andDeploy();
}
break;
}
case REDEPLOY:
{
completeBuilder = builder.replace(name, content);
break;
}
case UNDEPLOY:
{
validateExistingDeployments(existingDeployments);
completeBuilder = undeployAndRemoveUndeployed(builder, existingDeployments);
break;
}
case UNDEPLOY_IGNORE_MISSING:
{
validateExistingDeployments(existingDeployments);
if (!existingDeployments.isEmpty())
{
completeBuilder = undeployAndRemoveUndeployed(builder, existingDeployments);
} else
{
return null;
}
}
}
if (completeBuilder != null)
{
ServerGroupDeploymentPlanBuilder groupDeploymentBuilder = null;
for (String serverGroupName : domain.getServerGroups())
{
groupDeploymentBuilder = (groupDeploymentBuilder == null ? completeBuilder
.toServerGroup(serverGroupName) : groupDeploymentBuilder
.toServerGroup(serverGroupName));
}
if (groupDeploymentBuilder == null)
{
throw new DeploymentFailureException(
"No server groups were defined for the deployment.");
}
return groupDeploymentBuilder.withRollback().build();
}
throw new IllegalStateException(String.format("Invalid type '%s' for deployment", type));
}
private DeploymentActionsCompleteBuilder undeployAndRemoveUndeployed(
final DeploymentPlanBuilder builder, final List<String> deploymentNames)
{
DeploymentActionsCompleteBuilder completeBuilder = null;
for (String deploymentName : deploymentNames)
{
completeBuilder = builder.undeploy(deploymentName).andRemoveUndeployed();
if (matchPatternStrategy == MatchPatternStrategy.FIRST)
{
break;
}
}
return completeBuilder;
}
private void validateExistingDeployments(List<String> existingDeployments)
throws DeploymentFailureException
{
if (matchPattern == null)
{
return;
}
if (matchPatternStrategy == MatchPatternStrategy.FAIL && existingDeployments.size() > 1)
{
throw new DeploymentFailureException(String.format(
"Deployment failed, found %d deployed artifacts for pattern '%s' (%s)",
existingDeployments.size(), matchPattern, existingDeployments));
}
}
public Status execute() throws DeploymentExecutionException, DeploymentFailureException
{
try
{
validate();
final DomainDeploymentManager manager = client.getDeploymentManager();
final DeploymentPlanBuilder builder = manager.newDeploymentPlan();
DeploymentPlan plan = createPlan(builder);
if (plan != null)
{
executePlan(manager, plan);
}
} catch (DeploymentFailureException e)
{
throw e;
} catch (DeploymentExecutionException e)
{
throw e;
} catch (Exception e)
{
throw new DeploymentExecutionException(e, "Error executing %s", type);
}
return Status.SUCCESS;
}
void validate() throws DeploymentFailureException
{
final Map<ServerIdentity, ServerStatus> statuses = client.getServerStatuses();
// Check for NPE
final List<String> serverGroups = domain.getServerGroups();
for (String serverGroup : serverGroups)
{
boolean notFound = true;
// Check the servers
for (ServerIdentity serverId : statuses.keySet())
{
if (serverGroup.equals(serverId.getServerGroupName()))
{
ServerStatus currentStatus = statuses.get(serverId);
if (currentStatus != ServerStatus.STARTED)
{
throw new DeploymentFailureException(
"Status of server group '%s' is '%s', but is required to be '%s'.",
serverGroup, currentStatus, ServerStatus.STARTED);
}
notFound = false;
break;
}
}
if (notFound)
{
throw new DeploymentFailureException(
"Server group '%s' does not exist on the server.", serverGroup);
}
}
}
public Type getType()
{
return type;
}
private void executePlan(final DomainDeploymentManager manager, final DeploymentPlan plan)
throws DeploymentExecutionException, ExecutionException, InterruptedException
{
if (plan.getDeploymentActions().size() > 0)
{
final DeploymentPlanResult planResult = manager.execute(plan).get();
final Map<UUID, DeploymentActionResult> actionResults = planResult
.getDeploymentActionResults();
for (UUID uuid : actionResults.keySet())
{
final Map<String, ServerGroupDeploymentActionResult> groupDeploymentActionResults = actionResults
.get(uuid).getResultsByServerGroup();
for (String serverGroup2 : groupDeploymentActionResults.keySet())
{
final Map<String, ServerUpdateResult> serverUpdateResults = groupDeploymentActionResults
.get(serverGroup2).getResultByServer();
for (String server : serverUpdateResults.keySet())
{
final Throwable t = serverUpdateResults.get(server).getFailureResult();
if (t != null)
{
throw new DeploymentExecutionException(t, "Error executing %s", type);
}
}
}
}
}
}
}