/*
* JBoss, Home of Professional Open Source.
* Copyright 2008, Red Hat Middleware LLC, 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.jpa.deployers;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Properties;
import javax.naming.InitialContext;
import javax.naming.NamingException;
import org.jboss.beans.metadata.api.annotations.Inject;
import org.jboss.beans.metadata.api.annotations.MapValue;
import org.jboss.beans.metadata.plugins.AbstractBeanMetaData;
import org.jboss.beans.metadata.spi.BeanMetaData;
import org.jboss.beans.metadata.spi.builder.BeanMetaDataBuilder;
import org.jboss.deployers.spi.DeploymentException;
import org.jboss.deployers.spi.deployer.helpers.AbstractSimpleRealDeployer;
import org.jboss.deployers.structure.spi.DeploymentUnit;
import org.jboss.deployers.vfs.spi.structure.VFSDeploymentUnit;
import org.jboss.jpa.deployment.PersistenceUnitDeployment;
import org.jboss.jpa.resolvers.DataSourceDependencyResolver;
import org.jboss.jpa.resolvers.PersistenceUnitDependencyResolver;
import org.jboss.metadata.jpa.spec.PersistenceUnitMetaData;
/**
* @author <a href="mailto:cdewolf@redhat.com">Carlo de Wolf</a>
* @author <a href="mailto:ales.justin@jboss.com">Ales Justin</a>
* @version $Revision: $
*/
public class PersistenceUnitDeployer extends AbstractSimpleRealDeployer<PersistenceUnitMetaData>
{
private Properties defaultPersistenceProperties;
private DataSourceDependencyResolver dataSourceDependencyResolver;
private PersistenceUnitDependencyResolver persistenceUnitDependencyResolver;
public PersistenceUnitDeployer()
{
super(PersistenceUnitMetaData.class);
// We want to process the components created by PersistenceDeployer, this seems to be the only way to get there.
setComponentsOnly(true);
addOutput(BeanMetaData.class);
}
private void addDependencies(BeanMetaDataBuilder builder, PersistenceUnitMetaData metaData)
{
// Initialize properties to default
Properties props = new Properties();
props.putAll(defaultPersistenceProperties);
// Add properties from metadata
Map<String, String> metadataProps = metaData.getProperties() != null
? metaData.getProperties()
: new HashMap<String, String>();
props.putAll(metadataProps);
if (!props.containsKey("jboss.no.implicit.datasource.dependency"))
{
if (metaData.getJtaDataSource() != null)
{
String ds = metaData.getJtaDataSource();
builder.addDependency(dataSourceDependencyResolver.resolveDataSourceSupplier(ds));
}
if (metaData.getNonJtaDataSource() != null)
{
String ds = metaData.getNonJtaDataSource();
builder.addDependency(dataSourceDependencyResolver.resolveDataSourceSupplier(ds));
}
}
for (Object prop : props.keySet())
{
String property = (String) prop;
if (property.startsWith("jboss.depends"))
{
builder.addDependency(props.get(property));
}
}
}
@Override
public void deploy(DeploymentUnit unit, PersistenceUnitMetaData metaData) throws DeploymentException
{
log.debug("deploy " + metaData);
try
{
String name = persistenceUnitDependencyResolver.createBeanName(unit, metaData.getName());
InitialContext initialContext = new InitialContext();
List<String> explicitEntityClasses = new ArrayList<String>();
VFSDeploymentUnit deploymentUnit = (VFSDeploymentUnit) unit.getParent();
PersistenceUnitDeployment pu = new PersistenceUnitDeployment(initialContext, null, explicitEntityClasses, metaData, name, deploymentUnit, defaultPersistenceProperties);
AbstractBeanMetaData beanMetaData = new AbstractBeanMetaData(name, PersistenceUnitDeployment.class.getName());
BeanMetaDataBuilder builder = BeanMetaDataBuilder.createBuilder(beanMetaData);
builder.setConstructorValue(pu);
addDependencies(builder, metaData);
DeploymentUnit parent = unit.getParent();
if (parent == null)
throw new IllegalArgumentException("Parent unit should not be null as this unit should be component: " + unit);
parent.addAttachment(BeanMetaData.class.getName() + "." + name, builder.getBeanMetaData(), BeanMetaData.class);
}
catch(NamingException e)
{
throw new DeploymentException(e);
}
}
@Inject
public void setDataSourceDependencyResolver(DataSourceDependencyResolver resolver)
{
this.dataSourceDependencyResolver = resolver;
}
@MapValue(keyClass=String.class, value={}, valueClass=String.class)
public void setDefaultPersistenceProperties(Properties p)
{
this.defaultPersistenceProperties = p;
}
@Inject
public void setPersistenceUnitDependencyResolver(PersistenceUnitDependencyResolver resolver)
{
this.persistenceUnitDependencyResolver = resolver;
}
}