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

説明を見る。

package org.objectweb.cjdbc.common.sql;

import java.io.Serializable;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.StringTokenizer;

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 CreateRequest extends AbstractWriteRequest implements Serializable
{
  private transient DatabaseTable table      = null;

  private transient ArrayList     fromTables = null;

  public CreateRequest(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 CreateRequest(String sqlQuery, boolean escapeProcessing, int timeout,
      String lineSeparator)
  {
    super(sqlQuery, escapeProcessing, timeout, lineSeparator);
    cacheable = RequestType.UNCACHEABLE;
    isParsed = false;
  }

  public void parse(DatabaseSchema schema, int granularity,
      boolean isCaseSensitive) throws SQLException
  {
    if (granularity == ParsingGranularities.NO_PARSING)
    {
      isParsed = true;
      return;
    }

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

    // Strip 'create [temporary] table '
    int tableIdx = sql.indexOf("table");
    sql = sql.substring(tableIdx + 5).trim();

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

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

    if (selectIdx != -1)
    {
      // Get the table on which CREATE occurs
      int nextSpaceIdx = sql.indexOf(" ");
      tableName = sql.substring(0, nextSpaceIdx).trim();
      table = new DatabaseTable(tableName);
      // Parse the select
      sql = sql.substring(selectIdx).trim();
      SelectRequest select = new SelectRequest(sql, false, 60,
          getLineSeparator());
      select.parse(schema, granularity, isCaseSensitive);
      fromTables = select.getFrom();
      if (granularity > ParsingGranularities.TABLE)
      {
        columns = select.getSelect();
      }

    }
    else
    {
      // Get the table on which CREATE occurs
      // Look for the parenthesis
      int openParenthesisIdx = sql.indexOf("(");
      int closeParenthesisIdx = sql.lastIndexOf(")");
      if ((openParenthesisIdx == -1) && (closeParenthesisIdx == -1))
      {
        // no parenthesis found
        table = new DatabaseTable(sql.trim());
        if (granularity > ParsingGranularities.TABLE)
        {
          columns = new ArrayList();
        }
        return;
      }
      else if ((openParenthesisIdx == -1) || (closeParenthesisIdx == -1)
          || (openParenthesisIdx > closeParenthesisIdx))
      {
        throw new SQLException("Syntax error in this CREATE statement: '"
            + sqlQuery + "'");
      }
      else
      {
        tableName = sql.substring(0, openParenthesisIdx).trim();
      }
      table = new DatabaseTable(tableName);

      // Get the column names
      if (granularity > ParsingGranularities.TABLE)
      {
        columns = new ArrayList();
        sql = sql.substring(openParenthesisIdx + 1, closeParenthesisIdx).trim();
        StringTokenizer columnTokens = new StringTokenizer(sql, ",");
        String word;
        String lowercaseWord;
        StringTokenizer wordTokens = null;
        String token;
        DatabaseColumn col = null;

        while (columnTokens.hasMoreTokens())
        {
          token = columnTokens.nextToken().trim();

          // work around to prevent bug: if the request contains for example:
          // INDEX foo (col1,col2)
          // we have to merge the 2 tokens: 'INDEX foo (col1' and 'col2)'
          if ((token.indexOf("(") != -1) && (token.indexOf(")") == -1))
          {
            if (columnTokens.hasMoreTokens())
              token = token + "," + columnTokens.nextToken().trim();
            else
            {
              tableName = null;
              columns = null;
              throw new SQLException("Syntax error in this CREATE statement: '"
                  + sqlQuery + "'");
            }
          }

          // First word of the line: either a column name or
          // a table constraint definition
          wordTokens = new StringTokenizer(token, " ");
          word = wordTokens.nextToken().trim();
          lowercaseWord = word.toLowerCase();

          // If it's a constraint, index or check keyword do not do anything
          // else parse the line
          if (!lowercaseWord.equals("constraint")
              && !lowercaseWord.equals("index")
              && !lowercaseWord.equals("check"))
          {
            String columnName;
            boolean isUnique = false;
            // Check for primary key or unique constraint
            if (lowercaseWord.equals("primary")
                || lowercaseWord.startsWith("unique"))
            {

              // Get the name of the column
              openParenthesisIdx = token.indexOf("(");
              closeParenthesisIdx = token.indexOf(")");
              if ((openParenthesisIdx == -1) || (closeParenthesisIdx == -1)
                  || (openParenthesisIdx > closeParenthesisIdx))
              {
                tableName = null;
                columns = null;
                throw new SQLException(
                    "Syntax error in this CREATE statement: '" + sqlQuery + "'");
              }

              columnName = token.substring(openParenthesisIdx + 1,
                  closeParenthesisIdx).trim();

              // Set this column to unique
              col = table.getColumn(columnName);
              // Test first if dbTable contains this column. This can fail with
              // some invalid request, for example:
              // CREATE TABLE categories(id INT4, name TEXT, PRIMARY KEY((id))
              if (col == null)
              {
                tableName = null;
                columns = null;
                throw new SQLException(
                    "Syntax error in this CREATE statement: '" + sqlQuery + "'");
              }
              else
                col.setIsUnique(true);
            }
            else
            {
              // It's a column name
              columnName = word;

              if (!wordTokens.hasMoreTokens())
              {
                // at least type declaration is required
                tableName = null;
                columns = null;
                throw new SQLException(
                    "Syntax error in this CREATE statement: '" + sqlQuery + "'");
              }

              // Check for primary key or unique constraints
              do
              {
                word = wordTokens.nextToken().trim().toLowerCase();
                if (word.equals("primary") || word.startsWith("unique"))
                {
                  // Create the column as unique
                  isUnique = true;
                  break;
                }
              }
              while (wordTokens.hasMoreTokens());

              // Add the column to the parsed columns list and
              // to the create DatabaseTable
              columns.add(new TableColumn(tableName, columnName));
              table.addColumn(new DatabaseColumn(columnName, isUnique));
            }
          }
        }
      }
    }
    isParsed = true;
  }

  public void cloneParsing(AbstractRequest request)
  {
    if (!request.isParsed())
      return;
    CreateRequest createRequest = (CreateRequest) request;
    cloneTableNameAndColumns((AbstractWriteRequest) request);
    table = createRequest.getDatabaseTable();
    fromTables = createRequest.getFromTables();
    isParsed = true;
  }

  public boolean isCreate()
  {
    return true;
  }

  public boolean isInsert()
  {
    return false;
  }

  public boolean isUpdate()
  {
    return false;
  }

  public boolean isDelete()
  {
    return false;
  }

  public boolean isDrop()
  {
    return false;
  }

  public DatabaseTable getDatabaseTable()
  {
    return table;
  }

  public ArrayList getFromTables()
  {
    return fromTables;
  }

  public void debug()
  {
    super.debug();
    if (tableName != null)
      System.out.println("Created table: " + tableName);
    else
      System.out.println("No information about created table");

    if (columns != null)
    {
      System.out.println("Created columns:");
      for (int i = 0; i < columns.size(); i++)
        System.out.println("  "
            + ((TableColumn) columns.get(i)).getColumnName());
    }
    else
      System.out.println("No information about created columns");

    System.out.println();
  }
  public boolean isAlter()
  {
    return false;
  }
}

---