Exemplo n.º 1
0
 public void initialize(
     Row row, Key hKey, SpatialColumnHandler spatialColumnHandler, long zValue) {
   pKeyAppends = 0;
   int indexField = 0;
   IndexRowComposition indexRowComp = index.indexRowComposition();
   while (indexField < indexRowComp.getLength()) {
     // handleSpatialColumn will increment pKeyAppends once for all spatial columns
     if (spatialColumnHandler != null
         && spatialColumnHandler.handleSpatialColumn(this, indexField, zValue)) {
       if (indexField == index.firstSpatialArgument()) {
         pKeyAppends++;
       }
     } else {
       if (indexRowComp.isInRowData(indexField)) {
         int position = indexRowComp.getFieldPosition(indexField);
         Column column = row.rowType().table().getColumnsIncludingInternal().get(position);
         ValueSource source = row.value(column.getPosition());
         pKeyTarget().append(source, column.getType());
       } else if (indexRowComp.isInHKey(indexField)) {
         PersistitKey.appendFieldFromKey(
             pKey(), hKey, indexRowComp.getHKeyPosition(indexField), index.getIndexName());
       } else {
         throw new IllegalStateException("Invalid IndexRowComposition: " + indexRowComp);
       }
       pKeyAppends++;
     }
     indexField++;
   }
 }
Exemplo n.º 2
0
 private static void handleModifyColumnNode(
     ModifyColumnNode modNode,
     AISBuilder builder,
     Table tableCopy,
     TypesTranslator typesTranslator) {
   AkibanInformationSchema aisCopy = tableCopy.getAIS();
   Column column = tableCopy.getColumn(modNode.getColumnName());
   assert column != null : modNode.getColumnName();
   switch (modNode.getNodeType()) {
     case NodeTypes.MODIFY_COLUMN_DEFAULT_NODE:
       if (modNode.isAutoincrementColumn()) {
         int autoIncType = (int) modNode.getAutoinc_create_or_modify_Start_Increment();
         switch (autoIncType) {
           case ColumnDefinitionNode.CREATE_AUTOINCREMENT:
             {
               if (column.getIdentityGenerator() != null) {
                 throw new ColumnAlreadyGeneratedException(column);
               }
               TableName name = tableCopy.getName();
               TableDDL.setAutoIncrement(
                   builder, name.getSchemaName(), name.getTableName(), modNode);
             }
             break;
           case ColumnDefinitionNode.MODIFY_AUTOINCREMENT_INC_VALUE:
             throw new UnsupportedSQLException("SET INCREMENT BY", modNode);
           case ColumnDefinitionNode.MODIFY_AUTOINCREMENT_RESTART_VALUE:
             // Note: Also handled above
             throw new UnsupportedSQLException("RESTART WITH", modNode);
           default:
             throw new IllegalStateException("Unknown autoIncType: " + autoIncType);
         }
       } else {
         // DROP DEFAULT will come though as a NULL default, clears both GENERATED and DEFAULT
         Sequence seq = column.getIdentityGenerator();
         if (seq != null) {
           column.setDefaultIdentity(null);
           column.setIdentityGenerator(null);
           aisCopy.removeSequence(seq.getSequenceName());
         }
         String[] defaultValueFunction =
             TableDDL.getColumnDefault(
                 modNode, tableCopy.getName().getSchemaName(), tableCopy.getName().getTableName());
         column.setDefaultValue(defaultValueFunction[0]);
         column.setDefaultFunction(defaultValueFunction[1]);
       }
       break;
     case NodeTypes.MODIFY_COLUMN_CONSTRAINT_NODE: // Type only comes from NULL
       column.setType(column.getType().withNullable(true));
       break;
     case NodeTypes.MODIFY_COLUMN_CONSTRAINT_NOT_NULL_NODE: // Type only comes from NOT NULL
       column.setType(column.getType().withNullable(false));
       break;
     case NodeTypes.MODIFY_COLUMN_TYPE_NODE: // All but [NOT] NULL comes from type
       {
         TInstance type =
             typesTranslator.typeForSQLType(modNode.getType()).withNullable(column.getNullable());
         if (false) {
           // TODO: Determine whether compatible, does affect sequence, etc.
           column.setType(type);
         } else {
           tableCopy.dropColumn(modNode.getColumnName());
           builder.column(
               tableCopy.getName().getSchemaName(),
               tableCopy.getName().getTableName(),
               column.getName(),
               column.getPosition(),
               type,
               false, // column.getInitialAutoIncrementValue() != null
               column.getDefaultValue(),
               column.getDefaultFunction());
         }
       }
       break;
     default:
       throw new IllegalStateException("Unexpected node type: " + modNode);
   }
 }