Avoid creation of list afterwards converted to set

Co-authored-by: Joshua Gleitze <[email protected]>
vitruv-tools · Feb 19, 2021 · bf273cb · bf273cb
1 parent 0b1745c
commit bf273cb
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/...ence/src/tools/vitruv/framework/correspondence/impl/InternalCorrespondenceModelImpl.xtend b/...ence/src/tools/vitruv/framework/correspondence/impl/InternalCorrespondenceModelImpl.xtend
@@ -123,13 +123,13 @@ class InternalCorrespondenceModelImpl extends ModelInstance implements InternalC
 		Predicate<C> correspondencesFilter, List<EObject> aEObjects, List<EObject> bEObjects, String tag) {
 		getCorrespondences(correspondenceType, correspondencesFilter, aEObjects, tag)
 			.filter[EcoreUtil.equals(bEObjects, resolveCorrespondingObjects(aEObjects))]
-			.flatMapFixed[removeCorrespondencesAndDependendCorrespondences].toSet
+			.flatMapFixedTo(new LinkedHashSet) [removeCorrespondencesAndDependendCorrespondences()]
 	}
 
 	override <C extends Correspondence> Set<Correspondence> removeCorrespondencesFor(Class<C> correspondenceType,
 		Predicate<C> correspondencesFilter, List<EObject> eObjects, String tag) {
 		getCorrespondences(correspondenceType, correspondencesFilter, eObjects, tag)
-			.flatMapFixed[removeCorrespondencesAndDependendCorrespondences].toSet
+			.flatMapFixedTo(new LinkedHashSet) [removeCorrespondencesAndDependendCorrespondences()]
 	}
 
 	private def Set<Correspondence> removeCorrespondencesAndDependendCorrespondences(Correspondence correspondence) {
@@ -177,7 +177,7 @@ class InternalCorrespondenceModelImpl extends ModelInstance implements InternalC
 	override <C extends Correspondence> Set<List<EObject>> getCorrespondingEObjects(Class<C> correspondenceType,
 		Predicate<C> correspondencesFilter, List<EObject> eObjects, String tag) {
 		return getCorrespondences(correspondenceType, correspondencesFilter, eObjects, tag)
-			.mapFixed[resolveCorrespondingObjects(eObjects)].toSet
+			.mapFixedTo(new LinkedHashSet) [resolveCorrespondingObjects(eObjects)]
 	}
 
 	override List<EObject> getCorrespondingEObjectsInCorrespondence(Correspondence correspondence,