/*****************************************************************
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* KIND, either express or implied. See the License for the
* specific language governing permissions and limitations
* under the License.
****************************************************************/
package org.apache.cayenne.merge;
import java.util.Collections;
import java.util.List;
import org.apache.cayenne.CayenneRuntimeException;
import org.apache.cayenne.dba.DbAdapter;
import org.apache.cayenne.dba.QuotingStrategy;
import org.apache.cayenne.dba.TypesMapping;
import org.apache.cayenne.map.DbAttribute;
import org.apache.cayenne.map.DbEntity;
/**
* An {@link MergerToken} to use to set type, length and precision.
*/
public class SetColumnTypeToDb extends AbstractToDbToken.Entity {
private DbAttribute columnOriginal;
private DbAttribute columnNew;
public SetColumnTypeToDb(DbEntity entity, DbAttribute columnOriginal, DbAttribute columnNew) {
super(entity);
this.columnOriginal = columnOriginal;
this.columnNew = columnNew;
}
/**
* append the part of the token before the actual column data type
* @param context
*/
protected void appendPrefix(StringBuffer sqlBuffer, QuotingStrategy context) {
sqlBuffer.append("ALTER TABLE ");
sqlBuffer.append(context.quoteFullyQualifiedName(getEntity()));
sqlBuffer.append(" ALTER ");
sqlBuffer.append(context.quoteString(columnNew.getName()));
sqlBuffer.append(" TYPE ");
}
@Override
public List<String> createSql(DbAdapter adapter) {
StringBuffer sqlBuffer = new StringBuffer();
QuotingStrategy context = adapter.getQuotingStrategy(getEntity()
.getDataMap()
.isQuotingSQLIdentifiers());
appendPrefix(sqlBuffer, context);
// copied from JdbcAdapter.createTableAppendColumn
String[] types = adapter.externalTypesForJdbcType(columnNew.getType());
if (types == null || types.length == 0) {
String entityName = columnNew.getEntity() != null ? ((DbEntity) columnNew
.getEntity()).getFullyQualifiedName() : "<null>";
throw new CayenneRuntimeException("Undefined type for attribute '"
+ entityName
+ "."
+ columnNew.getName()
+ "': "
+ columnNew.getType());
}
String type = types[0];
sqlBuffer.append(type);
// append size and precision (if applicable)
if (TypesMapping.supportsLength(columnNew.getType())) {
int len = columnNew.getMaxLength();
int scale = TypesMapping.isDecimal(columnNew.getType()) ? columnNew.getScale() : -1;
// sanity check
if (scale > len) {
scale = -1;
}
if (len > 0) {
sqlBuffer.append('(').append(len);
if (scale >= 0) {
sqlBuffer.append(", ").append(scale);
}
sqlBuffer.append(')');
}
}
return Collections.singletonList(sqlBuffer.toString());
}
public String getTokenName() {
return "Set Column Type";
}
@Override
public String getTokenValue() {
StringBuilder sb = new StringBuilder();
sb.append(getEntity().getName());
sb.append(".");
sb.append(columnNew.getName());
if (columnOriginal.getType() != columnNew.getType()) {
sb.append(" type: ");
sb.append(TypesMapping.getSqlNameByType(columnOriginal.getType()));
sb.append(" -> ");
sb.append(TypesMapping.getSqlNameByType(columnNew.getType()));
}
if (columnOriginal.getMaxLength() != columnNew.getMaxLength()) {
sb.append(" maxLength: ");
sb.append(columnOriginal.getMaxLength());
sb.append(" -> ");
sb.append(columnNew.getMaxLength());
}
if (columnOriginal.getAttributePrecision() != columnNew.getAttributePrecision()) {
sb.append(" precision: ");
sb.append(columnOriginal.getAttributePrecision());
sb.append(" -> ");
sb.append(columnNew.getAttributePrecision());
}
if (columnOriginal.getScale() != columnNew.getScale()) {
sb.append(" scale: ");
sb.append(columnOriginal.getScale());
sb.append(" -> ");
sb.append(columnNew.getScale());
}
return sb.toString();
}
public MergerToken createReverse(MergerFactory factory) {
return factory.createSetColumnTypeToModel(getEntity(), columnNew, columnOriginal);
}
public DbAttribute getColumnOriginal() {
return columnOriginal;
}
public DbAttribute getColumnNew() {
return columnNew;
}
}