use of com.blazebit.persistence.view.CTEProvider in project blaze-persistence by Blazebit.
the class WithCTEMappingTest method testCteBindingWithSubview.
@Test
public void testCteBindingWithSubview() {
EntityViewManager evm = build(DocumentWithCTE.class, PersonWithPartnerDocument.class, PersonWithPartnerDocumentFullAged.class, PersonWithPartnerDocumentUnderAged.class);
CriteriaBuilder<Person> cb = cbf.create(em, Person.class).orderByAsc("id");
EntityViewSetting<PersonWithPartnerDocument, CriteriaBuilder<PersonWithPartnerDocument>> setting;
setting = EntityViewSetting.create(PersonWithPartnerDocument.class);
List<PersonWithPartnerDocument> list = evm.applySetting(setting, cb).getResultList();
assertEquals(3, list.size());
assertEquals("doc1", list.get(0).getPartnerDocument().getName());
assertEquals(Long.valueOf(1), list.get(0).getPartnerDocument().getOwnedDocumentCount());
boolean containsProvider = false;
ManagedViewType<?> viewType = evm.getMetamodel().managedView(PersonWithPartnerDocumentFullAged.class);
for (CTEProvider provider : viewType.getCteProviders()) {
if (provider instanceof FullAgedCTEProvider) {
containsProvider = true;
}
}
assertTrue(containsProvider);
}
use of com.blazebit.persistence.view.CTEProvider 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;
}
use of com.blazebit.persistence.view.CTEProvider in project blaze-persistence by Blazebit.
the class ManagedViewTypeImpl method renderSecondaryMappings.
@Override
public void renderSecondaryMappings(String viewPath, BaseQueryBuilder<?, ?> baseQueryBuilder, Map<String, Object> optionalParameters, boolean renderFetches) {
if (baseQueryBuilder instanceof CTEBuilder) {
CTEBuilder<?> cteBuilder = (CTEBuilder<?>) baseQueryBuilder;
for (CTEProvider cteProvider : getCteProviders()) {
cteProvider.applyCtes(cteBuilder, optionalParameters);
}
}
for (ViewRoot viewRoot : viewRoots) {
String entityViewRootName = viewRoot.getName();
CorrelationProvider correlationProvider = viewRoot.getCorrelationProviderFactory().create(baseQueryBuilder, optionalParameters);
ExpressionFactory expressionFactory = baseQueryBuilder.getService(ExpressionFactory.class);
Limiter limiter = createLimiter(expressionFactory, viewPath, viewRoot.getLimitExpression(), viewRoot.getOffsetExpression(), viewRoot.getOrderByItems());
String correlationAlias;
if (limiter == null) {
correlationAlias = entityViewRootName;
} else {
correlationAlias = "_sub_" + entityViewRootName;
}
JoinCorrelationBuilder correlationBuilder = new JoinCorrelationBuilder(baseQueryBuilder, optionalParameters, baseQueryBuilder, viewPath, correlationAlias, entityViewRootName, null, viewRoot.getJoinType(), limiter);
correlationProvider.applyCorrelation(correlationBuilder, viewPath);
correlationBuilder.finish();
if (renderFetches && baseQueryBuilder instanceof FetchBuilder<?>) {
((FetchBuilder<?>) baseQueryBuilder).fetch(viewRoot.getFetches());
}
}
}
Aggregations