package org.springframework.roo.addon.layers.repository.jpa.addon;
import static org.springframework.roo.model.RooJavaType.ROO_REPOSITORY_JPA_CUSTOM;
import org.apache.commons.lang3.StringUtils;
import org.apache.commons.lang3.Validate;
import org.apache.commons.lang3.tuple.Pair;
import org.apache.felix.scr.annotations.Component;
import org.apache.felix.scr.annotations.Service;
import org.osgi.framework.BundleContext;
import org.osgi.service.component.ComponentContext;
import org.springframework.roo.addon.javabean.addon.JavaBeanMetadata;
import org.springframework.roo.addon.jpa.addon.JpaOperations;
import org.springframework.roo.addon.jpa.addon.entity.JpaEntityMetadata;
import org.springframework.roo.addon.jpa.addon.entity.JpaEntityMetadata.RelationInfo;
import org.springframework.roo.addon.layers.repository.jpa.addon.finder.parser.FinderMethod;
import org.springframework.roo.addon.layers.repository.jpa.annotations.RooJpaRepositoryCustom;
import org.springframework.roo.classpath.PhysicalTypeIdentifier;
import org.springframework.roo.classpath.PhysicalTypeMetadata;
import org.springframework.roo.classpath.customdata.taggers.CustomDataKeyDecorator;
import org.springframework.roo.classpath.customdata.taggers.CustomDataKeyDecoratorTracker;
import org.springframework.roo.classpath.details.ClassOrInterfaceTypeDetails;
import org.springframework.roo.classpath.details.FieldMetadata;
import org.springframework.roo.classpath.details.ItdTypeDetails;
import org.springframework.roo.classpath.details.MemberHoldingTypeDetails;
import org.springframework.roo.classpath.details.annotations.AnnotationAttributeValue;
import org.springframework.roo.classpath.details.annotations.NestedAnnotationAttributeValue;
import org.springframework.roo.classpath.itd.AbstractMemberDiscoveringItdMetadataProvider;
import org.springframework.roo.classpath.itd.ItdTypeDetailsProvidingMetadataItem;
import org.springframework.roo.classpath.layers.LayerTypeMatcher;
import org.springframework.roo.metadata.MetadataDependencyRegistry;
import org.springframework.roo.metadata.MetadataIdentificationUtils;
import org.springframework.roo.metadata.MetadataItem;
import org.springframework.roo.metadata.internal.MetadataDependencyRegistryTracker;
import org.springframework.roo.model.JavaSymbolName;
import org.springframework.roo.model.JavaType;
import org.springframework.roo.model.RooJavaType;
import org.springframework.roo.project.LogicalPath;
import org.springframework.roo.support.logging.HandlerUtils;
import java.util.ArrayList;
import java.util.Iterator;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.logging.Logger;
/**
* Implementation of {@link RepositoryJpaCustomMetadataProvider}.
*
* @author Paula Navarro
* @author Juan Carlos GarcĂa
* @author Sergio Clares
* @since 2.0
*/
@Component
@Service
public class RepositoryJpaCustomMetadataProviderImpl extends
AbstractMemberDiscoveringItdMetadataProvider implements RepositoryJpaCustomMetadataProvider {
protected final static Logger LOGGER = HandlerUtils
.getLogger(RepositoryJpaCustomMetadataProviderImpl.class);
private final Map<JavaType, String> domainTypeToRepositoryMidMap =
new LinkedHashMap<JavaType, String>();
private final Map<String, JavaType> repositoryMidToDomainTypeMap =
new LinkedHashMap<String, JavaType>();
protected MetadataDependencyRegistryTracker registryTracker = null;
protected CustomDataKeyDecoratorTracker keyDecoratorTracker = null;
/**
* This service is being activated so setup it:
* <ul>
* <li>Create and open the {@link MetadataDependencyRegistryTracker}.</li>
* <li>Create and open the {@link CustomDataKeyDecoratorTracker}.</li>
* <li>Registers {@link RooJavaType#ROO_REPOSITORY_JPA_CUSTOM} as additional
* JavaType that will trigger metadata registration.</li>
* <li>Set ensure the governor type details represent a class.</li>
* </ul>
*/
@Override
@SuppressWarnings("unchecked")
protected void activate(final ComponentContext cContext) {
super.activate(cContext);
BundleContext localContext = cContext.getBundleContext();
super.setDependsOnGovernorBeingAClass(false);
this.registryTracker =
new MetadataDependencyRegistryTracker(localContext, this,
PhysicalTypeIdentifier.getMetadataIdentiferType(), getProvidesType());
this.registryTracker.open();
addMetadataTrigger(ROO_REPOSITORY_JPA_CUSTOM);
this.keyDecoratorTracker =
new CustomDataKeyDecoratorTracker(localContext, getClass(), new LayerTypeMatcher(
ROO_REPOSITORY_JPA_CUSTOM, new JavaSymbolName(RooJpaRepositoryCustom.ENTITY_ATTRIBUTE)));
this.keyDecoratorTracker.open();
}
/**
* This service is being deactivated so unregister upstream-downstream
* dependencies, triggers, matchers and listeners.
*
* @param context
*/
protected void deactivate(final ComponentContext context) {
MetadataDependencyRegistry registry = this.registryTracker.getService();
registry.removeNotificationListener(this);
registry.deregisterDependency(PhysicalTypeIdentifier.getMetadataIdentiferType(),
getProvidesType());
this.registryTracker.close();
removeMetadataTrigger(ROO_REPOSITORY_JPA_CUSTOM);
CustomDataKeyDecorator keyDecorator = this.keyDecoratorTracker.getService();
keyDecorator.unregisterMatchers(getClass());
this.keyDecoratorTracker.close();
}
@Override
protected String createLocalIdentifier(final JavaType javaType, final LogicalPath path) {
return RepositoryJpaCustomMetadata.createIdentifier(javaType, path);
}
@Override
protected String getGovernorPhysicalTypeIdentifier(final String metadataIdentificationString) {
final JavaType javaType = RepositoryJpaCustomMetadata.getJavaType(metadataIdentificationString);
final LogicalPath path = RepositoryJpaCustomMetadata.getPath(metadataIdentificationString);
return PhysicalTypeIdentifier.createIdentifier(javaType, path);
}
public String getItdUniquenessFilenameSuffix() {
return "Jpa_Repository_Custom";
}
@Override
protected String getLocalMidToRequest(final ItdTypeDetails itdTypeDetails) {
// Determine the governor for this ITD, and whether any metadata is even
// hoping to hear about changes to that JavaType and its ITDs
final JavaType governor = itdTypeDetails.getName();
final String localMid = domainTypeToRepositoryMidMap.get(governor);
if (localMid != null) {
return localMid;
}
final MemberHoldingTypeDetails memberHoldingTypeDetails =
getTypeLocationService().getTypeDetails(governor);
if (memberHoldingTypeDetails != null) {
for (final JavaType type : memberHoldingTypeDetails.getLayerEntities()) {
final String localMidType = domainTypeToRepositoryMidMap.get(type);
if (localMidType != null) {
return localMidType;
}
}
}
return null;
}
@Override
protected ItdTypeDetailsProvidingMetadataItem getMetadata(
final String metadataIdentificationString, final JavaType aspectName,
final PhysicalTypeMetadata governorPhysicalTypeMetadata, final String itdFilename) {
final RepositoryJpaCustomAnnotationValues annotationValues =
new RepositoryJpaCustomAnnotationValues(governorPhysicalTypeMetadata);
// Getting repository custom
JavaType entity = annotationValues.getEntity();
Validate
.notNull(
entity,
"ERROR: Repository custom interface should be contain an entity on @RooJpaRepositoryCustom annotation");
final ClassOrInterfaceTypeDetails repositoryClass =
getRepositoryJpaLocator().getRepository(entity);
final String repositoryMedatadataId = RepositoryJpaMetadata.createIdentifier(repositoryClass);
registerDependency(repositoryMedatadataId, metadataIdentificationString);
RepositoryJpaMetadata repositoryMetadata = getMetadataService().get(repositoryMedatadataId);
// This metadata is not available yet.
if (repositoryMetadata == null) {
return null;
}
// Add dependency between modules
ClassOrInterfaceTypeDetails cid = governorPhysicalTypeMetadata.getMemberHoldingTypeDetails();
String module = cid.getName().getModule();
getTypeLocationService().addModuleDependency(module, entity);
getTypeLocationService().addModuleDependency(module, repositoryMetadata.getDefaultReturnType());
// Get field which entity is field part
List<Pair<FieldMetadata, RelationInfo>> relationsAsChild =
getJpaOperations().getFieldChildPartOfRelation(entity);
for (Pair<FieldMetadata, RelationInfo> fieldInfo : relationsAsChild) {
// Add dependency between modules
getTypeLocationService().addModuleDependency(module, fieldInfo.getLeft().getFieldType());
}
// Register dependency between JavaBeanMetadata and this one
final ClassOrInterfaceTypeDetails entityDetails =
getTypeLocationService().getTypeDetails(entity);
final String javaBeanMetadataKey = JavaBeanMetadata.createIdentifier(entityDetails);
registerDependency(javaBeanMetadataKey, metadataIdentificationString);
String entityMetadataKey = JpaEntityMetadata.createIdentifier(entityDetails);
JpaEntityMetadata entityMetadata = getMetadataService().get(entityMetadataKey);
// Entity metadata is not available
if (entityMetadata == null) {
return null;
}
return new RepositoryJpaCustomMetadata(metadataIdentificationString, aspectName,
governorPhysicalTypeMetadata, annotationValues, entityMetadata.getCurrentIndentifierField()
.getFieldType(), entity, repositoryMetadata, relationsAsChild);
}
private JpaOperations getJpaOperations() {
return getServiceManager().getServiceInstance(this, JpaOperations.class);
}
private RepositoryJpaLocator getRepositoryJpaLocator() {
return getServiceManager().getServiceInstance(this, RepositoryJpaLocator.class);
}
protected void registerDependency(final String upstreamDependency,
final String downStreamDependency) {
if (getMetadataDependencyRegistry() != null
&& StringUtils.isNotBlank(upstreamDependency)
&& StringUtils.isNotBlank(downStreamDependency)
&& !upstreamDependency.equals(downStreamDependency)
&& !MetadataIdentificationUtils.getMetadataClass(downStreamDependency).equals(
MetadataIdentificationUtils.getMetadataClass(upstreamDependency))) {
getMetadataDependencyRegistry().registerDependency(upstreamDependency, downStreamDependency);
}
}
public String getProvidesType() {
return RepositoryJpaCustomMetadata.getMetadataIdentiferType();
}
}