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;
}
private EntityMetadataInformation determineSingleEntityMetadataInformation(
SessionFactoryImplementor sessionFactory, List<Return> customQueryReturns) {
EntityMetadataInformation metadataInformation = null;
for (Return queryReturn : customQueryReturns) {
if (queryReturn instanceof RootReturn) {
if (metadataInformation != null) {
return null;
}
RootReturn rootReturn = (RootReturn) queryReturn;
OgmEntityPersister persister =
(OgmEntityPersister) sessionFactory.getEntityPersister(rootReturn.getEntityName());
metadataInformation =
new EntityMetadataInformation(
persister.getEntityKeyMetadata(), rootReturn.getEntityName());
}
}
return metadataInformation;
}