private Map<AssociationKeyMetadata, EmbeddedNeo4jAssociationQueries> initializeAssociationQueries(
     SessionFactoryImplementor sessionFactoryImplementor) {
   Map<AssociationKeyMetadata, EmbeddedNeo4jAssociationQueries> queryMap =
       new HashMap<AssociationKeyMetadata, EmbeddedNeo4jAssociationQueries>();
   Collection<CollectionPersister> collectionPersisters =
       sessionFactoryImplementor.getCollectionPersisters().values();
   for (CollectionPersister collectionPersister : collectionPersisters) {
     if (collectionPersister instanceof OgmCollectionPersister) {
       OgmCollectionPersister ogmCollectionPersister =
           (OgmCollectionPersister) collectionPersister;
       EntityKeyMetadata ownerEntityKeyMetadata =
           ((OgmEntityPersister) (ogmCollectionPersister.getOwnerEntityPersister()))
               .getEntityKeyMetadata();
       AssociationKeyMetadata associationKeyMetadata =
           ogmCollectionPersister.getAssociationKeyMetadata();
       queryMap.put(
           associationKeyMetadata,
           new EmbeddedNeo4jAssociationQueries(ownerEntityKeyMetadata, associationKeyMetadata));
     }
   }
   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;
  }