Open Source Repository

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


org/hibernate/hql/ast/ParameterTranslationsImpl.java
package org.hibernate.hql.ast;

import org.hibernate.hql.ParameterTranslations;
import org.hibernate.type.Type;
import org.hibernate.param.ParameterSpecification;
import org.hibernate.param.PositionalParameterSpecification;
import org.hibernate.param.NamedParameterSpecification;
import org.hibernate.util.ArrayHelper;

import java.util.Map;
import java.util.Set;
import java.util.List;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.Iterator;
import java.io.Serializable;

/**
 * Defines the information available for parameters encountered during
 * query translation through the antlr-based parser.
 *
 @author Steve Ebersole
 */
public class ParameterTranslationsImpl implements ParameterTranslations {

  private final Map namedParameters;
  private final ParameterInfo[] ordinalParameters;

  public boolean supportsOrdinalParameterMetadata() {
    return true;
  }

  public int getOrdinalParameterCount() {
    return ordinalParameters.length;
  }

  public ParameterInfo getOrdinalParameterInfo(int ordinalPosition) {
    // remember that ordinal parameters numbers are 1-based!!!
    return ordinalParameters[ordinalPosition - 1];
  }

  public int getOrdinalParameterSqlLocation(int ordinalPosition) {
    return getOrdinalParameterInfoordinalPosition ).getSqlLocations()[0];
  }

  public Type getOrdinalParameterExpectedType(int ordinalPosition) {
    return getOrdinalParameterInfoordinalPosition ).getExpectedType();
  }

  public Set getNamedParameterNames() {
    return namedParameters.keySet();
  }

  public ParameterInfo getNamedParameterInfo(String name) {
    return ParameterInfo namedParameters.getname );
  }

  public int[] getNamedParameterSqlLocations(String name) {
    return getNamedParameterInfoname ).getSqlLocations();
  }

  public Type getNamedParameterExpectedType(String name) {
    return getNamedParameterInfoname ).getExpectedType();
  }

  /**
   * Constructs a parameter metadata object given a list of parameter
   * specifications.
   </p>
   * Note: the order in the incoming list denotes the parameter's
   * psudeo-position within the resulting sql statement.
   *
   @param parameterSpecifications
   */
  public ParameterTranslationsImpl(List parameterSpecifications) {

    class NamedParamTempHolder {
      String name;
      Type type;
      List positions = new ArrayList();
    }

    int size = parameterSpecifications.size();
    List ordinalParameterList = new ArrayList();
    Map namedParameterMap = new HashMap();
    for int i = 0; i < size; i++ ) {
      final ParameterSpecification spec = ParameterSpecification parameterSpecifications.get);
      if PositionalParameterSpecification.class.isAssignableFromspec.getClass() ) ) {
        PositionalParameterSpecification ordinalSpec = PositionalParameterSpecification spec;
        ordinalParameterList.addnew ParameterInfoi, ordinalSpec.getExpectedType() ) );
      }
      else if NamedParameterSpecification.class.isAssignableFromspec.getClass() ) ) {
        NamedParameterSpecification namedSpec = NamedParameterSpecification spec;
        NamedParamTempHolder paramHolder = NamedParamTempHolder namedParameterMap.getnamedSpec.getName() );
        if paramHolder == null ) {
          paramHolder = new NamedParamTempHolder();
          paramHolder.name = namedSpec.getName();
          paramHolder.type = namedSpec.getExpectedType();
          namedParameterMap.putnamedSpec.getName(), paramHolder );
        }
        paramHolder.positions.addnew Integer) );
      }
      else {
        // don't care about other param types here, just those explicitly user-defined...
      }
    }

    ordinalParameters = ParameterInfo[] ) ordinalParameterList.toArraynew ParameterInfo[ordinalParameterList.size()] );

    if namedParameterMap.isEmpty() ) {
      namedParameters = java.util.Collections.EMPTY_MAP;
    }
    else {
      Map namedParametersBacking = new HashMapnamedParameterMap.size() );
      Iterator itr = namedParameterMap.values().iterator();
      whileitr.hasNext() ) {
        final NamedParamTempHolder holder = NamedParamTempHolder itr.next();
        namedParametersBacking.put(
            holder.name,
                new ParameterInfoArrayHelper.toIntArrayholder.positions ), holder.type )
        );
      }
      namedParameters = java.util.Collections.unmodifiableMapnamedParametersBacking );
    }
  }

  public static class ParameterInfo implements Serializable {
    private final int[] sqlLocations;
    private final Type expectedType;

    public ParameterInfo(int[] sqlPositions, Type expectedType) {
      this.sqlLocations = sqlPositions;
      this.expectedType = expectedType;
    }

    public ParameterInfo(int sqlPosition, Type expectedType) {
      this.sqlLocations = new int[] { sqlPosition };
      this.expectedType = expectedType;
    }

    public int[] getSqlLocations() {
      return sqlLocations;
    }

    public Type getExpectedType() {
      return expectedType;
    }
  }
}