1 files changed, 37 insertions, 0 deletions
diff --git a/rba.tool.editor.ui/src/rba/tool/editor/ui/resource/RBAModelResourceSetProvider.xtend b/rba.tool.editor.ui/src/rba/tool/editor/ui/resource/RBAModelResourceSetProvider.xtend
new file mode 100644
index 0000000..5c4a6dd
--- /dev/null
+++ b/rba.tool.editor.ui/src/rba/tool/editor/ui/resource/RBAModelResourceSetProvider.xtend
@@ -0,0 +1,37 @@
+package rba.tool.editor.ui.resource
+
+import com.google.inject.Singleton
+import java.util.concurrent.ConcurrentHashMap
+import org.eclipse.core.resources.IProject
+import org.eclipse.emf.ecore.resource.ResourceSet
+import org.eclipse.xtext.ui.resource.SimpleResourceSetProvider
+
+@Singleton
+class RBAModelResourceSetProvider extends SimpleResourceSetProvider {
+
+	private ConcurrentHashMap<IProject, ResourceSet> projectXtextResourceSetMap = new ConcurrentHashMap<IProject, ResourceSet>();
+
+	override get(IProject project) {
+		val ResourceSet result = super.get(project);
+		if(project !== null) {
+			if(projectXtextResourceSetMap.containsKey(project)) {
+				projectXtextResourceSetMap.remove(project);
+			}
+			projectXtextResourceSetMap.put(project, result);
+		}
+		return result;
+	}
+
+	def public ResourceSet getResourceSet(IProject project) {
+		if(project === null) {
+			return get(project);
+		} else {
+			return projectXtextResourceSetMap.get(project) ?: get(project);
+		}
+	}
+
+	def public void setResourceSet(IProject project, ResourceSet resourceSet) {
+		projectXtextResourceSetMap.put(project, resourceSet);
+	}
+
+}