Java tutorial
/***************************************************************** * 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.modeler.dialog.pref; import java.awt.Component; import java.io.File; import java.sql.Connection; import java.sql.Driver; import java.sql.SQLException; import java.util.ArrayList; import java.util.Arrays; import java.util.Collection; import java.util.Iterator; import java.util.List; import java.util.Map; import java.util.StringTokenizer; import java.util.prefs.Preferences; import javax.swing.DefaultComboBoxModel; import javax.swing.JOptionPane; import org.apache.cayenne.datasource.DriverDataSource; import org.apache.cayenne.map.event.MapEvent; import org.apache.cayenne.modeler.FileClassLoadingService; import org.apache.cayenne.modeler.ProjectController; import org.apache.cayenne.modeler.event.DataSourceModificationEvent; import org.apache.cayenne.modeler.pref.DBConnectionInfo; import org.apache.cayenne.modeler.util.CayenneController; import org.apache.cayenne.pref.CayennePreferenceEditor; import org.apache.cayenne.pref.ChildrenMapPreference; import org.apache.cayenne.pref.PreferenceEditor; import org.apache.cayenne.swing.BindingBuilder; import org.apache.cayenne.util.Util; import org.apache.commons.collections.CollectionUtils; import org.apache.commons.collections.Transformer; /** * Editor for the local DataSources configured in preferences. * */ public class DataSourcePreferences extends CayenneController { protected DataSourcePreferencesView view; protected String dataSourceKey; protected Map dataSources; protected ChildrenMapPreference dataSourcePreferences; protected CayennePreferenceEditor editor; public DataSourcePreferences(PreferenceDialog parentController) { super(parentController); this.view = new DataSourcePreferencesView(this); PreferenceEditor editor = parentController.getEditor(); if (editor instanceof CayennePreferenceEditor) { this.editor = (CayennePreferenceEditor) editor; } // init view data this.dataSourcePreferences = getApplication().getCayenneProjectPreferences() .getDetailObject(DBConnectionInfo.class); this.dataSources = dataSourcePreferences.getChildrenPreferences(); Object[] keys = dataSources.keySet().toArray(); Arrays.sort(keys); DefaultComboBoxModel dataSourceModel = new DefaultComboBoxModel(keys); view.getDataSources().setModel(dataSourceModel); initBindings(); // show first item if (keys.length > 0) { view.getDataSources().setSelectedIndex(0); editDataSourceAction(); } } public Component getView() { return view; } protected void initBindings() { BindingBuilder builder = new BindingBuilder(getApplication().getBindingFactory(), this); builder.bindToAction(view.getAddDataSource(), "newDataSourceAction()"); builder.bindToAction(view.getDuplicateDataSource(), "duplicateDataSourceAction()"); builder.bindToAction(view.getRemoveDataSource(), "removeDataSourceAction()"); builder.bindToAction(view.getTestDataSource(), "testDataSourceAction()"); builder.bindToComboSelection(view.getDataSources(), "dataSourceKey"); } public Map getDataSources() { return dataSources; } public String getDataSourceKey() { return dataSourceKey; } public void setDataSourceKey(String dataSourceKey) { this.dataSourceKey = dataSourceKey; editDataSourceAction(); } public DBConnectionInfo getConnectionInfo() { return (DBConnectionInfo) dataSourcePreferences.getObject(dataSourceKey); } /** * Shows a dialog to create new local DataSource configuration. */ public void newDataSourceAction() { DataSourceCreator creatorWizard = new DataSourceCreator(this); DBConnectionInfo dataSource = creatorWizard.startupAction(); if (dataSource != null) { dataSourcePreferences.create(creatorWizard.getName(), dataSource); dataSources = dataSourcePreferences.getChildrenPreferences(); Object[] keys = dataSources.keySet().toArray(); Arrays.sort(keys); view.getDataSources().setModel(new DefaultComboBoxModel(keys)); view.getDataSources().setSelectedItem(creatorWizard.getName()); editDataSourceAction(); fireEvent(creatorWizard.getName(), MapEvent.ADD); } } /** * Shows a dialog to duplicate an existing local DataSource configuration. */ public void duplicateDataSourceAction() { Object selected = view.getDataSources().getSelectedItem(); if (selected != null) { DataSourceDuplicator wizard = new DataSourceDuplicator(this, selected.toString()); DBConnectionInfo dataSource = wizard.startupAction(); if (dataSource != null) { dataSourcePreferences.create(wizard.getName(), dataSource); dataSources = dataSourcePreferences.getChildrenPreferences(); Object[] keys = dataSources.keySet().toArray(); Arrays.sort(keys); view.getDataSources().setModel(new DefaultComboBoxModel(keys)); view.getDataSources().setSelectedItem(wizard.getName()); editDataSourceAction(); fireEvent(wizard.getName(), MapEvent.ADD); } } } /** * Removes current DataSource. */ public void removeDataSourceAction() { String key = getDataSourceKey(); if (key != null) { dataSourcePreferences.remove(key); dataSources = dataSourcePreferences.getChildrenPreferences(); Object[] keys = dataSources.keySet().toArray(); Arrays.sort(keys); view.getDataSources().setModel(new DefaultComboBoxModel(keys)); editDataSourceAction(keys.length > 0 ? keys[0] : null); fireEvent(key, MapEvent.REMOVE); } } private void fireEvent(String dataSourceKey, int eventId) { DataSourceModificationEvent event = new DataSourceModificationEvent(this, dataSourceKey, eventId); getApplication().getFrameController().getProjectController().fireDataSourceModificationEvent(event); } /** * Opens specified DataSource in the editor. */ public void editDataSourceAction(Object dataSourceKey) { view.getDataSources().setSelectedItem(dataSourceKey); editDataSourceAction(); } /** * Opens current DataSource in the editor. */ public void editDataSourceAction() { this.view.getDataSourceEditor().setConnectionInfo(getConnectionInfo()); } /** * Tries to establish a DB connection, reporting the status of this * operation. */ public void testDataSourceAction() { DBConnectionInfo currentDataSource = getConnectionInfo(); if (currentDataSource == null) { return; } if (currentDataSource.getJdbcDriver() == null) { JOptionPane.showMessageDialog(null, "No JDBC Driver specified", "Warning", JOptionPane.WARNING_MESSAGE); return; } if (currentDataSource.getUrl() == null) { JOptionPane.showMessageDialog(null, "No Database URL specified", "Warning", JOptionPane.WARNING_MESSAGE); return; } try { FileClassLoadingService classLoader = new FileClassLoadingService(); List<File> oldPathFiles = ((FileClassLoadingService) getApplication().getClassLoadingService()) .getPathFiles(); Collection<String> details = new ArrayList<>(); for (File oldPathFile : oldPathFiles) { details.add(oldPathFile.getAbsolutePath()); } Preferences classPathPreferences = getApplication().getPreferencesNode(ClasspathPreferences.class, ""); if (editor.getChangedPreferences().containsKey(classPathPreferences)) { Map<String, String> map = editor.getChangedPreferences().get(classPathPreferences); for (Map.Entry<String, String> en : map.entrySet()) { String key = en.getKey(); if (!details.contains(key)) { details.add(key); } } } if (editor.getRemovedPreferences().containsKey(classPathPreferences)) { Map<String, String> map = editor.getRemovedPreferences().get(classPathPreferences); for (Map.Entry<String, String> en : map.entrySet()) { String key = en.getKey(); if (details.contains(key)) { details.remove(key); } } } if (details.size() > 0) { // transform preference to file... Transformer transformer = new Transformer() { public Object transform(Object object) { String pref = (String) object; return new File(pref); } }; classLoader.setPathFiles(CollectionUtils.collect(details, transformer)); } Class<Driver> driverClass = classLoader.loadClass(Driver.class, currentDataSource.getJdbcDriver()); Driver driver = driverClass.newInstance(); // connect via Cayenne DriverDataSource - it addresses some driver // issues... // can't use try with resource here as we can loose meaningful exception Connection c = new DriverDataSource(driver, currentDataSource.getUrl(), currentDataSource.getUserName(), currentDataSource.getPassword()).getConnection(); try { c.close(); } catch (SQLException ignored) { // i guess we can ignore this... } JOptionPane.showMessageDialog(null, "Connected Successfully", "Success", JOptionPane.INFORMATION_MESSAGE); } catch (Throwable th) { th = Util.unwindException(th); String message = "Error connecting to DB: " + th.getLocalizedMessage(); StringTokenizer st = new StringTokenizer(message); StringBuilder sbMessage = new StringBuilder(); int len = 0; String tempString; while (st.hasMoreTokens()) { tempString = st.nextElement().toString(); if (len < 110) { len = len + tempString.length() + 1; } else { sbMessage.append("\n"); len = 0; } sbMessage.append(tempString).append(" "); } JOptionPane.showMessageDialog(null, sbMessage.toString(), "Warning", JOptionPane.WARNING_MESSAGE); } } }