/*
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You under the Apache License, Version 2.0
* (the "License"); you may not use this file except in compliance with
* the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.apache.openejb.assembler;
import org.apache.openejb.ClassLoaderUtil;
import org.apache.openejb.NoSuchApplicationException;
import org.apache.openejb.OpenEJBException;
import org.apache.openejb.OpenEJBRuntimeException;
import org.apache.openejb.UndeployException;
import org.apache.openejb.assembler.classic.AppInfo;
import org.apache.openejb.assembler.classic.Assembler;
import org.apache.openejb.config.AppModule;
import org.apache.openejb.config.ConfigurationFactory;
import org.apache.openejb.config.DeploymentLoader;
import org.apache.openejb.config.DeploymentModule;
import org.apache.openejb.config.sys.AdditionalDeployments;
import org.apache.openejb.config.sys.Deployments;
import org.apache.openejb.config.sys.JaxbOpenejb;
import org.apache.openejb.loader.IO;
import org.apache.openejb.loader.SystemInstance;
import org.apache.openejb.util.LogCategory;
import org.apache.openejb.util.Logger;
import javax.ejb.Remote;
import javax.ejb.Stateless;
import javax.ejb.TransactionManagement;
import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.math.BigInteger;
import java.security.SecureRandom;
import java.util.Collection;
import java.util.Iterator;
import java.util.Map;
import java.util.Properties;
import java.util.TreeMap;
import static javax.ejb.TransactionManagementType.BEAN;
import static org.apache.openejb.config.ConfigurationFactory.ADDITIONAL_DEPLOYMENTS;
import static org.apache.openejb.loader.ProvisioningUtil.realLocation;
@Stateless(name = "openejb/Deployer")
@Remote(Deployer.class)
@TransactionManagement(BEAN)
public class DeployerEjb implements Deployer {
public static final String OPENEJB_DEPLOYER_FORCED_APP_ID_PROP = "openejb.deployer.forced.appId";
public static final Logger LOGGER = Logger.getInstance(LogCategory.OPENEJB, DeployerEjb.class);
private final static File uniqueFile;
static {
String uniqueName = "OpenEJB-" + new BigInteger(128, new SecureRandom()).toString(Character.MAX_RADIX);
String tempDir = System.getProperty("java.io.tmpdir");
File unique = null;
try {
unique = new File(tempDir, uniqueName).getCanonicalFile();
unique.createNewFile();
} catch (IOException e) {
// same trying in work directory
unique = new File(SystemInstance.get().getHome().getDirectory(), "work");
if (unique.exists()) {
try {
unique = new File(unique, uniqueName).getCanonicalFile();
unique.createNewFile();
} catch (IOException e1) {
throw new OpenEJBRuntimeException(e);
}
} else {
throw new OpenEJBRuntimeException("can't create unique file, please set java.io.tmpdir to a writable folder or create work folder", e);
}
}
uniqueFile = unique;
uniqueFile.deleteOnExit();
}
private final DeploymentLoader deploymentLoader;
private final ConfigurationFactory configurationFactory;
private final Assembler assembler;
public DeployerEjb() {
deploymentLoader = new DeploymentLoader();
configurationFactory = new ConfigurationFactory();
assembler = (Assembler) SystemInstance.get().getComponent(org.apache.openejb.spi.Assembler.class);
}
public String getUniqueFile() {
return uniqueFile.getAbsolutePath();
}
public Collection<AppInfo> getDeployedApps() {
return assembler.getDeployedApplications();
}
public AppInfo deploy(String location) throws OpenEJBException {
return deploy(location, null);
}
public AppInfo deploy(Properties properties) throws OpenEJBException {
return deploy(null, properties);
}
public AppInfo deploy(String inLocation, Properties properties) throws OpenEJBException {
String rawLocation = inLocation;
if (rawLocation == null && properties == null) {
throw new NullPointerException("location and properties are null");
}
if (rawLocation == null) {
rawLocation = properties.getProperty(FILENAME);
}
if (properties == null) {
properties = new Properties();
}
AppModule appModule = null;
final File file = new File(realLocation(rawLocation));
AppInfo appInfo;
try {
appModule = deploymentLoader.load(file);
// Add any alternate deployment descriptors to the modules
Map<String, DeploymentModule> modules = new TreeMap<String, DeploymentModule>();
for (DeploymentModule module : appModule.getEjbModules()) {
modules.put(module.getModuleId(), module);
}
for (DeploymentModule module : appModule.getClientModules()) {
modules.put(module.getModuleId(), module);
}
for (DeploymentModule module : appModule.getWebModules()) {
modules.put(module.getModuleId(), module);
}
for (DeploymentModule module : appModule.getConnectorModules()) {
modules.put(module.getModuleId(), module);
}
for (Map.Entry<Object, Object> entry : properties.entrySet()) {
String name = (String) entry.getKey();
if (name.startsWith(ALT_DD + "/")) {
name = name.substring(ALT_DD.length() + 1);
DeploymentModule module;
int slash = name.indexOf('/');
if (slash > 0) {
String moduleId = name.substring(0, slash);
name = name.substring(slash + 1);
module = modules.get(moduleId);
} else {
module = appModule;
}
if (module != null) {
String value = (String) entry.getValue();
File dd = new File(value);
if (dd.canRead()) {
module.getAltDDs().put(name, dd.toURI().toURL());
} else {
module.getAltDDs().put(name, value);
}
}
}
}
appInfo = configurationFactory.configureApplication(appModule);
if (properties != null && properties.containsKey(OPENEJB_DEPLOYER_FORCED_APP_ID_PROP)) {
appInfo.appId = properties.getProperty(OPENEJB_DEPLOYER_FORCED_APP_ID_PROP);
}
assembler.createApplication(appInfo);
saveDeployment(file, true);
return appInfo;
} catch (Throwable e) {
// destroy the class loader for the failed application
if (appModule != null) {
ClassLoaderUtil.destroyClassLoader(appModule.getJarLocation());
}
e.printStackTrace();
if (e instanceof javax.validation.ValidationException) {
throw (javax.validation.ValidationException) e;
}
if (e instanceof OpenEJBException) {
if (e.getCause() instanceof javax.validation.ValidationException) {
throw (javax.validation.ValidationException) e.getCause();
}
throw (OpenEJBException) e;
}
throw new OpenEJBException(e);
}
}
private synchronized void saveDeployment(final File file, boolean add) {
final Deployments deps = new Deployments();
if (file.isDirectory()) {
deps.setDir(file.getAbsolutePath());
} else {
deps.setJar(file.getAbsolutePath());
}
File config;
try {
config = SystemInstance.get().getBase().getFile(ADDITIONAL_DEPLOYMENTS, false);
} catch (IOException e) {
config = null;
}
if (config == null || !config.getParentFile().exists()) {
LOGGER.error("can't save the added app because the conf folder doesn't exist, it will not be present next time you'll start");
return;
}
// dump it
OutputStream os = null;
try {
final AdditionalDeployments additionalDeployments;
if (config.exists() && config.length() > 0) {
final InputStream fis = IO.read(config);
try {
additionalDeployments = JaxbOpenejb.unmarshal(AdditionalDeployments.class, fis);
} finally {
IO.close(fis);
}
} else {
additionalDeployments = new AdditionalDeployments();
}
if (add) {
additionalDeployments.getDeployments().add(deps);
} else {
Iterator<Deployments> it = additionalDeployments.getDeployments().iterator();
while (it.hasNext()) {
final Deployments current = it.next();
if (deps.getDir() != null && deps.getDir().equals(current.getDir())) {
it.remove();
break;
} else if (deps.getJar() != null && deps.getJar().equals(current.getJar())) {
it.remove();
break;
}
}
}
os = IO.write(config);
JaxbOpenejb.marshal(AdditionalDeployments.class, additionalDeployments, os);
} catch (Exception e) {
LOGGER.error("can't save the added app, will not be present next time you'll start", e);
} finally {
IO.close(os);
}
}
public void undeploy(String moduleId) throws UndeployException, NoSuchApplicationException {
try {
assembler.destroyApplication(moduleId);
} catch (NoSuchApplicationException nsae) {
try {
assembler.destroyApplication(realLocation(moduleId));
} catch (Exception e) {
try {
assembler.destroyApplication(new File(moduleId).getAbsolutePath());
} catch (Exception e2) {
try {
assembler.destroyApplication(new File(realLocation(moduleId)).getAbsolutePath());
} catch (Exception e3) {
throw nsae;
}
}
}
}
saveDeployment(new File(moduleId), false);
}
}