@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;
}
