/*****************************************************************
* 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.token.db;
import org.apache.cayenne.access.DataNode;
import org.apache.cayenne.dba.DbAdapter;
import org.apache.cayenne.dbsync.merge.context.MergerContext;
import org.apache.cayenne.dbsync.merge.factory.MergerTokenFactory;
import org.apache.cayenne.dbsync.merge.token.MergerToken;
import org.apache.cayenne.map.DbAttribute;
import org.apache.cayenne.map.DbEntity;
import org.apache.cayenne.validation.SimpleValidationFailure;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
public class CreateTableToDb extends AbstractToDbToken.Entity {
public CreateTableToDb(DbEntity entity) {
super("Create Table", 40, entity);
}
@Override
public List<String> createSql(DbAdapter adapter) {
List<String> sqls = new ArrayList<>();
if(needAutoPkSupport()) {
sqls.addAll(adapter.getPkGenerator().createAutoPkStatements(
Collections.singletonList(getEntity())));
}
sqls.add(adapter.createTable(getEntity()));
return sqls;
}
@Override
public void execute(MergerContext mergerContext) {
try {
DataNode node = mergerContext.getDataNode();
DbAdapter adapter = node.getAdapter();
if(needAutoPkSupport()) {
adapter.getPkGenerator().createAutoPk(
node,
Collections.singletonList(getEntity()));
}
executeSql(mergerContext, adapter.createTable(getEntity()));
}
catch (Exception e) {
mergerContext.getValidationResult().addFailure(
new SimpleValidationFailure(this, e.getMessage()));
}
}
private boolean needAutoPkSupport() {
DbEntity entity = getEntity();
if(entity.getPrimaryKeyGenerator() != null) {
return false;
}
for(DbAttribute attribute : entity.getPrimaryKeys()) {
if(attribute.isGenerated()) {
return false;
}
}
return true;
}
public MergerToken createReverse(MergerTokenFactory factory) {
return factory.createDropTableToModel(getEntity());
}
}