use of com.blazebit.persistence.view.EntityViewRoots in project blaze-persistence by Blazebit.
the class AnnotationMappingReader method readViewMapping.
@Override
public ViewMapping readViewMapping(Class<?> entityViewClass) {
ViewMapping existingMapping = context.getViewMapping(entityViewClass);
if (existingMapping != null) {
return existingMapping;
}
EntityView entityView = AnnotationUtils.findAnnotation(entityViewClass, EntityView.class);
if (entityView == null) {
return null;
}
Class<?> entityClass = entityView.value();
ViewMapping viewMapping = new ViewMappingImpl(entityViewClass, entityClass, context);
context.addViewMapping(entityViewClass, viewMapping);
boolean isAbstract = entityViewClass.isInterface() || Modifier.isAbstract(entityViewClass.getModifiers());
BatchFetch batchFetch = AnnotationUtils.findAnnotation(entityViewClass, BatchFetch.class);
if (batchFetch != null) {
viewMapping.setDefaultBatchSize(batchFetch.size());
}
Set<Class<? extends CTEProvider>> cteProviders = new LinkedHashSet<>();
Map<String, Class<? extends ViewFilterProvider>> viewFilterProviders = new HashMap<>();
Map<String, EntityViewRootMapping> viewRootMappings = new LinkedHashMap<>();
for (Annotation a : AnnotationUtils.getAllAnnotations(entityViewClass)) {
if (a instanceof With) {
cteProviders.addAll(Arrays.asList(((With) a).value()));
} else if (a instanceof ViewFilter) {
ViewFilter viewFilter = (ViewFilter) a;
addFilterMapping(viewFilter.name(), viewFilter.value(), viewFilterProviders, entityViewClass, context);
} else if (a instanceof ViewFilters) {
ViewFilters viewFilters = (ViewFilters) a;
for (ViewFilter viewFilter : viewFilters.value()) {
addFilterMapping(viewFilter.name(), viewFilter.value(), viewFilterProviders, entityViewClass, context);
}
} else if (a instanceof EntityViewRoot) {
EntityViewRoot entityViewRoot = (EntityViewRoot) a;
addEntityViewRootMapping(entityViewRoot, viewRootMappings, entityViewClass, context);
} else if (a instanceof EntityViewRoots) {
EntityViewRoots entityViewRoots = (EntityViewRoots) a;
for (EntityViewRoot entityViewRoot : entityViewRoots.value()) {
addEntityViewRootMapping(entityViewRoot, viewRootMappings, entityViewClass, context);
}
}
}
viewMapping.setCteProviders(cteProviders);
viewMapping.setViewFilterProviders(viewFilterProviders);
viewMapping.setEntityViewRoots(viewRootMappings.isEmpty() ? Collections.<EntityViewRootMapping>emptySet() : new LinkedHashSet<>(viewRootMappings.values()));
UpdatableEntityView updatableEntityView = AnnotationUtils.findAnnotation(entityViewClass, UpdatableEntityView.class);
if (updatableEntityView != null) {
if (isAbstract) {
viewMapping.setUpdatable(true);
viewMapping.setFlushMode(updatableEntityView.mode());
viewMapping.setFlushStrategy(updatableEntityView.strategy());
viewMapping.setLockMode(updatableEntityView.lockMode());
} else {
context.addError("Only abstract class entity views can be updatable! Remove the @UpdatableEntityView annotation from the entity view class '" + entityViewClass.getName() + "' or make it abstract.");
}
}
CreatableEntityView creatableEntityView = AnnotationUtils.findAnnotation(entityViewClass, CreatableEntityView.class);
if (creatableEntityView != null) {
if (isAbstract) {
viewMapping.setCreatable(true);
viewMapping.setValidatePersistability(creatableEntityView.validatePersistability());
viewMapping.getExcludedAttributes().addAll(Arrays.asList(creatableEntityView.excludedEntityAttributes()));
} else {
context.addError("Only abstract class entity views can be creatable! Remove the @CreatableEntityView annotation from the entity view class '" + entityViewClass.getName() + "' or make it abstract.");
}
}
if (updatableEntityView != null || creatableEntityView != null) {
LockOwner lockOwner = AnnotationUtils.findAnnotation(entityViewClass, LockOwner.class);
if (lockOwner != null) {
viewMapping.setLockOwner(lockOwner.value());
}
} else {
viewMapping.setLockMode(LockMode.NONE);
}
// Inheritance
// Note that the usage of Class.getAnnotation is on purpose
// For the full discussion see: https://github.com/Blazebit/blaze-persistence/issues/475
EntityViewInheritance inheritanceAnnotation = entityViewClass.getAnnotation(EntityViewInheritance.class);
EntityViewInheritanceMapping inheritanceMappingAnnotation = entityViewClass.getAnnotation(EntityViewInheritanceMapping.class);
String inheritanceMapping;
if (inheritanceMappingAnnotation != null) {
inheritanceMapping = inheritanceMappingAnnotation.value();
} else {
inheritanceMapping = null;
}
if (isAbstract) {
viewMapping.setInheritanceMapping(inheritanceMapping);
if (inheritanceAnnotation == null) {
viewMapping.setInheritanceSubtypesResolved(true);
} else if (inheritanceAnnotation.value().length > 0) {
viewMapping.getInheritanceSubtypeClasses().addAll(Arrays.asList(inheritanceAnnotation.value()));
viewMapping.setInheritanceSubtypesResolved(true);
}
} else if (inheritanceMapping != null || inheritanceAnnotation != null) {
context.addError("Only abstract class entity views can use inheritance mappings! Remove the inheritance annotations from the entity view class '" + entityViewClass.getName() + "' or make it abstract.");
} else {
viewMapping.setInheritanceSubtypesResolved(true);
}
// Attributes
if (isAbstract) {
readAbstractClassMappings(entityViewClass, viewMapping);
} else {
readClassMappings(entityViewClass, viewMapping);
}
return viewMapping;
}
Aggregations