use of org.talend.sqlbuilder.erdiagram.ui.nodes.Table in project tdi-studio-se by Talend.
the class ErDiagramComposite method clearAll.
public void clearAll() {
List<Table> tables = new ArrayList<Table>();
tables.addAll(erDiagram.getTables());
for (Table table : tables) {
erDiagram.removeTable(table);
}
//$NON-NLS-1$
this.sqlText.setText("");
}
use of org.talend.sqlbuilder.erdiagram.ui.nodes.Table in project tdi-studio-se by Talend.
the class ErDiagramComposite method getSqlStatement.
//$NON-NLS-1$
@SuppressWarnings("unchecked")
private String getSqlStatement() {
//$NON-NLS-1$
String sql = "";
List<String> tables = new ArrayList<String>();
List<String> columns = new ArrayList<String>();
List<String> wheres = new ArrayList<String>();
if (editor != null) {
if (editor.getViewer().getContents() instanceof ErDiagramPart) {
//$NON-NLS-1$ //$NON-NLS-2$ //$NON-NLS-3$
String schemaPrefix = "".equals(getSchema()) ? "" : getSchema() + ".";
//$NON-NLS-1$ //$NON-NLS-2$ //$NON-NLS-3$
String schemaPrefixWithDoubleQuotes = "".equals(getSchema()) ? "" : "\"" + getSchema() + "\".";
ErDiagramPart er = (ErDiagramPart) editor.getViewer().getContents();
for (Object object : er.getChildren()) {
if (object instanceof TablePart) {
TablePart tablePart = (TablePart) object;
Table table = (Table) tablePart.getModel();
if (TextUtil.isDoubleQuotesNeededDbType(getCurrentDbType())) {
//$NON-NLS-1$
tables.add(schemaPrefixWithDoubleQuotes + "\"" + table.getElementName() + "\"");
} else {
tables.add(schemaPrefix + TalendTextUtils.addQuotesWithSpaceField(table.getElementName(), getCurrentDbType()));
}
boolean oracleDbType = TextUtil.isOracleDbType(getCurrentDbType());
for (Object obj : tablePart.getChildren()) {
if (obj instanceof ColumnPart) {
ColumnPart columnPart = (ColumnPart) obj;
Column column = (Column) columnPart.getModel();
CheckBox isSelected = columnPart.getPrimativeFigure().getFigureCustomColumnIsSelectedFigure();
if (isSelected != null && isSelected.isSelected() && !column.getElementName().equals("*")) {
//$NON-NLS-1$
if (TextUtil.isDoubleQuotesNeededDbType(getCurrentDbType())) {
//$NON-NLS-1$
columns.add(schemaPrefixWithDoubleQuotes + "\"" + table.getElementName() + "\".\"" + column.getElementName() + //$NON-NLS-1$ //$NON-NLS-2$ //$NON-NLS-3$
"\"");
} else {
// added by hyWang
String leftQuote = TalendTextUtils.getQuoteByDBType(getCurrentDbType(), true);
String rightQuote = TalendTextUtils.getQuoteByDBType(getCurrentDbType(), false);
String columnContent = column.getElementName();
//$NON-NLS-1$
Pattern pattern = Pattern.compile("\\w+");
Matcher matcher = pattern.matcher(columnContent);
EDatabaseTypeName dbType = EDatabaseTypeName.getTypeFromDbType(getCurrentDbType());
// modify for bug 12092
boolean sqlKeyword = KeywordsValidator.isSqlKeyword(column.getElementName(), dbType.getProduct());
if (!matcher.matches() || (sqlKeyword && oracleDbType)) {
columns.add(schemaPrefix + TalendTextUtils.addQuotesWithSpaceField(table.getElementName(), getCurrentDbType()) + //$NON-NLS-1$
"." + TalendTextUtils.addQuotesWithSpaceField(leftQuote + column.getElementName() + rightQuote, getCurrentDbType()));
} else {
columns.add(schemaPrefix + TalendTextUtils.addQuotesWithSpaceField(table.getElementName(), getCurrentDbType()) + //$NON-NLS-1$
"." + TalendTextUtils.addQuotesWithSpaceField(column.getElementName(), getCurrentDbType()));
}
}
}
for (Relation rel : (List<Relation>) column.getOutputs()) {
Column source = rel.getSource();
Column target = rel.getTarget();
if (TextUtil.isDoubleQuotesNeededDbType(getCurrentDbType())) {
String where1 = schemaPrefixWithDoubleQuotes + "\"" + source.getTable().getElementName() + //$NON-NLS-1$ //$NON-NLS-2$
"\".\"" + source.getElementName() + "\"= " + schemaPrefixWithDoubleQuotes + "\"" + //$NON-NLS-1$
target.getTable().getElementName() + "\".\"" + target.getElementName() + //$NON-NLS-1$ //$NON-NLS-2$
"\"";
if (!wheres.contains(where1)) {
wheres.add(where1);
}
} else {
String where1 = schemaPrefix + TalendTextUtils.addQuotesWithSpaceField(source.getTable().getElementName(), getCurrentDbType()) + //$NON-NLS-1$
"." + TalendTextUtils.addQuotesWithSpaceField(source.getElementName(), getCurrentDbType()) + //$NON-NLS-1$
"=" + schemaPrefix + TalendTextUtils.addQuotesWithSpaceField(target.getTable().getElementName(), getCurrentDbType()) + //$NON-NLS-1$
"." + TalendTextUtils.addQuotesWithSpaceField(target.getElementName(), getCurrentDbType());
if (!wheres.contains(where1)) {
wheres.add(where1);
}
}
}
}
}
}
}
}
}
// Mssql query need add catalog and schema before the table, like this "catolog.schema.table"
Connection conn = null;
if (rootNode != null) {
Item connectionItem = rootNode.getObject().getProperty().getItem();
if (connectionItem instanceof ConnectionItem) {
conn = ((ConnectionItem) connectionItem).getConnection();
}
}
if (getCurrentDbType() != null && (getCurrentDbType().equals(EDatabaseTypeName.MSSQL.getDisplayName()) || getCurrentDbType().equals(EDatabaseTypeName.MSSQL.name())) && conn != null) {
List<String> newTables = new ArrayList<String>();
for (String str : tables) {
newTables.add(getMssqlCatalog(str, conn));
}
tables = newTables;
}
sql = getSelectStatement(tables, columns, wheres);
if (sql.endsWith(",")) {
//$NON-NLS-1$
return sql.substring(0, sql.length() - 1);
} else if (sql.endsWith(" and ")) {
//$NON-NLS-1$
return sql.substring(0, sql.length() - 5);
}
//$NON-NLS-1$
return "";
}
use of org.talend.sqlbuilder.erdiagram.ui.nodes.Table in project tdi-studio-se by Talend.
the class ErDiagramComposite method createErDiagram.
/**
* admin Comment method "createErDiagram".
*
* @return
*/
private ErDiagram createErDiagram(boolean isShowDesignerPage) {
erDiagram = new ErDiagram();
erDiagram.setErDiagramComposite(this);
if (isShowDesignerPage) {
IRunnableWithProgress progress = new IRunnableWithProgress() {
public void run(IProgressMonitor monitor) throws InvocationTargetException, InterruptedException {
//$NON-NLS-1$
monitor.beginTask("", IProgressMonitor.UNKNOWN);
try {
List<MetadataColumn> selectedColumns = new ArrayList<MetadataColumn>();
List<MetadataTable> tables = EMFRepositoryNodeManager.getInstance().getTables(getNodes(), selectedColumns);
erDiagram.setMetadataTables(tables);
List<String[]> fks = EMFRepositoryNodeManager.getInstance().getPKFromTables(tables, erDiagram.getErDiagramComposite().getDialog().getSelectedContext());
for (MetadataTable metadataTable : tables) {
Table table = new Table();
table.setMetadataTable(metadataTable, selectedColumns);
table.setErDiagram(erDiagram);
erDiagram.addTable(table);
}
erDiagram.setRelations(fks);
} finally {
monitor.done();
}
}
};
UIUtils.runWithProgress(progress, true, dialog.getProgressMonitor(), dialog.getShell());
} else {
erDiagram.setMetadataTables(new ArrayList<MetadataTable>());
erDiagram.setRelations(new ArrayList<String[]>());
}
return erDiagram;
}
use of org.talend.sqlbuilder.erdiagram.ui.nodes.Table in project tdi-studio-se by Talend.
the class TableEditPolicy method createDeleteCommand.
@Override
protected Command createDeleteCommand(GroupRequest deleteRequest) {
Object parent = getHost().getParent().getModel();
if (!(parent instanceof ErDiagram)) {
return null;
}
List<Table> nodeList = new ArrayList<Table>();
for (int i = 0; i < deleteRequest.getEditParts().size(); i++) {
if (deleteRequest.getEditParts().get(i) instanceof TablePart) {
nodeList.add((Table) ((TablePart) deleteRequest.getEditParts().get(i)).getModel());
}
}
DeleteTableCommand command = new DeleteTableCommand((ErDiagram) parent, nodeList);
return command;
}
use of org.talend.sqlbuilder.erdiagram.ui.nodes.Table in project tdi-studio-se by Talend.
the class TablePart method refreshVisuals.
@Override
protected void refreshVisuals() {
Table table = (Table) getModel();
Point loc = table.getLocation();
Dimension size = table.getSize();
Rectangle rectangle = new Rectangle(loc, size);
((GraphicalEditPart) getParent()).setLayoutConstraint(this, getFigure(), rectangle);
}
Aggregations