LazyEObjectImpl.java

/*******************************************************************************
 * Copyright (c) 2014 BestSolution.at and others.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *     Tom Schindl <tom.schindl@bestsolution.at> - initial API and implementation
 *******************************************************************************/
package at.bestsolution.persistence.model;

import java.util.HashMap;
import java.util.Map;

import org.eclipse.emf.ecore.EReference;
import org.eclipse.emf.ecore.EStructuralFeature;
import org.eclipse.emf.ecore.impl.EObjectImpl;

// TODO add missing emf methods (see CGLibObjectProxyInterceptor)
public abstract class LazyEObjectImpl extends EObjectImpl implements LazyEObject {

	private Map<EReference, Boolean> resolved = new HashMap<EReference, Boolean>();
	private ResolveDelegate delegate;
	private Object proxyData;

	@Override
	public Object eGet(EStructuralFeature eFeature, boolean resolve) {
		if( delegate != null ) {
			if( eFeature instanceof EReference ) {
				if( ! resolved.containsKey(eFeature) ) {
					if( delegate.resolve(this,proxyData,eFeature) ) {
						resolved.put((EReference) eFeature, Boolean.TRUE);
					}
				}
			}
		}

		return super.eGet(eFeature, resolve);
	}

	public void setProxyData(Object proxyData) {
		this.proxyData = proxyData;
	}

	@Override
	public void eSet(EStructuralFeature eFeature, Object newValue) {
		super.eSet(eFeature, newValue);
		if( eFeature instanceof EReference ) {
			resolved.put((EReference) eFeature, Boolean.TRUE);
		}
	}

	@Override
	public void setProxyDelegate(ResolveDelegate delegate) {
		this.delegate = delegate;
	}

	@Override
	public boolean isEnhanced() {
		return delegate != null;
	}

	@Override
	public boolean isResolved(EReference f) {
		return resolved.get(f) == Boolean.TRUE;
	}
}