/*****************************************************************
* 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.dbsync.merge.factory;
import org.apache.cayenne.dba.DbAdapter;
import org.apache.cayenne.dba.QuotingStrategy;
import org.apache.cayenne.dbsync.merge.token.MergerToken;
import org.apache.cayenne.dbsync.merge.token.db.DropRelationshipToDb;
import org.apache.cayenne.dbsync.merge.token.db.SetAllowNullToDb;
import org.apache.cayenne.dbsync.merge.token.db.SetColumnTypeToDb;
import org.apache.cayenne.dbsync.merge.token.db.SetGeneratedFlagToDb;
import org.apache.cayenne.dbsync.merge.token.db.SetNotNullToDb;
import org.apache.cayenne.dbsync.merge.token.db.SetPrimaryKeyToDb;
import org.apache.cayenne.map.DbAttribute;
import org.apache.cayenne.map.DbEntity;
import org.apache.cayenne.map.DbRelationship;
import java.util.Collection;
import java.util.Collections;
import java.util.List;
public class MySQLMergerTokenFactory extends DefaultMergerTokenFactory {
@Override
public MergerToken createSetNotNullToDb(
final DbEntity entity,
final DbAttribute column) {
return new SetNotNullToDb(entity, column) {
@Override
public List<String> createSql(DbAdapter adapter) {
StringBuffer sqlBuffer = new StringBuffer();
QuotingStrategy context = adapter.getQuotingStrategy();
sqlBuffer.append("ALTER TABLE ");
sqlBuffer.append(context.quotedFullyQualifiedName(getEntity()));
sqlBuffer.append(" CHANGE ");
sqlBuffer.append(context.quotedName(getColumn()));
sqlBuffer.append(" ");
adapter.createTableAppendColumn(sqlBuffer, column);
return Collections.singletonList(sqlBuffer.toString());
}
};
}
@Override
public MergerToken createSetAllowNullToDb(
final DbEntity entity,
final DbAttribute column) {
return new SetAllowNullToDb(entity, column) {
@Override
public List<String> createSql(DbAdapter adapter) {
StringBuffer sqlBuffer = new StringBuffer();
QuotingStrategy context = adapter.getQuotingStrategy();
sqlBuffer.append("ALTER TABLE ");
sqlBuffer.append(context.quotedFullyQualifiedName(getEntity()));
sqlBuffer.append(" CHANGE ");
sqlBuffer.append(context.quotedName(getColumn()));
sqlBuffer.append(" ");
adapter.createTableAppendColumn(sqlBuffer, column);
return Collections.singletonList(sqlBuffer.toString());
}
};
}
@Override
public MergerToken createSetColumnTypeToDb(
final DbEntity entity,
DbAttribute columnOriginal,
final DbAttribute columnNew) {
return new SetColumnTypeToDb(entity, columnOriginal, columnNew) {
@Override
protected void appendPrefix(StringBuffer sqlBuffer, QuotingStrategy context) {
// http://dev.mysql.com/tech-resources/articles/mysql-cluster-50.html
sqlBuffer.append("ALTER TABLE ");
sqlBuffer.append(context.quotedFullyQualifiedName(entity));
sqlBuffer.append(" MODIFY ");
sqlBuffer.append(context.quotedName(columnNew));
sqlBuffer.append(" ");
}
};
}
@Override
public MergerToken createDropRelationshipToDb(
final DbEntity entity,
DbRelationship rel) {
return new DropRelationshipToDb(entity, rel) {
@Override
public List<String> createSql(DbAdapter adapter) {
String fkName = getFkName();
if (fkName == null) {
return Collections.emptyList();
}
QuotingStrategy context = adapter.getQuotingStrategy();
// http://dev.mysql.com/tech-resources/articles/mysql-cluster-50.html
return Collections.singletonList("ALTER TABLE " + context.quotedFullyQualifiedName(entity) + " DROP FOREIGN KEY " + fkName);
}
};
}
@Override
public MergerToken createSetPrimaryKeyToDb(
DbEntity entity,
Collection<DbAttribute> primaryKeyOriginal,
Collection<DbAttribute> primaryKeyNew,
String detectedPrimaryKeyName) {
return new SetPrimaryKeyToDb(
entity,
primaryKeyOriginal,
primaryKeyNew,
detectedPrimaryKeyName) {
@Override
protected void appendDropOriginalPrimaryKeySQL(
DbAdapter adapter,
List<String> sqls) {
sqls.add("ALTER TABLE "
+ adapter.getQuotingStrategy()
.quotedFullyQualifiedName(getEntity())
+ " DROP PRIMARY KEY");
}
};
}
@Override
public MergerToken createSetGeneratedFlagToDb(DbEntity entity, DbAttribute column, boolean isGenerated) {
return new SetGeneratedFlagToDb(entity, column, isGenerated) {
protected void appendAlterColumnClause(DbAdapter adapter, StringBuffer builder) {
builder.append(" MODIFY ");
}
@Override
protected void appendAutoIncrement(DbAdapter adapter, StringBuffer builder) {
adapter.createTableAppendColumn(builder, this.getColumn());
}
/**
* To drop AUTO_INCREMENT flag update column with all information but w/o AUTO_INCREMENT
*/
@Override
protected void appendDropAutoIncrement(DbAdapter adapter, StringBuffer builder) {
adapter.createTableAppendColumn(builder, this.getColumn());
}
@Override
public boolean isEmpty() {
return false;
}
};
}
}