Java tutorial
/* * Copyright 2009-2011 Amazon Technologies, Inc. * * Licensed 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://aws.amazon.com/apache2.0 * * This file 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 com.amazonaws.eclipse.datatools.sqltools.tablewizard.simpledb.ui.popup.actions; import java.sql.Connection; import org.eclipse.core.runtime.IProgressMonitor; import org.eclipse.datatools.connectivity.sqm.core.connection.ConnectionInfo; import org.eclipse.datatools.connectivity.sqm.core.containment.ContainmentService; import org.eclipse.datatools.connectivity.sqm.core.internal.ui.explorer.providers.content.virtual.ColumnNode; import org.eclipse.datatools.connectivity.sqm.core.internal.ui.explorer.virtual.IColumnNode; import org.eclipse.datatools.connectivity.sqm.core.rte.ICatalogObject; import org.eclipse.datatools.connectivity.sqm.core.ui.explorer.virtual.IVirtualNode; import org.eclipse.datatools.connectivity.sqm.internal.core.RDBCorePlugin; import org.eclipse.datatools.connectivity.sqm.internal.core.containment.GroupID; import org.eclipse.datatools.connectivity.sqm.internal.core.util.ConnectionUtil; import org.eclipse.datatools.modelbase.sql.schema.Database; import org.eclipse.datatools.modelbase.sql.schema.Schema; import org.eclipse.datatools.modelbase.sql.tables.Table; import org.eclipse.emf.ecore.EObject; import org.eclipse.jface.dialogs.IInputValidator; import org.eclipse.jface.dialogs.InputDialog; import org.eclipse.jface.viewers.IStructuredSelection; import org.eclipse.jface.viewers.SelectionChangedEvent; import org.eclipse.jface.viewers.StructuredSelection; import org.eclipse.jface.viewers.TreePath; import org.eclipse.jface.window.Window; import org.eclipse.swt.widgets.Display; import org.eclipse.ui.actions.WorkspaceModifyOperation; import org.eclipse.ui.navigator.CommonViewer; import com.amazonaws.eclipse.datatools.enablement.simpledb.driver.JdbcConnection; import com.amazonaws.eclipse.datatools.sqltools.tablewizard.simpledb.ui.Messages; import com.amazonaws.eclipse.datatools.sqltools.tablewizard.simpledb.ui.UIPlugin; public class NewAttributeWizardAction extends AbstractEditorClosingAction { private static final String TEXT = Messages.attributeNewMenu; @Override public void selectionChanged(final SelectionChangedEvent event) { super.selectionChanged(event); if (event.getSelection() instanceof IStructuredSelection && ((IStructuredSelection) event.getSelection()).getFirstElement() instanceof EObject) { EObject o = (EObject) ((IStructuredSelection) event.getSelection()).getFirstElement(); ContainmentService containmentService = RDBCorePlugin.getDefault().getContainmentService(); String groupID = containmentService.getGroupId(o); setEnabled((groupID != null) && (groupID.startsWith(GroupID.CORE_PREFIX))); } } @Override public void initialize() { initializeAction(null, null, TEXT, TEXT); } @Override public void run() { if (!this.event.getSelection().isEmpty()) { try { Display display = Display.getCurrent(); Object o = ((IStructuredSelection) this.event.getSelection()).getFirstElement(); Table table = findTable(o); Database db = findDatabase(o); if (table != null && db != null) { if (!closeEditors(table)) { return; } ConnectionInfo info = ConnectionUtil.getConnectionForEObject(db); if (info != null) { InputDialog dlg = new InputDialog(display.getActiveShell(), Messages.CreateNewAttribute, Messages.NewAttributeName, "", new IInputValidator() { //$NON-NLS-1$ public String isValid(final String newText) { return newText != null && newText.trim().length() > 0 ? null : Messages.EmptyAttributeName; } }); if (dlg.open() == Window.OK) { performFinish(info, table.getName(), dlg.getValue(), o); } } } } catch (Exception e) { throw new RuntimeException(e); } } } private Table findTable(final Object o) { if (o instanceof IColumnNode) { ColumnNode node = (ColumnNode) o; Object parent = node.getParent(); if (parent instanceof Table) { Table table = (Table) parent; return table; } } return null; } private Database findDatabase(final Object o) { if (o instanceof IColumnNode) { ColumnNode node = (ColumnNode) o; Object parent = node.getParent(); if (parent instanceof Table) { Table table = (Table) parent; Schema schema = table.getSchema(); Database db = schema.getDatabase() == null ? schema.getCatalog().getDatabase() : schema.getDatabase(); // JDBCColumn o2 = new JDBCColumn(); // o2.setName("FooBarBaz"); // DatabaseDefinition definition = RDBCorePlugin.getDefault().getDatabaseDefinitionRegistry().getDefinition(db); // PredefinedDataTypeDefinition pdtd = definition.getPredefinedDataTypeDefinition("TEXT"); // PredefinedDataType pdt = definition.getPredefinedDataType(pdtd); // o2.setDataType(pdt); // ((JDBCTable) table).getColumns().add(o2); // node.addChildren(o2); // CommonViewer cViewer = (CommonViewer) this.event.getSource(); // // cViewer.getControl().setRedraw(false); // cViewer.add(node, o2); // // cViewer.getControl().setRedraw(true); return db; } } return null; } public boolean performFinish(final ConnectionInfo info, final String domainName, final String attrName, final Object node) { WorkspaceModifyOperation operation = new WorkspaceModifyOperation() { @Override protected void execute(final IProgressMonitor monitor) { try { // SQLDevToolsConfiguration f = SQLToolsFacade.getConfigurationByProfileName(info.getConnectionProfile().getName()); // ConnectionService conService = f.getConnectionService(); // DatabaseVendorDefinitionId dbVendorId = ProfileUtil.getDatabaseVendorDefinitionId(profileName); // DatabaseIdentifier databaseIdentifier = new DatabaseIdentifier(info.getConnectionProfile().getName(), info.getDatabaseName()); // Connection conn = conService.createConnection(databaseIdentifier, true); Connection conn = info.getSharedConnection(); // conn.prepareStatement("create domain " + domainName).executeUpdate(); JdbcConnection jdbcConn = (JdbcConnection) conn; jdbcConn.addPendingColumn(domainName, attrName); CommonViewer cViewer = (CommonViewer) NewAttributeWizardAction.this.event.getSource(); TreePath[] expanded = cViewer.getExpandedTreePaths(); // Object selected = ((IStructuredSelection) NewAttributeWizardAction.this.event.getSelection()) // .getFirstElement(); IVirtualNode vnode = (IVirtualNode) node; ICatalogObject table = (ICatalogObject) vnode.getParent(); table.refresh(); // if (NewAttributeWizardAction.this.event.getSelection() instanceof IStructuredSelection // && ((IStructuredSelection) NewAttributeWizardAction.this.event.getSelection()).getFirstElement() instanceof EObject) { // EObject o = (EObject) ((IStructuredSelection) NewAttributeWizardAction.this.event.getSelection()) // .getFirstElement(); // cViewer.expandToLevel(o, 1); // } // cViewer.refresh(); // Object col = ((JDBCTable) table).getColumns().get(((JDBCTable) table).getColumns().size() - 1); for (TreePath path : expanded) { cViewer.expandToLevel(path, 0); } cViewer.setSelection(new StructuredSelection(table), true); // cViewer.reveal(table); // cViewer.expandToLevel(table, 2); // cViewer.setSelection(new StructuredSelection(table)); // cViewer.reveal(((IStructuredSelection) NewAttributeWizardAction.this.event.getSelection()).getFirstElement()); // cViewer.expandToLevel(((IStructuredSelection) NewAttributeWizardAction.this.event.getSelection()) // .getFirstElement(), 1); // cViewer.setSelection(new StructuredSelection(((IStructuredSelection) NewAttributeWizardAction.this.event // .getSelection()).getFirstElement())); } catch (Exception e) { e.printStackTrace(); } finally { if (monitor != null) { monitor.done(); } } } }; try { UIPlugin.getDefault().getWorkbench().getActiveWorkbenchWindow().run(false, true, operation); } catch (Exception e) { e.printStackTrace(); } return true; } }