@Override public Set<EntityKeyMetadata> getAllEntityKeyMetadata() { Set<EntityKeyMetadata> allEntityKeyMetadata = new HashSet<>(); for (EntityPersister entityPersister : factory.getEntityPersisters().values()) { allEntityKeyMetadata.add(((OgmEntityPersister) entityPersister).getEntityKeyMetadata()); } return allEntityKeyMetadata; }
/** * Returns all the persistent id generators which potentially require the creation of an object in * the schema. */ private Iterable<PersistentNoSqlIdentifierGenerator> getPersistentGenerators() { Map<String, EntityPersister> entityPersisters = factory.getEntityPersisters(); Set<PersistentNoSqlIdentifierGenerator> persistentGenerators = new HashSet<PersistentNoSqlIdentifierGenerator>(entityPersisters.size()); for (EntityPersister persister : entityPersisters.values()) { if (persister.getIdentifierGenerator() instanceof PersistentNoSqlIdentifierGenerator) { persistentGenerators.add( (PersistentNoSqlIdentifierGenerator) persister.getIdentifierGenerator()); } } return persistentGenerators; }
private Map<EntityKeyMetadata, EmbeddedNeo4jEntityQueries> initializeEntityQueries( SessionFactoryImplementor sessionFactoryImplementor) { Map<EntityKeyMetadata, EmbeddedNeo4jEntityQueries> queryMap = new HashMap<EntityKeyMetadata, EmbeddedNeo4jEntityQueries>(); Collection<EntityPersister> entityPersisters = sessionFactoryImplementor.getEntityPersisters().values(); for (EntityPersister entityPersister : entityPersisters) { if (entityPersister instanceof OgmEntityPersister) { OgmEntityPersister ogmEntityPersister = (OgmEntityPersister) entityPersister; queryMap.put( ogmEntityPersister.getEntityKeyMetadata(), new EmbeddedNeo4jEntityQueries(ogmEntityPersister.getEntityKeyMetadata())); } } return queryMap; }
@Override public Set<AssociationKeyMetadata> getAllAssociationKeyMetadata() { Set<AssociationKeyMetadata> allAssociationKeyMetadata = new HashSet<>(); for (CollectionPersister associationPersister : factory.getCollectionPersisters().values()) { allAssociationKeyMetadata.add( ((OgmCollectionPersister) associationPersister).getAssociationKeyMetadata()); } for (EntityPersister entityPersister : factory.getEntityPersisters().values()) { for (String property : entityPersister.getPropertyNames()) { AssociationKeyMetadata inverseOneToOneAssociationKeyMetadata = ((OgmEntityPersister) entityPersister) .getInverseOneToOneAssociationKeyMetadata(property); if (inverseOneToOneAssociationKeyMetadata != null) { allAssociationKeyMetadata.add(inverseOneToOneAssociationKeyMetadata); } } } return allAssociationKeyMetadata; }