Examples with FetchGroupTracker org.eclipse.persistence.queries.FetchGroupTracker used on opensource projects

Example 1 with FetchGroupTracker

use of org.eclipse.persistence.queries.FetchGroupTracker in project cuba by cuba-platform.

the class PersistenceTools method getReferenceId.

/**
 * Returns an ID of directly referenced entity without loading it from DB.
 *
 * If the view does not contain the reference and {@link View#loadPartialEntities()} is true,
 * the returned {@link RefId} will have {@link RefId#isLoaded()} = false.
 *
 * <p>Usage example:
 * <pre>
 *   PersistenceTools.RefId refId = persistenceTools.getReferenceId(doc, "currency");
 *   if (refId.isLoaded()) {
 *       String currencyCode = (String) refId.getValue();
 *   }
 * </pre>
 *
 * @param entity   entity instance in managed state
 * @param property name of reference property
 * @return {@link RefId} instance which contains the referenced entity ID
 * @throws IllegalArgumentException if the specified property is not a reference
 * @throws IllegalStateException if the entity is not in Managed state
 * @throws RuntimeException if anything goes wrong when retrieving the ID
 */
public RefId getReferenceId(BaseGenericIdEntity entity, String property) {
    MetaClass metaClass = metadata.getClassNN(entity.getClass());
    MetaProperty metaProperty = metaClass.getPropertyNN(property);
    if (!metaProperty.getRange().isClass() || metaProperty.getRange().getCardinality().isMany())
        throw new IllegalArgumentException("Property is not a reference");
    if (!PersistenceHelper.isManaged(entity))
        throw new IllegalStateException("Entity must be in managed state");
    String[] inaccessibleAttributes = BaseEntityInternalAccess.getInaccessibleAttributes(entity);
    if (inaccessibleAttributes != null) {
        for (String inaccessibleAttr : inaccessibleAttributes) {
            if (inaccessibleAttr.equals(property))
                return RefId.createNotLoaded(property);
        }
    }
    if (entity instanceof FetchGroupTracker) {
        FetchGroup fetchGroup = ((FetchGroupTracker) entity)._persistence_getFetchGroup();
        if (fetchGroup != null) {
            if (!fetchGroup.getAttributeNames().contains(property))
                return RefId.createNotLoaded(property);
            else {
                Entity refEntity = (Entity) entity.getValue(property);
                return RefId.create(property, refEntity == null ? null : refEntity.getId());
            }
        }
    }
    try {
        Class<?> declaringClass = metaProperty.getDeclaringClass();
        if (declaringClass == null) {
            throw new RuntimeException("Property does not belong to persistent class");
        }
        Method vhMethod = declaringClass.getDeclaredMethod(String.format("_persistence_get_%s_vh", property));
        vhMethod.setAccessible(true);
        ValueHolderInterface vh = (ValueHolderInterface) vhMethod.invoke(entity);
        if (vh instanceof DatabaseValueHolder) {
            AbstractRecord row = ((DatabaseValueHolder) vh).getRow();
            if (row != null) {
                Session session = persistence.getEntityManager().getDelegate().unwrap(Session.class);
                ClassDescriptor descriptor = session.getDescriptor(entity);
                DatabaseMapping mapping = descriptor.getMappingForAttributeName(property);
                Vector<DatabaseField> fields = mapping.getFields();
                if (fields.size() != 1) {
                    throw new IllegalStateException("Invalid number of columns in mapping: " + fields);
                }
                Object value = row.get(fields.get(0));
                if (value != null) {
                    ClassDescriptor refDescriptor = mapping.getReferenceDescriptor();
                    DatabaseMapping refMapping = refDescriptor.getMappingForAttributeName(metadata.getTools().getPrimaryKeyName(metaClass));
                    if (refMapping instanceof AbstractColumnMapping) {
                        Converter converter = ((AbstractColumnMapping) refMapping).getConverter();
                        if (converter != null) {
                            return RefId.create(property, converter.convertDataValueToObjectValue(value, session));
                        }
                    }
                }
                return RefId.create(property, value);
            } else {
                return RefId.create(property, null);
            }
        }
        return RefId.createNotLoaded(property);
    } catch (Exception e) {
        throw new RuntimeException(String.format("Error retrieving reference ID from %s.%s", entity.getClass().getSimpleName(), property), e);
    }
}

Example 2 with FetchGroupTracker

use of org.eclipse.persistence.queries.FetchGroupTracker in project cuba by cuba-platform.

the class AttributeSecuritySupport method applySecurityToFetchGroup.

