-
Notifications
You must be signed in to change notification settings - Fork 45
/
EditingContextServices.java
81 lines (73 loc) · 2.98 KB
/
EditingContextServices.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
/*******************************************************************************
* Copyright (c) 2021, 2023 Obeo.
* This program and the accompanying materials
* are made available under the terms of the Eclipse Public License v2.0
* which accompanies this distribution, and is available at
* https://www.eclipse.org/legal/epl-2.0/
*
* SPDX-License-Identifier: EPL-2.0
*
* Contributors:
* Obeo - initial API and implementation
*******************************************************************************/
package org.eclipse.sirius.components.emf.query;
import java.util.Collection;
import java.util.Objects;
import java.util.Optional;
import java.util.Spliterator;
import java.util.Spliterators;
import java.util.stream.Stream;
import java.util.stream.StreamSupport;
import org.eclipse.emf.common.notify.Notifier;
import org.eclipse.emf.common.util.EList;
import org.eclipse.emf.ecore.EObject;
import org.eclipse.emf.ecore.resource.Resource;
import org.eclipse.emf.ecore.resource.ResourceSet;
import org.eclipse.emf.edit.domain.EditingDomain;
import org.eclipse.sirius.components.core.api.IEditingContext;
import org.eclipse.sirius.components.emf.services.api.IEMFEditingContext;
/**
* An utility class providing various query services.
*
* @author fbarbin
*/
public final class EditingContextServices {
public Collection<EObject> allContents(IEditingContext editingContext) {
return this.getResourceset(editingContext)
.stream()
.flatMap(this::collectAllContent)
.toList();
}
public Collection<EObject> contents(IEditingContext editingContext) {
return this.getResourceset(editingContext)
.stream()
.map(ResourceSet::getResources)
.flatMap(EList::stream)
.map(Resource::getContents)
.flatMap(EList::stream)
.toList();
}
public EObject getObjectById(IEditingContext editingContext, String id) {
return this.getResourceset(editingContext)
.stream()
.map(ResourceSet::getResources)
.flatMap(EList::stream)
.map(resource -> resource.getEObject(id))
.filter(Objects::nonNull)
.findFirst()
.orElse(null);
}
private Optional<ResourceSet> getResourceset(IEditingContext editingContext) {
return Optional.of(editingContext)
.filter(IEMFEditingContext.class::isInstance)
.map(IEMFEditingContext.class::cast)
.map(IEMFEditingContext::getDomain)
.map(EditingDomain::getResourceSet);
}
private Stream<EObject> collectAllContent(ResourceSet resourceSet) {
Spliterator<Notifier> spliterator = Spliterators.spliteratorUnknownSize(resourceSet.getAllContents(), Spliterator.ORDERED);
return StreamSupport.stream(spliterator, false)
.filter(EObject.class::isInstance)
.map(EObject.class::cast);
}
}