@Override
public void run() {
try {
CompoundCommand commands = new CompoundCommand();
List models = getViewer().getContents().getChildren();
NodeList graphNodes = new NodeList();
EdgeList graphEdges = new EdgeList();
// nodes
for (int i = 0; i < models.size(); i++) {
Object obj = models.get(i);
if (obj instanceof TableEditPart) {
TableEditPart editPart = (TableEditPart) obj;
Table model = (Table) editPart.getModel();
EntityNode node = new EntityNode();
node.model = model;
node.width = editPart.getFigure().getSize().width;
node.height = editPart.getFigure().getSize().height + 40;
graphNodes.add(node);
}
}
// edge
for (int i = 0; i < models.size(); i++) {
Object obj = models.get(i);
if (obj instanceof TableEditPart) {
TableEditPart tableEditpart = (TableEditPart) obj;
List outgoing = tableEditpart.getSourceConnections();
for (int j = 0; j < outgoing.size(); j++) {
RelationEditPart conn = (RelationEditPart) outgoing.get(j);
EntityNode source =
(EntityNode) getNode(graphNodes, (Table) conn.getSource().getModel());
EntityNode target =
(EntityNode) getNode(graphNodes, (Table) conn.getTarget().getModel());
if (source != null && target != null) {
ConnectionEdge edge = new ConnectionEdge(source, target);
Relation relation = (Relation) conn.getModel();
edge.model = relation.getSource();
graphEdges.add(edge);
}
}
}
}
DirectedGraph graph = new DirectedGraph();
graph.nodes = graphNodes;
graph.edges = graphEdges;
new DirectedGraphLayout().visit(graph);
for (int i = 0; i < graph.nodes.size(); i++) {
EntityNode node = (EntityNode) graph.nodes.getNode(i);
commands.add(new LayoutCommand(node.model, node.x, node.y));
}
getViewer().getEditDomain().getCommandStack().execute(commands);
} catch (Exception e) {
logger.error(Messages.AutoLayoutAction_2, e);
Status errStatus =
new Status(IStatus.ERROR, Activator.PLUGIN_ID, e.getMessage(), e); // $NON-NLS-1$
ExceptionDetailsErrorDialog.openError(
PlatformUI.getWorkbench().getActiveWorkbenchWindow().getShell(),
"Error",
Messages.AutoLayoutAction_3,
errStatus); //$NON-NLS-1$
}
}
/**
* 테이블 관계를 구성합니다.
*
* @param userDB
* @param mapDBTables
* @param db
* @throws Exception
*/
public static void calRelation(
UserDBDAO userDB,
Map<String, Table> mapDBTables,
DB db,
List<ReferencedTableDAO> referenceTableList)
throws Exception {
RdbFactory tadpoleFactory = RdbFactory.eINSTANCE;
// 디비에서 관계 정보를 찾아서 넣어준다.
for (ReferencedTableDAO refTabDAO : referenceTableList) {
Table soTabMod = mapDBTables.get(refTabDAO.getTable_name());
Table tarTabMod = mapDBTables.get(refTabDAO.getReferenced_table_name());
// 소스테이블에 인덱스가 없고, 타겟 테이블이 있으면 추가한다.
if (soTabMod != null && tarTabMod != null) {
// 이미 추가된 relation인지 검사합니다.
boolean isAlrealyAppend = false;
for (Relation relation : soTabMod.getOutgoingLinks()) {
if (relation.getConstraint_name() != null && refTabDAO.getConstraint_name() != null) {
if (relation.getConstraint_name().equals(refTabDAO.getConstraint_name())) {
isAlrealyAppend = true;
break;
}
}
}
for (Relation relation : soTabMod.getIncomingLinks()) {
if (relation.getConstraint_name() != null && refTabDAO.getConstraint_name() != null) {
if (relation.getConstraint_name().equals(refTabDAO.getConstraint_name())) {
isAlrealyAppend = true;
break;
}
}
}
// TODO 현재 자신의 테이블을 키로 가자고 있는 항목은 다음과 같은 이유로 제거 합니다.
// java.lang.RuntimeException: Cycle detected in graph
if (refTabDAO.getTable_name().equals(refTabDAO.getReferenced_table_name())) continue;
// 이미 추가 되어 있는가?
if (isAlrealyAppend) continue;
// 새롭게 추가될 요소 이면.
Relation relation = tadpoleFactory.createRelation();
/* 저장시 아래와 같은 오류가 발생하여 추가한 코드
* 여유를 가지고 디버깅을 해봐야 하는코드
*
* org.eclipse.emf.ecore.resource.Resource$IOWrappedException:
* The object 'com.hangum.tadpole.model.impl.RelationImpl@5e44efa0 (source_kind: ONLY_ONE, target_kind: ONE_MANY, column_name: country_id, referenced_column_name: country_id, bendpoint: [], constraint_name: null)'
* is not contained in a resource.
*/
relation.setDb(db);
relation.setConstraint_name(refTabDAO.getConstraint_name());
relation.setColumn_name(refTabDAO.getColumn_name());
relation.setReferenced_column_name(refTabDAO.getReferenced_column_name());
/*
* 위의 경우의 수를 이용하여 릴레이션을 생성합니다.
*/
Map<String, Column> sourceColumnsMap = new HashMap<String, Column>();
Map<String, Column> targetColumnMap = new HashMap<String, Column>();
for (Column column : soTabMod.getColumns()) sourceColumnsMap.put(column.getField(), column);
for (Column column : tarTabMod.getColumns()) targetColumnMap.put(column.getField(), column);
// source 컬럼 정보
Column col = sourceColumnsMap.get(refTabDAO.getColumn_name());
// target 컬럼 정보
Column colR = targetColumnMap.get(refTabDAO.getReferenced_column_name());
if (logger.isDebugEnabled()) {
if (col == null || colR == null) {
logger.debug(
"###[table index]###############################################################################");
logger.debug(db.getUrl() + ":" + refTabDAO.toString());
logger.debug(
"###[table index]###############################################################################");
}
}
if (col == null || colR == null) continue;
// logger.debug("\t [source ]" + col.getField() + " [key]" + col.getKey());
// logger.debug("\t [target ]" + colR.getField() + " [key]" + colR.getKey());
relation.setSource_kind(calcRelationCol(col, colR));
relation.setTarget_kind(calcRelationCol(colR, col));
// 관계형성
soTabMod.getIncomingLinks().add(relation);
tarTabMod.getOutgoingLinks().add(relation);
relation.setSource(soTabMod);
relation.setTarget(tarTabMod);
} // if(souceModel != null && targetModel != null
} // for
}