protected void applySecurityToFetchGroup(Entity entity) {
    if (entity == null) {
        return;
    }
    MetaClass metaClass = metadata.getClassNN(entity.getClass());
    FetchGroupTracker fetchGroupTracker = (FetchGroupTracker) entity;
    FetchGroup fetchGroup = fetchGroupTracker._persistence_getFetchGroup();
    SecurityState securityState = getSecurityState(entity);
    if (fetchGroup != null) {
        List<String> attributesToRemove = new ArrayList<>();
        for (String attrName : fetchGroup.getAttributeNames()) {
            String[] parts = attrName.split("\\.");
            if (parts.length > 0 && BaseEntityInternalAccess.isHiddenOrReadOnly(securityState, parts[0])) {
                attributesToRemove.add(attrName);
            } else {
                MetaClass currentMetaClass = metaClass;
                for (String part : parts) {
                    if (!security.isEntityAttrUpdatePermitted(currentMetaClass, part)) {
                        attributesToRemove.add(attrName);
                        break;
                    }
                    MetaProperty metaProperty = currentMetaClass.getPropertyNN(part);
                    if (metaProperty.getRange().isClass()) {
                        currentMetaClass = metaProperty.getRange().asClass();
                    }
                }
            }
        }
        if (!attributesToRemove.isEmpty()) {
            List<String> attributeNames = new ArrayList<>(fetchGroup.getAttributeNames());
            attributeNames.removeAll(attributesToRemove);
            fetchGroupTracker._persistence_setFetchGroup(new CubaEntityFetchGroup(attributeNames));
        }
    } else {
        List<String> attributeNames = new ArrayList<>();
        for (MetaProperty metaProperty : metaClass.getProperties()) {
            String propertyName = metaProperty.getName();
            if (metadataTools.isSystem(metaProperty)) {
                attributeNames.add(propertyName);
            }
            if (security.isEntityAttrUpdatePermitted(metaClass, propertyName) && !BaseEntityInternalAccess.isHiddenOrReadOnly(securityState, propertyName)) {
                attributeNames.add(metaProperty.getName());
            }
        }
        fetchGroupTracker._persistence_setFetchGroup(new CubaEntityFetchGroup(attributeNames));
    }
}

Example 3 with FetchGroupTracker

use of org.eclipse.persistence.queries.FetchGroupTracker in project cuba by cuba-platform.

the class EclipseLinkDescriptorEventListener method postClone.

@Override
public void postClone(DescriptorEvent event) {
    // in shared cache mode, postBuild event is missed, so we repeat it here
    if (event.getObject() instanceof BaseGenericIdEntity) {
        BaseEntityInternalAccess.setNew((BaseGenericIdEntity) event.getObject(), false);
    }
    if (event.getObject() instanceof FetchGroupTracker) {
        FetchGroupTracker entity = (FetchGroupTracker) event.getObject();
        FetchGroup fetchGroup = entity._persistence_getFetchGroup();
        if (fetchGroup != null && !(fetchGroup instanceof CubaEntityFetchGroup))
            entity._persistence_setFetchGroup(new CubaEntityFetchGroup(fetchGroup));
    }
    if (event.getObject() instanceof Entity)
        support.registerInstance((Entity) event.getObject(), event.getSession());
}

Example 4 with FetchGroupTracker

use of org.eclipse.persistence.queries.FetchGroupTracker in project cuba by cuba-platform.

the class EclipseLinkDescriptorEventListener method postRefresh.

@Override
public void postRefresh(DescriptorEvent event) {
    if (event.getObject() instanceof FetchGroupTracker) {
        FetchGroupTracker entity = (FetchGroupTracker) event.getObject();
        FetchGroup fetchGroup = entity._persistence_getFetchGroup();
        if (fetchGroup != null && !(fetchGroup instanceof CubaEntityFetchGroup))
            entity._persistence_setFetchGroup(new CubaEntityFetchGroup(fetchGroup));
    }
}

Example 5 with FetchGroupTracker

use of org.eclipse.persistence.queries.FetchGroupTracker in project cuba by cuba-platform.

the class EclipseLinkDescriptorEventListener method postBuild.

@Override
public void postBuild(DescriptorEvent event) {
    if (event.getObject() instanceof BaseGenericIdEntity) {
        BaseEntityInternalAccess.setNew((BaseGenericIdEntity) event.getObject(), false);
    }
    if (event.getObject() instanceof FetchGroupTracker) {
        FetchGroupTracker entity = (FetchGroupTracker) event.getObject();
        FetchGroup fetchGroup = entity._persistence_getFetchGroup();
        if (fetchGroup != null && !(fetchGroup instanceof CubaEntityFetchGroup))
            entity._persistence_setFetchGroup(new CubaEntityFetchGroup(fetchGroup));
    }
}