src/org/objectweb/cjdbc/common/sql/AlterRequest.java

説明を見る。

package org.objectweb.cjdbc.common.sql;

import java.sql.SQLException;
import java.util.ArrayList;

import org.objectweb.cjdbc.common.sql.schema.DatabaseColumn;
import org.objectweb.cjdbc.common.sql.schema.DatabaseSchema;
import org.objectweb.cjdbc.common.sql.schema.DatabaseTable;
import org.objectweb.cjdbc.common.sql.schema.TableColumn;

public class AlterRequest extends AbstractWriteRequest
{

  private transient DatabaseTable  table  = null;

  private transient DatabaseColumn column = null;

  private transient boolean        isDrop = false;
  private transient boolean        isAdd  = false;

  public AlterRequest(String sqlQuery, boolean escapeProcessing, int timeout,
      String lineSeparator)
  {
    super(sqlQuery, escapeProcessing, timeout, lineSeparator);
  }

  public AlterRequest(String sqlQuery, boolean escapeProcessing, int timeout,
      String lineSeparator, DatabaseSchema schema, int granularity,
      boolean isCaseSensitive) throws SQLException
  {
    this(sqlQuery, escapeProcessing, timeout, lineSeparator);
    parse(schema, granularity, isCaseSensitive);
  }

  public boolean isReadRequest()
  {
    return false;
  }

  public boolean isWriteRequest()
  {
    return true;
  }

  public boolean isUnknownRequest()
  {
    return false;
  }

  public void parse(DatabaseSchema schema, int granularity,
      boolean isCaseSensitive) throws SQLException
  {
    /*
     * Example Alter statement: ALTER TABLE table_name ADD column_name datatype
     * ALTER TABLE table_name DROP COLUMN column_name
     */

    if (granularity == ParsingGranularities.NO_PARSING)
    {
      isParsed = true;
      return;
    }

    String originalSQL = this.trimCarriageReturn();
    String sql = originalSQL.toLowerCase();

    // Strip 'alter table '
    int tableIdx = sql.indexOf("table");
    if (tableIdx == -1)
      throw new SQLException(
          "Malformed Alter Request. Should start with [ALTER TABLE]");
    sql = sql.substring(tableIdx + 5).trim();

    // Does the query contain a add?
    int addIdx = sql.indexOf(" add ");

    // Does the query contain a drop?
    int dropIdx = sql.indexOf(" drop ");

    if (addIdx != -1)
      isAdd = true;
    if (dropIdx != -1)
      isDrop = true;

    if (!isAdd && !isDrop)
      throw new SQLException(
          "Malformed Alter Request. No drop or add condition");

    if (isCaseSensitive) // Reverse to the original case
      sql = originalSQL.substring(originalSQL.length() - sql.length());

    int index = (isAdd) ? addIdx : dropIdx;

    tableName = sql.substring(0, index).trim();
    table = new DatabaseTable(tableName);


    if (granularity > ParsingGranularities.TABLE)
    {
      

      int subsIndex = index + 6 + 2; // index +
                                                               // column.length()
                                                               // + space
      if (isAdd)
        subsIndex += 3;
      else
        // Drop
        subsIndex += 4;
      
      columns = new ArrayList();
      sql = sql.substring(subsIndex).trim();

      if (isAdd)
      {
        int colIndex = sql.indexOf(' ');
        String colName = sql.substring(0, colIndex);

        int uniqueIndex = sql.toLowerCase().indexOf("unique");
        int primary = sql.toLowerCase().indexOf("primary");
        if (uniqueIndex != -1 || primary != -1)
          column = new DatabaseColumn(colName, true);
        else
          column = new DatabaseColumn(colName, false);
        columns.add(new TableColumn(tableName, colName));
      }
      else if (isDrop)
      {
        String colName = sql.trim();
        column = schema.getTable(tableName).getColumn(colName);
        columns.add(new TableColumn(tableName, colName));
      }
    }
    isParsed  = true;
  }

  public void cloneParsing(AbstractRequest request)
  {
    if (!request.isParsed())
      return;
    AlterRequest alterRequest = (AlterRequest) request;
    cloneTableNameAndColumns((AbstractWriteRequest) request);
    table = alterRequest.getDatabaseTable();
    column = alterRequest.getColumn();
    isParsed = true;
  }

  public boolean isAlter()
  {
    return true;
  }

  public boolean isCreate()
  {
    return false;
  }

  public boolean isDelete()
  {
    return false;
  }

  public boolean isDrop()
  {
    return false;
  }

  public boolean isInsert()
  {
    return false;
  }

  public boolean isUpdate()
  {
    return false;
  }

  public DatabaseTable getDatabaseTable()
  {
    return table;
  }
  public DatabaseColumn getColumn()
  {
    return column;
  }
  public boolean isAdd()
  {
    return isAdd;
  }
}

---