/* uDig - User Friendly Desktop Internet GIS client
* http://udig.refractions.net
* (C) 2011, Refractions Research Inc.
*
* All rights reserved. This program and the accompanying materials
* are made available under the terms of the Eclipse Public License v1.0
* (http://www.eclipse.org/legal/epl-v10.html), and the Refractions BSD
* License v1.0 (http://udig.refractions.net/files/bsd3-v10.html).
*/
package org.locationtech.udig.catalog.teradata;
import static org.geotools.data.teradata.TeradataDataStoreFactory.DBTYPE;
import static org.geotools.data.teradata.TeradataDataStoreFactory.PORT;
import static org.geotools.jdbc.JDBCDataStoreFactory.DATABASE;
import static org.geotools.jdbc.JDBCDataStoreFactory.HOST;
import static org.geotools.jdbc.JDBCDataStoreFactory.PASSWD;
import static org.geotools.jdbc.JDBCDataStoreFactory.USER;
import java.io.Serializable;
import java.lang.reflect.InvocationTargetException;
import java.sql.Connection;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Statement;
import java.util.HashMap;
import java.util.Map;
import java.util.Set;
import java.util.TreeSet;
import org.locationtech.udig.catalog.service.database.DatabaseConnectionRunnable;
import org.locationtech.udig.catalog.service.database.DatabaseWizardLocalization;
import org.apache.commons.dbcp.BasicDataSource;
import org.eclipse.core.runtime.IProgressMonitor;
public class TeradataDatabaseConnectionRunnable implements
DatabaseConnectionRunnable {
private String host;
private int port;
private String username;
private String password;
private DatabaseWizardLocalization localization;
private boolean ran = false;
private String result;
private Set<String> databaseNames = new TreeSet<String>();
public TeradataDatabaseConnectionRunnable(String host, int port,
String username, String password,
DatabaseWizardLocalization localization) {
this.host = host;
this.port = port;
this.username = username;
this.password = password;
this.localization = localization;
}
public void run(IProgressMonitor monitor) throws InvocationTargetException,
InterruptedException {
try {
Map<String, Serializable> params = new HashMap<String, Serializable>();
params.put(DBTYPE.key, (Serializable) new TeradataDialect().dbType);
params.put(HOST.key, host);
params.put(PORT.key, port);
params.put(USER.key, username);
params.put(PASSWD.key, password);
params.put(DATABASE.key, "dbc");
BasicDataSource source = TeradataServiceExtension.getFactory()
.createDataSource(params);
Connection connection = source.getConnection();
try {
Statement statement = connection.createStatement();
if (statement
.execute("SELECT F_TABLE_SCHEMA FROM SYSSPATIAL.GEOMETRY_COLUMNS")) {
ResultSet resultSet = statement.getResultSet();
while (resultSet.next()) {
databaseNames.add(resultSet.getString(1).trim());
}
}
statement.close();
} finally {
if (connection != null) {
connection.close();
}
if (source != null) {
source.close();
}
}
} catch (SQLException e) {
checkSqlException(e);
} catch (Exception e) {
if (e.getCause() instanceof SQLException) {
checkSqlException((SQLException) e.getCause());
} else {
Activator.log("Error connecting to datasource", e);
result = "Unrecognized connection failure. Check parameters and database.";
}
}
ran = true;
}
private void checkSqlException(SQLException e) {
if (e.getMessage().contains("[Error 3807]")) {
result = "SYSSPATIAL.GEOMETRY_COLUMNS is either inaccessible or does not exist. The table must both exist and be accessible to current.";
} else if (e.getMessage().contains("[Error 8017]")) { //$NON-NLS-1$//$NON-NLS-2$
// this is understandable the template1 database is not accessible
// to this user/location so it is not an error
result = "Username or password is incorrect";
} else {
Activator.log("Error connecting to database dbc", e);
result = "Unrecognized connection failure. Check parameters and database.";
}
}
public String canConnect() throws IllegalStateException {
if (!ran) {
throw new IllegalStateException(
"run must complete running before this method is called.");
}
return result;
}
public String[] getDatabaseNames() {
if (!ran) {
throw new IllegalStateException(
"run must complete running before this method is called.");
}
return databaseNames.toArray(new String[0]);
}
}