plugins/org.eclipse.emf.cdo.server.db/src/org/eclipse/emf/cdo/server/internal/db/AttributeMapping.java

/***************************************************************************
 * Copyright (c) 2004 - 2008 Eike Stepper, Germany.
 * 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:
 *    Eike Stepper - initial API and implementation
 **************************************************************************/
package org.eclipse.emf.cdo.server.internal.db;

import org.eclipse.emf.cdo.common.id.CDOIDUtil;
import org.eclipse.emf.cdo.common.model.CDOFeature;
import org.eclipse.emf.cdo.common.revision.CDORevision;
import org.eclipse.emf.cdo.server.db.IAttributeMapping;
import org.eclipse.emf.cdo.spi.common.revision.InternalCDORevision;

import org.eclipse.net4j.db.DBException;
import org.eclipse.net4j.db.IDBAdapter;
import org.eclipse.net4j.db.ddl.IDBField;

import java.sql.ResultSet;
import java.sql.SQLException;

/**
 * @author Eike Stepper
 */
public abstract class AttributeMapping extends FeatureMapping implements IAttributeMapping
{
  private IDBField field;

  public AttributeMapping(ClassMapping classMapping, CDOFeature feature)
  {
    super(classMapping, feature);
    field = classMapping.addField(feature, classMapping.getTable());
  }

  public IDBField getField()
  {
    return field;
  }

  public void appendValue(StringBuilder builder, CDORevision revision)
  {
    IDBAdapter dbAdapter = getDBAdapter();
    Object value = getRevisionValue(revision);
    dbAdapter.appendValue(builder, field, value);
  }

  public Object getRevisionValue(CDORevision revision)
  {
    CDOFeature feature = getFeature();
    return ((InternalCDORevision)revision).getValue(feature);
  }

  public void extractValue(ResultSet resultSet, int column, CDORevision revision)
  {
    try
    {
      Object value = getResultSetValue(resultSet, column);
      if (resultSet.wasNull())
      {
        value = null;
      }

      ((InternalCDORevision)revision).setValue(getFeature(), value);
    }
    catch (SQLException ex)
    {
      throw new DBException(ex);
    }
  }

  protected abstract Object getResultSetValue(ResultSet resultSet, int column) throws SQLException;

  /**
   * @author Eike Stepper
   */
  public static class AMString extends AttributeMapping
  {
    public AMString(ClassMapping classMapping, CDOFeature feature)
    {
      super(classMapping, feature);
    }

    @Override
    protected Object getResultSetValue(ResultSet resultSet, int column) throws SQLException
    {
      return resultSet.getString(column);
    }
  }

  /**
   * @author Eike Stepper
   */
  public static class AMShort extends AttributeMapping
  {
    public AMShort(ClassMapping classMapping, CDOFeature feature)
    {
      super(classMapping, feature);
    }

    @Override
    protected Object getResultSetValue(ResultSet resultSet, int column) throws SQLException
    {
      return resultSet.getShort(column);
    }
  }

  /**
   * @author Eike Stepper
   */
  public static class AMObject extends AttributeMapping
  {
    public AMObject(ClassMapping classMapping, CDOFeature feature)
    {
      super(classMapping, feature);
    }

    @Override
    protected Object getResultSetValue(ResultSet resultSet, int column) throws SQLException
    {
      long id = resultSet.getLong(column);
      if (resultSet.wasNull())
      {
        return null;
      }

      return CDOIDUtil.createLong(id);
    }
  }

  /**
   * @author Eike Stepper
   */
  public static class AMLong extends AttributeMapping
  {
    public AMLong(ClassMapping classMapping, CDOFeature feature)
    {
      super(classMapping, feature);
    }

    @Override
    protected Object getResultSetValue(ResultSet resultSet, int column) throws SQLException
    {
      return resultSet.getLong(column);
    }
  }

  /**
   * @author Eike Stepper
   */
  public static class AMInteger extends AttributeMapping
  {
    public AMInteger(ClassMapping classMapping, CDOFeature feature)
    {
      super(classMapping, feature);
    }

    @Override
    protected Object getResultSetValue(ResultSet resultSet, int column) throws SQLException
    {
      return resultSet.getInt(column);
    }
  }

  /**
   * @author Eike Stepper
   */
  public static class AMFloat extends AttributeMapping
  {
    public AMFloat(ClassMapping classMapping, CDOFeature feature)
    {
      super(classMapping, feature);
    }

    @Override
    protected Object getResultSetValue(ResultSet resultSet, int column) throws SQLException
    {
      return resultSet.getFloat(column);
    }
  }

  /**
   * @author Eike Stepper
   */
  public static class AMDouble extends AttributeMapping
  {
    public AMDouble(ClassMapping classMapping, CDOFeature feature)
    {
      super(classMapping, feature);
    }

    @Override
    protected Object getResultSetValue(ResultSet resultSet, int column) throws SQLException
    {
      return resultSet.getDouble(column);
    }
  }

  /**
   * @author Eike Stepper
   */
  public static class AMDate extends AttributeMapping
  {
    public AMDate(ClassMapping classMapping, CDOFeature feature)
    {
      super(classMapping, feature);
    }

    @Override
    protected Object getResultSetValue(ResultSet resultSet, int column) throws SQLException
    {
      // TODO Is getDate() correct?
      return resultSet.getDate(column);
    }
  }

  /**
   * @author Eike Stepper
   */
  public static class AMCharacter extends AttributeMapping
  {
    public AMCharacter(ClassMapping classMapping, CDOFeature feature)
    {
      super(classMapping, feature);
    }

    @Override
    protected Object getResultSetValue(ResultSet resultSet, int column) throws SQLException
    {
      String str = resultSet.getString(column);
      if (resultSet.wasNull())
      {
        return null;
      }

      return str.charAt(0);
    }
  }

  /**
   * @author Eike Stepper
   */
  public static class AMByte extends AttributeMapping
  {
    public AMByte(ClassMapping classMapping, CDOFeature feature)
    {
      super(classMapping, feature);
    }

    @Override
    protected Object getResultSetValue(ResultSet resultSet, int column) throws SQLException
    {
      return resultSet.getByte(column);
    }
  }

  /**
   * @author Eike Stepper
   */
  public static class AMBoolean extends AttributeMapping
  {
    public AMBoolean(ClassMapping classMapping, CDOFeature feature)
    {
      super(classMapping, feature);
    }

    @Override
    protected Object getResultSetValue(ResultSet resultSet, int column) throws SQLException
    {
      return resultSet.getBoolean(column);
    }
  }
}