Open Source Repository

Home /hibernate/hibernate-3.2.7.ga | Repository Home


org/hibernate/proxy/pojo/BasicLazyInitializer.java
//$Id: BasicLazyInitializer.java 9210 2006-02-03 22:15:19Z steveebersole $
package org.hibernate.proxy.pojo;

import java.io.Serializable;
import java.lang.reflect.Method;

import org.hibernate.engine.EntityKey;
import org.hibernate.engine.SessionImplementor;
import org.hibernate.type.AbstractComponentType;
import org.hibernate.util.MarkerObject;
import org.hibernate.util.ReflectHelper;
import org.hibernate.proxy.AbstractLazyInitializer;

/**
 * Lazy initializer for POJOs
 @author Gavin King
 */
public abstract class BasicLazyInitializer extends AbstractLazyInitializer {

  protected static final Object INVOKE_IMPLEMENTATION = new MarkerObject("INVOKE_IMPLEMENTATION");

  protected Class persistentClass;
  protected Method getIdentifierMethod;
  protected Method setIdentifierMethod;
  protected boolean overridesEquals;
  private Object replacement;
  protected AbstractComponentType componentIdType;

  protected BasicLazyInitializer(
      String entityName,
          Class persistentClass,
          Serializable id,
          Method getIdentifierMethod,
          Method setIdentifierMethod,
          AbstractComponentType componentIdType,
          SessionImplementor session) {
    super(entityName, id, session);
    this.persistentClass = persistentClass;
    this.getIdentifierMethod = getIdentifierMethod;
    this.setIdentifierMethod = setIdentifierMethod;
    this.componentIdType = componentIdType;
    overridesEquals = ReflectHelper.overridesEquals(persistentClass);
  }

  protected abstract Object serializableProxy();

  protected final Object invoke(Method method, Object[] args, Object proxythrows Throwable {

    String methodName = method.getName();
    int params = args.length;

    if params==) {

      if "writeReplace".equals(methodName) ) {
        return getReplacement();
      }
      else if !overridesEquals && "hashCode".equals(methodName) ) {
        return new IntegerSystem.identityHashCode(proxy) );
      }
      else if isUninitialized() && method.equals(getIdentifierMethod) ) {
        return getIdentifier();
      }

      else if "getHibernateLazyInitializer".equals(methodName) ) {
        return this;
      }

    }
    else if params==) {

      if !overridesEquals && "equals".equals(methodName) ) {
        return args[0]==proxy ? Boolean.TRUE : Boolean.FALSE;
      }
      else if method.equals(setIdentifierMethod) ) {
        initialize();
        setIdentifier( (Serializableargs[0] );
        return INVOKE_IMPLEMENTATION;
      }

    }

    //if it is a property of an embedded component, invoke on the "identifier"
    if componentIdType!=null && componentIdType.isMethodOf(method) ) {
      return method.invokegetIdentifier(), args );
    }

    // otherwise:
    return INVOKE_IMPLEMENTATION;

  }

  private Object getReplacement() {

    final SessionImplementor session = getSession();
    if isUninitialized() && session != null && session.isOpen()) {
      final EntityKey key = new EntityKey(
          getIdentifier(),
              session.getFactory().getEntityPersistergetEntityName() ),
              session.getEntityMode()
        );
      final Object entity = session.getPersistenceContext().getEntity(key);
      if (entity!=nullsetImplementationentity );
    }

    if isUninitialized() ) {
      if (replacement==null) {
        replacement = serializableProxy();
      }
      return replacement;
    }
    else {
      return getTarget();
    }

  }

  public final Class getPersistentClass() {
    return persistentClass;
  }

}