/*****************************************************************
* 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.model;
import java.util.Collection;
import java.util.Iterator;
import org.apache.cayenne.dbsync.merge.context.MergeDirection;
import org.apache.cayenne.dbsync.merge.token.AbstractMergerToken;
import org.apache.cayenne.dbsync.merge.token.MergerToken;
import org.apache.cayenne.dbsync.reverse.dbload.ModelMergeDelegate;
import org.apache.cayenne.map.DbAttribute;
import org.apache.cayenne.map.DbEntity;
import org.apache.cayenne.map.DbRelationship;
import org.apache.cayenne.map.ObjEntity;
import org.apache.cayenne.map.ObjRelationship;
/**
* Common abstract superclass for all {@link MergerToken}s going from the database to the
* model.
*/
public abstract class AbstractToModelToken extends AbstractMergerToken {
protected AbstractToModelToken(String tokenName, int sortingWeight) {
super(tokenName, sortingWeight);
}
protected static void remove(ModelMergeDelegate mergerContext, DbRelationship rel, boolean reverse) {
if (rel == null) {
return;
}
if (reverse) {
remove(mergerContext, rel.getReverseRelationship(), false);
}
DbEntity dbEntity = rel.getSourceEntity();
for (ObjEntity objEntity : dbEntity.mappedObjEntities()) {
remove(mergerContext, objEntity.getRelationshipForDbRelationship(rel), true);
}
rel.getSourceEntity().removeRelationship(rel.getName());
mergerContext.dbRelationshipRemoved(rel);
}
protected static void remove(ModelMergeDelegate mergerContext, ObjRelationship rel, boolean reverse) {
if (rel == null) {
return;
}
if (reverse) {
remove(mergerContext, rel.getReverseRelationship(), false);
}
rel.getSourceEntity().removeRelationship(rel.getName());
mergerContext.objRelationshipRemoved(rel);
}
@Override
public final MergeDirection getDirection() {
return MergeDirection.TO_MODEL;
}
abstract static class Entity extends AbstractToModelToken {
private final DbEntity entity;
protected Entity(String tokenName, int sortingWeight, DbEntity entity) {
super(tokenName, sortingWeight);
this.entity = entity;
}
public DbEntity getEntity() {
return entity;
}
/**
* @return ObjEntities mapped to current DbEntity excluding inherited
*/
public Collection<ObjEntity> getMappedObjEntities() {
Collection<ObjEntity> entities = entity.mappedObjEntities();
Iterator<ObjEntity> iterator = entities.iterator();
while(iterator.hasNext()) {
if(iterator.next().getSuperEntity() != null) {
iterator.remove();
}
}
return entities;
}
public String getTokenValue() {
return getEntity().getName();
}
}
abstract static class EntityAndColumn extends Entity {
private final DbAttribute column;
protected EntityAndColumn(String tokenName, int sortingWeight, DbEntity entity, DbAttribute column) {
super(tokenName, sortingWeight, entity);
this.column = column;
}
public DbAttribute getColumn() {
return column;
}
@Override
public String getTokenValue() {
return getEntity().getName() + "." + getColumn().getName();
}
}
}