package org.jboss.deployment;
import java.io.BufferedInputStream;
import java.io.BufferedOutputStream;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.net.MalformedURLException;
import java.net.URL;
import java.net.URLClassLoader;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Collections;
import java.util.Comparator;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Iterator;
import java.util.LinkedList;
import java.util.List;
import java.util.ListIterator;
import java.util.Map;
import java.util.StringTokenizer;
import java.util.jar.Attributes;
import java.util.jar.Manifest;
import javax.management.JMException;
import javax.management.MBeanServer;
import javax.management.MalformedObjectNameException;
import javax.management.Notification;
import javax.management.ObjectName;
import org.jboss.mx.util.JMXExceptionDecoder;
import org.jboss.system.ServiceContext;
import org.jboss.system.ServiceMBeanSupport;
import org.jboss.system.server.ServerConfig;
import org.jboss.system.server.ServerConfigLocator;
import org.jboss.util.file.Files;
import org.jboss.util.file.JarUtils;
import org.jboss.util.stream.Streams;
public class MainDeployer
extends ServiceMBeanSupport
implements Deployer, MainDeployerMBean
{
private ObjectName serviceController;
private final LinkedList deployers = new LinkedList();
private final Map deploymentMap = Collections.synchronizedMap(new HashMap());
private final List deploymentList = new ArrayList();
private final List waitingDeployments = new ArrayList();
private final DeploymentSorter sorter = new DeploymentSorter();
private final Comparator infoSorter = new DeploymentInfoComparator(sorter);
private final SuffixOrderHelper suffixOrderHelper = new SuffixOrderHelper(sorter);
private boolean copyFiles = true;
private File tempDir;
private String tempDirString;
public MainDeployer()
{
String localCopy = System.getProperty("jboss.deploy.localcopy");
if (localCopy != null && (
localCopy.equalsIgnoreCase("false") ||
localCopy.equalsIgnoreCase("no") ||
localCopy.equalsIgnoreCase("off")))
{
log.debug("Disabling local copies of file: urls");
copyFiles = false;
}
}
public boolean getCopyFiles()
{
return copyFiles;
}
public void setCopyFiles(boolean copyFiles)
{
this.copyFiles = copyFiles;
}
public File getTempDir()
{
return tempDir;
}
public void setTempDir(File tempDir)
{
this.tempDir = tempDir;
}
public String getTempDirString()
{
return tempDirString;
}
public String[] getSuffixOrder()
{
return suffixOrderHelper.getSuffixOrder();
}
public String[] getEnhancedSuffixOrder()
{
return suffixOrderHelper.getEnhancedSuffixOrder();
}
public void setEnhancedSuffixOrder(String[] enhancedSuffixOrder)
{
suffixOrderHelper.setEnhancedSuffixOrder(enhancedSuffixOrder);
}
public void setServiceController(final ObjectName serviceController)
{
this.serviceController = serviceController;
}
public Collection listDeployed()
{
synchronized (deploymentList)
{
log.info("deployment list string: " + deploymentList);
return new ArrayList(deploymentList);
}
}
public Collection listDeployedModules()
{
log.debug("listDeployedModules");
HashMap map = new HashMap();
synchronized (deploymentList)
{
Collection col = new ArrayList(deploymentList);
for (Iterator it = col.iterator(); it.hasNext();)
{
DeploymentInfo info = (DeploymentInfo) it.next();
map.put(info.url, new SerializableDeploymentInfo(info));
}
for (Iterator it = col.iterator(); it.hasNext();)
{
DeploymentInfo info = (DeploymentInfo) it.next();
SerializableDeploymentInfo sinfo = (SerializableDeploymentInfo)map.get(info.url);
if (info.parent != null)
{
sinfo.parent = (SerializableDeploymentInfo)map.get(info.parent.url);
sinfo.parent.subDeployments.add(sinfo);
}
}
}
return new ArrayList(map.values());
}
public String listDeployedAsString()
{
return "<pre>" + listDeployed() + "</pre>";
}
public Collection listIncompletelyDeployed()
{
List id = new ArrayList();
List copy;
synchronized (deploymentList)
{
copy = new ArrayList(deploymentList);
}
for (Iterator i = copy.iterator(); i.hasNext();)
{
DeploymentInfo di = (DeploymentInfo)i.next();
if (!"Deployed".equals(di.status) && !"Starting".equals(di.status))
{
id.add(di);
}
} return id;
}
public Collection listWaitingForDeployer()
{
synchronized (waitingDeployments)
{
return new ArrayList(waitingDeployments);
}
}
public void addDeployer(final SubDeployer deployer)
{
log.debug("Adding deployer: " + deployer);
synchronized(deployers)
{
deployers.addFirst(deployer);
suffixOrderHelper.addSuffixes(deployer.getSuffixes(), deployer.getRelativeOrder());
}
Notification msg = new Notification(ADD_DEPLOYER, this, getNextNotificationSequenceNumber());
msg.setUserData(deployer.getServiceName());
sendNotification(msg);
synchronized (waitingDeployments)
{
List copy = new ArrayList(waitingDeployments);
waitingDeployments.clear();
for (Iterator i = copy.iterator(); i.hasNext();)
{
DeploymentInfo di = (DeploymentInfo)i.next();
log.debug("trying to deploy with new deployer: " + di.shortName);
try
{
di.setServer(server);
deploy(di);
}
catch (DeploymentException e)
{
log.error("DeploymentException while trying to deploy a package with a new deployer", e);
}
} }
}
public void removeDeployer(final SubDeployer deployer)
{
log.debug("Removing deployer: " + deployer);
boolean removed = false;
synchronized(deployers)
{
removed = deployers.remove(deployer);
suffixOrderHelper.removeSuffixes(deployer.getSuffixes(), deployer.getRelativeOrder());
}
if( removed )
{
Notification msg = new Notification(REMOVE_DEPLOYER, this,
getNextNotificationSequenceNumber());
msg.setUserData(deployer.getServiceName());
sendNotification(msg);
}
List copy;
synchronized (deploymentList)
{
copy = new ArrayList(deploymentList);
}
for (Iterator i = copy.iterator(); i.hasNext(); )
{
DeploymentInfo di = (DeploymentInfo)i.next();
if (di.deployer == deployer)
{
undeploy(di);
di.deployer = null;
synchronized (waitingDeployments)
{
waitingDeployments.add(di);
}
}
}
}
public Collection listDeployers()
{
ArrayList deployerNames = new ArrayList();
synchronized(deployers)
{
for(int n = 0; n < deployers.size(); n ++)
{
SubDeployer deployer = (SubDeployer) deployers.get(n);
ObjectName name = deployer.getServiceName();
deployerNames.add(name);
}
}
return deployerNames;
}
protected ObjectName getObjectName(MBeanServer server, ObjectName name)
throws MalformedObjectNameException
{
return name == null ? OBJECT_NAME : name;
}
protected void createService() throws Exception
{
ServerConfig config = ServerConfigLocator.locate();
File basedir = config.getServerTempDir();
tempDir = new File(basedir, "deploy");
Files.delete(tempDir);
tempDir.mkdirs();
tempDirString = tempDir.toURL().toString();
suffixOrderHelper.init();
}
public void shutdown()
{
int deployCounter = 0;
List copy;
synchronized (deploymentList)
{
copy = new ArrayList(deploymentList);
}
for (ListIterator i = copy.listIterator(copy.size()); i.hasPrevious(); )
{
try
{
undeploy((DeploymentInfo)i.previous(), true);
deployCounter++;
}
catch (Exception e)
{
log.info("exception trying to undeploy during shutdown", e);
}
}
log.debug("Undeployed " + deployCounter + " deployed packages");
}
public void redeploy(String urlspec)
throws DeploymentException, MalformedURLException
{
redeploy(new URL(urlspec));
}
public void redeploy(URL url) throws DeploymentException
{
DeploymentInfo sdi = getDeployment(url);
if (sdi!= null)
{
redeploy(sdi);
}
else
{
deploy(url);
} }
public void redeploy(DeploymentInfo sdi) throws DeploymentException
{
try
{
undeploy(sdi);
}
catch (Throwable t)
{
log.info("Throwable from undeployment attempt: ", t);
} sdi.setServer(server);
deploy(sdi);
}
public void undeploy(URL url) throws DeploymentException
{
DeploymentInfo sdi = getDeployment(url);
if (sdi!= null)
{
undeploy(sdi);
}
}
public void undeploy(String urlspec)
throws DeploymentException, MalformedURLException
{
undeploy(new URL(urlspec));
}
public void undeploy(DeploymentInfo di)
{
undeploy(di, false);
}
protected void undeploy(DeploymentInfo di, boolean isShutdown)
{
log.debug("Undeploying "+di.url);
stop(di);
destroy(di);
}
private void stop(DeploymentInfo di)
{
ArrayList reverseSortedSubs = new ArrayList(di.subDeployments);
Collections.sort(reverseSortedSubs, infoSorter);
Collections.reverse(reverseSortedSubs);
for (Iterator subs = reverseSortedSubs.iterator(); subs.hasNext();)
{
DeploymentInfo sub = (DeploymentInfo) subs.next();
log.debug("Stopping sub deployment: "+sub.url);
stop(sub);
}
try
{
if (di.deployer != null)
{
di.deployer.stop(di);
}
}
catch (Throwable t)
{
log.error("Undeployment failed: " + di.url, t);
}
}
private void destroy(DeploymentInfo di)
{
ArrayList reverseSortedSubs = new ArrayList(di.subDeployments);
Collections.sort(reverseSortedSubs, infoSorter);
Collections.reverse(reverseSortedSubs);
for (Iterator subs = reverseSortedSubs.iterator(); subs.hasNext();)
{
DeploymentInfo sub = (DeploymentInfo) subs.next();
log.debug("Destroying sub deployment: "+sub.url);
destroy(sub);
}
try
{
if (di.deployer != null)
{
di.deployer.destroy(di);
}
}
catch (Throwable t)
{
log.error("Undeployment failed: " + di.url, t);
}
try
{
synchronized (deploymentList)
{
deploymentMap.remove(di.url);
if (deploymentList.lastIndexOf(di) != -1)
{
deploymentList.remove(deploymentList.lastIndexOf(di));
}
}
synchronized (waitingDeployments)
{
waitingDeployments.remove(di);
}
di.cleanup();
log.debug("Undeployed "+di.url);
}
catch (Throwable t)
{
log.error("Undeployment cleanup failed: " + di.url, t);
}
}
public void deploy(String urlspec)
throws DeploymentException, MalformedURLException
{
URL url;
try
{
url = new URL(urlspec);
}
catch (MalformedURLException e)
{
File file = new File(urlspec);
url = file.toURL();
}
deploy(url);
}
public void deploy(URL url) throws DeploymentException
{
DeploymentInfo sdi = getDeployment(url);
if (sdi == null)
{
sdi = new DeploymentInfo(url, null, getServer());
deploy(sdi);
}
if( sdi.state != DeploymentState.STARTED )
checkIncompleteDeployments();
}
public void deploy(DeploymentInfo deployment)
throws DeploymentException
{
if (isDeployed(deployment.url))
{
log.info("Package: " + deployment.url + " is already deployed");
return;
}
log.debug("Starting deployment of package: " + deployment.url);
boolean inited = false;
try
{
inited = init(deployment);
}
catch (Throwable t)
{
log.error("Could not initialise deployment: " + deployment.url, t);
DeploymentException.rethrowAsDeploymentException("Could not initialise deployment: " + deployment.url, t);
}
if ( inited )
{
create(deployment);
start(deployment);
log.debug("Deployed package: " + deployment.url);
} else
{
log.debug("Deployment of package: " + deployment.url + " is waiting for an appropriate deployer.");
} }
private boolean init(DeploymentInfo deployment) throws DeploymentException
{
if (isDeployed(deployment.url))
{
log.info("Package: " + deployment.url + " is already deployed");
return false;
}
log.debug("Starting deployment (init step) of package at: " + deployment.url);
try
{
if (deployment.localUrl == null)
{
makeLocalCopy(deployment);
URL[] localCl = new URL[]{deployment.localUrl};
deployment.localCl = new URLClassLoader(localCl);
}
findDeployer(deployment);
if(deployment.deployer == null)
{
deployment.state = DeploymentState.INIT_WAITING_DEPLOYER;
log.debug("deployment waiting for deployer: " + deployment.url);
synchronized (waitingDeployments)
{
if (waitingDeployments.contains(deployment) == false)
waitingDeployments.add(deployment);
}
return false;
}
deployment.state = DeploymentState.INIT_DEPLOYER;
deployment.deployer.init(deployment);
deployment.createClassLoaders();
deployment.state = DeploymentState.INITIALIZED;
synchronized (deploymentList)
{
deploymentMap.put(deployment.url, deployment);
}
parseManifestLibraries(deployment);
log.debug("found " + deployment.subDeployments.size() + " subpackages of " + deployment.url);
ArrayList sortedSubs = new ArrayList(deployment.subDeployments);
Collections.sort(sortedSubs, infoSorter);
for (Iterator lt = sortedSubs.listIterator(); lt.hasNext();)
{
init((DeploymentInfo) lt.next());
}
}
catch (Exception e)
{
deployment.state = DeploymentState.FAILED;
DeploymentException.rethrowAsDeploymentException("exception in init of " + deployment.url, e);
}
finally
{
try
{
URL url = deployment.localUrl == null ? deployment.url : deployment.localUrl;
long lastModified = -1;
if (url.getProtocol().equals("file"))
lastModified = new File(url.getFile()).lastModified();
else
lastModified = url.openConnection().getLastModified();
deployment.lastModified=lastModified;
deployment.lastDeployed=System.currentTimeMillis();
}
catch (IOException ignore)
{
deployment.lastModified=System.currentTimeMillis();
deployment.lastDeployed=System.currentTimeMillis();
}
synchronized (deploymentList)
{
if (!inLocalCopyDir(deployment.url) && deploymentList.contains(deployment) == false)
{
deploymentList.add(deployment);
log.debug("Watching new file: " + deployment.url);
}
}
}
return true;
}
private void create(DeploymentInfo deployment) throws DeploymentException
{
log.debug("create step for deployment " + deployment.url);
try
{
ArrayList sortedSubs = new ArrayList(deployment.subDeployments);
Collections.sort(sortedSubs, infoSorter);
for (Iterator lt = sortedSubs.listIterator(); lt.hasNext();)
{
create((DeploymentInfo) lt.next());
}
deployment.state = DeploymentState.CREATE_SUBDEPLOYMENTS;
if (deployment.deployer != null)
{
deployment.state = DeploymentState.CREATE_DEPLOYER;
deployment.deployer.create(deployment);
deployment.state = DeploymentState.CREATED;
deployment.status="Created";
log.debug("Done with create step of deploying " + deployment.shortName);
}
else
{
log.debug("Still no deployer for package in create step: " + deployment.shortName);
} }
catch (Throwable t)
{
log.error("could not create deployment: " + deployment.url, t);
deployment.status = "Deployment FAILED reason: " + t.getMessage();
deployment.state = DeploymentState.FAILED;
DeploymentException.rethrowAsDeploymentException("Could not create deployment: " + deployment.url, t);
}
}
private void start(DeploymentInfo deployment) throws DeploymentException
{
deployment.status = "Starting";
log.debug("Begin deployment start " + deployment.url);
try
{
ArrayList sortedSubs = new ArrayList(deployment.subDeployments);
Collections.sort(sortedSubs, infoSorter);
for (Iterator lt = sortedSubs.listIterator(); lt.hasNext();)
{
start((DeploymentInfo) lt.next());
}
deployment.state = DeploymentState.START_SUBDEPLOYMENTS;
if (deployment.deployer != null)
{
deployment.state = DeploymentState.START_DEPLOYER;
deployment.deployer.start(deployment);
Object[] args = {deployment, DeploymentState.STARTED};
String[] sig = {"org.jboss.deployment.DeploymentInfo",
"org.jboss.deployment.DeploymentState"};
server.invoke(serviceController, "validateDeploymentState",args, sig);
deployment.status="Deployed";
log.debug("End deployment start on package: "+ deployment.shortName);
}
else
{
log.debug("Still no deployer for package in start step: " + deployment.shortName);
} }
catch (Throwable t)
{
log.error("could not start deployment: " + deployment.url, t);
deployment.state = DeploymentState.FAILED;
deployment.status = "Deployment FAILED reason: " + t.getMessage();
DeploymentException.rethrowAsDeploymentException("Could not create deployment: " + deployment.url, t);
}
}
private void findDeployer(DeploymentInfo sdi)
{
if( sdi.deployer != null )
{
log.debug("using existing deployer "+sdi.deployer);
return;
}
synchronized(deployers)
{
for (Iterator iterator = deployers.iterator(); iterator.hasNext(); )
{
SubDeployer deployer = (SubDeployer) iterator.next();
if (deployer.accepts(sdi))
{
sdi.deployer = deployer;
log.debug("using deployer "+deployer);
return;
}
}
}
log.debug("No deployer found for url: " + sdi.url);
}
private void parseManifestLibraries(DeploymentInfo sdi)
{
String classPath = null;
Manifest mf = sdi.getManifest();
if( mf != null )
{
Attributes mainAttributes = mf.getMainAttributes();
classPath = mainAttributes.getValue(Attributes.Name.CLASS_PATH);
}
if (classPath != null)
{
StringTokenizer st = new StringTokenizer(classPath);
log.debug("resolveLibraries: "+classPath);
while (st.hasMoreTokens())
{
URL lib = null;
String tk = st.nextToken();
log.debug("new manifest entry for sdi at "+sdi.shortName+" entry is "+tk);
try
{
if (sdi.isDirectory)
{
File parentDir = new File(sdi.url.getPath()).getParentFile();
lib = new File(parentDir, tk).toURL();
}
else
{
lib = new URL(sdi.url, tk);
}
if ( deploymentMap.containsKey(lib) == false )
{
DeploymentInfo mfRef = new DeploymentInfo(lib, null, getServer());
makeLocalCopy(mfRef);
URL[] localURL = {mfRef.localUrl};
mfRef.localCl = new java.net.URLClassLoader(localURL);
findDeployer(mfRef);
SubDeployer deployer = mfRef.deployer;
if(deployer != null && (deployer instanceof JARDeployer) == false)
{
log.warn("Found non-jar deployer for " + tk + ": " + deployer);
}
sdi.addLibraryJar(lib);
}
}
catch (Exception ignore)
{
log.debug("The manifest entry in "+sdi.url+" references URL "+lib+
" which could not be opened, entry ignored", ignore);
}
}
}
}
private void makeLocalCopy(DeploymentInfo sdi)
{
try
{
if (sdi.url.getProtocol().equals("file") && (!copyFiles || sdi.isDirectory))
{
sdi.localUrl = sdi.url;
return;
}
else if (inLocalCopyDir(sdi.url))
{
sdi.localUrl = sdi.url;
return;
}
else
{
String shortName = sdi.shortName;
File localFile = File.createTempFile("tmp", shortName, tempDir);
sdi.localUrl = localFile.toURL();
copy(sdi.url, localFile);
}
}
catch (Exception e)
{
log.error("Could not make local copy for " + sdi.url, e);
}
}
private boolean inLocalCopyDir(URL url)
{
int i = 0;
String urlTest = url.toString();
if( urlTest.startsWith("jar:") )
i = 4;
return urlTest.startsWith(tempDirString, i);
}
protected void copy(URL src, File dest) throws IOException
{
log.debug("Copying " + src + " -> " + dest);
File dir = dest.getParentFile();
if (!dir.exists())
{
boolean created = dir.mkdirs();
if( created == false )
throw new IOException("mkdirs failed for: "+dir.getAbsolutePath());
}
if( dest.exists() == true )
{
boolean deleted = Files.delete(dest);
if( deleted == false )
throw new IOException("delete of previous content failed for: "+dest.getAbsolutePath());
}
if (src.getProtocol().equals("file"))
{
File srcFile = new File(src.getFile());
if (srcFile.isDirectory())
{
log.debug("Making zip copy of: " + srcFile);
OutputStream out = new BufferedOutputStream(new FileOutputStream(dest));
JarUtils.jar(out, srcFile.listFiles());
out.close();
return;
}
}
InputStream in = new BufferedInputStream(src.openStream());
OutputStream out = new BufferedOutputStream(new FileOutputStream(dest));
Streams.copy(in, out);
out.flush();
out.close();
in.close();
}
public boolean isDeployed(String url)
throws MalformedURLException
{
return isDeployed(new URL(url));
}
public boolean isDeployed(URL url)
{
DeploymentInfo di = getDeployment(url);
if (di == null)
{
return false;
} return di.state == DeploymentState.STARTED;
}
public DeploymentInfo getDeployment(URL url)
{
synchronized (deploymentList)
{
return (DeploymentInfo) deploymentMap.get(url);
}
}
public URL getWatchUrl(URL url)
{
DeploymentInfo info = getDeployment(url);
return info == null ? null : info.watch;
}
public void checkIncompleteDeployments() throws DeploymentException
{
try
{
Collection waitingForClasses = new HashSet();
Collection waitingForDepends = (Collection)server.invoke(serviceController,
"listIncompletelyDeployed",
new Object[] {},
new String[] {});
Collection allServices = (Collection) server.invoke(serviceController,
"listDeployed",
new Object[] {},
new String[] {});
Collection rootCause = new HashSet(waitingForDepends);
Collection missing = new HashSet();
for (Iterator i = rootCause.iterator(); i.hasNext();)
{
ServiceContext ctx = (ServiceContext) i.next();
for (Iterator j = ctx.iDependOn.iterator(); j.hasNext(); )
{
ServiceContext dependee = (ServiceContext) j.next();
if (dependee.state != ServiceContext.RUNNING)
{
if (allServices.contains(dependee) == false)
missing.add(dependee);
i.remove();
break;
}
}
}
rootCause.addAll(missing);
IncompleteDeploymentException ide = new IncompleteDeploymentException(
waitingForClasses,
waitingForDepends,
rootCause,
listIncompletelyDeployed(),
listWaitingForDeployer());
if (!ide.isEmpty())
{
throw ide;
} }
catch (JMException jme)
{
throw new DeploymentException(JMXExceptionDecoder.decode(jme));
} }
}