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
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
|
/*
* SPDX-FileCopyrightText: 2021-2023 The Refinery Authors <https://refinery.tools/>
*
* SPDX-License-Identifier: EPL-2.0
*/
package tools.refinery.language.web.xtext.server.push;
import com.google.common.collect.ImmutableList;
import org.eclipse.xtext.util.CancelIndicator;
import org.eclipse.xtext.web.server.IServiceResult;
import org.eclipse.xtext.web.server.model.AbstractCachedService;
import org.eclipse.xtext.web.server.model.DocumentSynchronizer;
import org.eclipse.xtext.web.server.model.XtextWebDocument;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import tools.refinery.language.web.generator.ModelGenerationManager;
import tools.refinery.language.web.xtext.server.ResponseHandlerException;
import java.util.ArrayList;
import java.util.List;
public class PushWebDocument extends XtextWebDocument {
private static final Logger LOG = LoggerFactory.getLogger(PushWebDocument.class);
private final List<PrecomputationListener> precomputationListeners = new ArrayList<>();
private final ModelGenerationManager modelGenerationManager = new ModelGenerationManager();
private final DocumentSynchronizer synchronizer;
public PushWebDocument(String resourceId, DocumentSynchronizer synchronizer) {
super(resourceId, synchronizer);
this.synchronizer = synchronizer;
}
public ModelGenerationManager getModelGenerationManager() {
return modelGenerationManager;
}
public void addPrecomputationListener(PrecomputationListener listener) {
synchronized (precomputationListeners) {
if (precomputationListeners.contains(listener)) {
return;
}
precomputationListeners.add(listener);
listener.onSubscribeToPrecomputationEvents(getResourceId(), this);
}
}
public void removePrecomputationListener(PrecomputationListener listener) {
synchronized (precomputationListeners) {
precomputationListeners.remove(listener);
}
}
public <T extends IServiceResult> void precomputeServiceResult(AbstractCachedService<T> service, String serviceName,
CancelIndicator cancelIndicator, boolean logCacheMiss) {
var result = getCachedServiceResult(service, cancelIndicator, logCacheMiss);
if (result != null) {
notifyPrecomputationListeners(serviceName, result);
}
}
public <T extends IServiceResult> void notifyPrecomputationListeners(String serviceName, T result) {
var resourceId = getResourceId();
if (resourceId == null) {
return;
}
var stateId = getStateId();
List<PrecomputationListener> copyOfListeners;
synchronized (precomputationListeners) {
copyOfListeners = ImmutableList.copyOf(precomputationListeners);
}
var toRemove = new ArrayList<PrecomputationListener>();
for (var listener : copyOfListeners) {
try {
listener.onPrecomputedServiceResult(resourceId, stateId, serviceName, result);
} catch (ResponseHandlerException e) {
LOG.error("Delivering precomputation push message failed", e);
toRemove.add(listener);
}
}
if (!toRemove.isEmpty()) {
synchronized (precomputationListeners) {
precomputationListeners.removeAll(toRemove);
}
}
}
public void cancelModelGeneration() {
modelGenerationManager.cancel();
}
public void dispose() {
synchronizer.setCanceled(true);
modelGenerationManager.dispose();
}
}
|