examples/org.eclipse.ocl.examples.interpreter/src/org/eclipse/ocl/examples/interpreter/console/DelegatingPackageRegistry.java

/*******************************************************************************
 * Copyright (c) 2006, 2018 IBM Corporation and others.
 * All rights reserved.   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
 * http://www.eclipse.org/legal/epl-v20.html
 *
 * Contributors:
 *   IBM - Initial API and implementation
 *******************************************************************************/
package org.eclipse.ocl.examples.interpreter.console;

import java.util.Collection;
import java.util.List;
import java.util.Map;
import java.util.Set;

import org.eclipse.emf.common.util.BasicEList;
import org.eclipse.emf.ecore.EFactory;
import org.eclipse.emf.ecore.EPackage;
import org.eclipse.emf.ecore.impl.EPackageRegistryImpl;
import org.eclipse.ocl.ecore.EcoreEnvironmentFactory;

/**
 * A package registry that delegates to multiple other registries (in order),
 * also delegating the {@link Map#values()} API in addition to package look-up.
 * This is safe to use with the {@link EcoreEnvironmentFactory} because Ecore
 * environments only iterate the values of a registry.
 *
 * @author Christian W. Damus (cdamus)
 */
public class DelegatingPackageRegistry extends EPackageRegistryImpl {
	private static final long serialVersionUID = -7606528898701523821L;

	private final List<EPackage.Registry> delegates;
	
	public DelegatingPackageRegistry(EPackage.Registry delegate) {
		delegates = new BasicEList.FastCompare<EPackage.Registry>(1);
		delegates.add(delegate);
	}
	
	public DelegatingPackageRegistry(EPackage.Registry delegate1, EPackage.Registry delegate2) {
		delegates = new BasicEList.FastCompare<EPackage.Registry>(2);
		delegates.add(delegate1);
		delegates.add(delegate2);
	}
	
	public DelegatingPackageRegistry(List<EPackage.Registry> delegates) {
		this.delegates = new BasicEList.FastCompare<EPackage.Registry>(delegates);
	}
	
	public Collection<Object> values() {
		Collection<Object> result = new java.util.ArrayList<Object>();
		
		Set<String> keys = new java.util.HashSet<String>(keySet());
		result.addAll(super.values());
		
		for (EPackage.Registry next : delegates) {
			// get values from the delegates for the keys that we do not yet have
			for (String key : next.keySet()) {
				if (!keys.contains(key)) {
					result.add(next.get(key));
				}
			}
		}
		
		return result;
	}
	
	protected EFactory delegatedGetEFactory(String nsURI) {
		EFactory result = null;
		
		for (EPackage.Registry next : delegates) {
			result = next.getEFactory(nsURI);
		}
		
		return result;
	}
	
	protected EPackage delegatedGetEPackage(String nsURI) {
		EPackage result = null;
		
		for (EPackage.Registry next : delegates) {
			result = next.getEPackage(nsURI);
		}
		
		return result;
	}
}