@SuppressWarnings("unchecked") private static Class<? extends Entity<?, ?>> convertAsEntityType(SegmentType type) { Class<? extends AbstractElement<?, ?>> cls = AbstractElement.toElementClass(type); if (Relationship.class.isAssignableFrom(cls)) { throw new IllegalArgumentException("Type filter not applicable to relationships."); } return (Class<? extends Entity<?, ?>>) cls; }
.map((s) -> AbstractElement.toElementClass(s.getSegment().getElementType())).collect(toSet()); Set<String> expectedIds = Arrays.asList(sources).stream().map((s) -> s.getSegment().getElementId()) .collect(toSet());