jpa/plugins/org.eclipse.jpt.db/src/org/eclipse/jpt/db/internal/DTPConnectionProfileWrapper.java - dali/webtools.dali - Git at Google

 /*******************************************************************************
  * Copyright (c) 2006, 2007 Oracle. All rights reserved.
  * This program and the accompanying materials are made available under the
  * terms of the Eclipse Public License v1.0, which accompanies this distribution
  * and is available at http://www.eclipse.org/legal/epl-v10.html.
  *
  * Contributors:
  *     Oracle - initial API and implementation
  ******************************************************************************/
 package org.eclipse.jpt.db.internal;

 import java.util.Properties;

 import org.eclipse.core.runtime.IStatus;
 import org.eclipse.datatools.connectivity.IManagedConnection;
 import org.eclipse.datatools.connectivity.db.generic.IDBDriverDefinitionConstants;
 import org.eclipse.datatools.connectivity.drivers.DriverManager;
 import org.eclipse.datatools.connectivity.sqm.core.connection.ConnectionInfo;
 import org.eclipse.datatools.sqltools.core.DatabaseIdentifier;
 import org.eclipse.datatools.sqltools.core.profile.ProfileUtil;

 /**
  *  Wrap a DTP ConnectionProfile
  */
 public final class DTPConnectionProfileWrapper extends ConnectionProfile {

 	final private org.eclipse.datatools.connectivity.IConnectionProfile dtpConnectionProfile;

 	private Connection connection; // Lazy initialized

     public static final String LIVE_CONNECTION_TYPE = "java.sql.Connection";  //$NON-NLS-1$
     public static final String OFFLINE_CONNECTION_TYPE = "org.eclipse.datatools.connectivity.sqm.core.connection.ConnectionInfo";  //$NON-NLS-1$

     public static final String CONNECTION_PROFILE_TYPE = "org.eclipse.datatools.connectivity.db.generic.connectionProfile";  //$NON-NLS-1$
     public static final String DATABASE_PRODUCT_PROPERTY = "org.eclipse.datatools.connectivity.server.version";  //$NON-NLS-1$
 	/**
 	 * This property is used in ConnectionProfile creation.
 	 */
 	public static final String DATABASE_SAVE_PWD_PROP_ID = IDBDriverDefinitionConstants.PROP_PREFIX + "savePWD"; //$NON-NLS-1$
 	/**
 	 * This property is used in ConnectionProfile creation.
 	 */
 	public static final String DRIVER_DEFINITION_PROP_ID = "org.eclipse.datatools.connectivity.driverDefinitionID"; //$NON-NLS-1$
 	/**
 	 * This property is used in DriverDefinition creation.
 	 */
 	public static final String DRIVER_DEFINITION_TYPE_PROP_ID = "org.eclipse.datatools.connectivity.drivers.defnType"; //$NON-NLS-1$
 	/**
 	 * This property is used in DriverDefinition creation.
 	 */
 	public static final String DRIVER_JAR_LIST_PROP_ID = "jarList"; //$NON-NLS-1$

 	public static final String POSTGRESQL_VENDOR = "postgres"; //$NON-NLS-1$

 	public static final String PUBLIC_SCHEMA = "public"; //$NON-NLS-1$

 	// ********** constructors **********

 	DTPConnectionProfileWrapper( ConnectionProfileRepository profileRepository, org.eclipse.datatools.connectivity.IConnectionProfile dtpConnectionProfile) {
 		super( profileRepository);
 		this.dtpConnectionProfile = dtpConnectionProfile;
 	}

 	// ********** listeners **********

 	@Override
 	public void addProfileListener( ProfileListener listener) {

 		this.getProfileRepository().addProfileListener( listener);
 	}

 	@Override
 	public void removeProfileListener( ProfileListener listener) {

 		this.getProfileRepository().removeProfileListener( listener);
 	}

 	@Override
 	public void addConnectionListener( ConnectionListener listener) {

 		this.getConnection().addConnectionListener( listener);
 	}

 	@Override
 	public void removeConnectionListener( ConnectionListener listener) {

 		this.getConnection().removeConnectionListener( listener);
 	}

 	// ********** behavior **********

 	IManagedConnection getDTPConnection() {
 		return this.dtpConnectionProfile.getManagedConnection( LIVE_CONNECTION_TYPE);
 	}

 	IManagedConnection getDTPOfflineConnection() {
 		return this.dtpConnectionProfile.getManagedConnection( OFFLINE_CONNECTION_TYPE);
 	}

 	/**
 	 * Connect using this profile.
 	 */
 	@Override
 	public void connect() {
 		if( ! this.connectionIsOnline()) {

 			IStatus status = this.dtpConnectionProfile.connect();
 			if( ! status.isOK()) {
 				if( status.isMultiStatus()) {
 					IStatus[] statusChildren = status.getChildren();
 					throw new RuntimeException( statusChildren[ 0].getMessage(), statusChildren[ 0].getException());
 				}
 				throw new RuntimeException( status.getMessage(), status.getException());
 			}
 		}
 	}

 	@Override
 	public void disconnect() {

 		IStatus status = this.dtpConnectionProfile.disconnect();
 		if( !status.isOK()) {
 			if( status.isMultiStatus()) {
 				IStatus[] statusChildren = status.getChildren();
 				throw new RuntimeException( statusChildren[ 0].getMessage(), statusChildren[ 0].getException());
 			}
 			throw new RuntimeException( status.getMessage(), status.getException());
 		}
 	}

 	@Override
 	protected void dispose() {
 		super.dispose();

 		this.disposeConnection();
 	}

 	private void disposeConnection() {
 		if( this.getConnection() != null) {
 			this.getConnection().dispose();
 		}
 	}

 	@Override
 	void databaseChanged( Database database, int eventType) {
 		this.getConnection().databaseChanged( database, eventType);
 		return;
 	}

 	@Override
 	 void catalogChanged( Catalog catalog, Database database, int eventType) {
 		 //TODO
 //		this.getConnection().catalogChanged( catalog, eventType);
 		 return;
 	}

 	@Override
 	void schemaChanged( Schema schema, Database database, int eventType) {
 		this.getConnection().schemaChanged( schema, database, eventType);
 	}

 	@Override
 	void tableChanged( Table table, Schema schema, Database database, int eventType) {
 		this.getConnection().tableChanged( table, schema, database, eventType);
 	}

 	protected Connection buildConnection() {

 		Connection connection = Connection.createConnection( this);  //$NON-NLS-1$
 		return connection;
 	}

 	@Override
 	protected Database buildDatabase() {
 		org.eclipse.datatools.modelbase.sql.schema.Database dtpDatabase;
 		if( this.isConnected()) {
 			if( this.isWorkingOffline()) {
 				ConnectionInfo connectionInfo = ( ConnectionInfo) this.getDTPOfflineConnection().getConnection().getRawConnection();
 				dtpDatabase = connectionInfo.getSharedDatabase();
 			}
 			else {
 				dtpDatabase = ProfileUtil.getDatabase( new DatabaseIdentifier( this.getName(), this.getDatabaseName()), false);
 			}
 			return Database.createDatabase( this, dtpDatabase);
 		}
 		return NullDatabase.instance();
 	}

 	public IStatus saveWorkOfflineData() {
 		return this.dtpConnectionProfile.saveWorkOfflineData();
 	}

 	public IStatus workOffline() {
 		return this.dtpConnectionProfile.workOffline();
 	}

 	@Override
 	boolean wraps( org.eclipse.datatools.connectivity.IConnectionProfile dtpProfile) {
 		return this.dtpConnectionProfile == dtpProfile;
 	}


 	// ********** queries **********

 	@Override
 	public boolean isConnected() {
 		return this.getDTPConnection().isConnected() || this.isWorkingOffline();
 	}

 	@Override
 	public boolean isWorkingOffline() {
 		return this.getDTPOfflineConnection().isWorkingOffline();
 	}

 	public boolean supportsWorkOfflineMode() {
 		return this.dtpConnectionProfile.supportsWorkOfflineMode();
 	}

 	public boolean canWorkOffline() {
 		return this.dtpConnectionProfile.canWorkOffline();
 	}

 	@Override
 	public boolean isNull() {
 		return false;
 	}

 	@Override
 	public String getName() {

 		return this.dtpConnectionProfile.getName();
 	}

 	@Override
 	public String getDatabaseName() {
 		return this.getProperties().getProperty( IDBDriverDefinitionConstants.DATABASE_NAME_PROP_ID);
 	}

 	@Override
 	public String getDatabaseProduct() {
 		return this.getProperties().getProperty( DATABASE_PRODUCT_PROPERTY);
 	}

 	@Override
 	public String getDatabaseVendor() {
 		return this.getProperties().getProperty( IDBDriverDefinitionConstants.DATABASE_VENDOR_PROP_ID);
 	}

 	@Override
 	public String getDatabaseVersion() {
 		return this.getProperties().getProperty( IDBDriverDefinitionConstants.DATABASE_VERSION_PROP_ID);
 	}

 	@Override
 	public String getUserName() {
 		return this.getProperties().getProperty( IDBDriverDefinitionConstants.USERNAME_PROP_ID);
 	}

 	@Override
 	public String getUserPassword() {
 		return this.getProperties().getProperty( IDBDriverDefinitionConstants.PASSWORD_PROP_ID);
 	}

 	@Override
 	public String getDefaultSchema() {
 		if( this.getDatabase().getVendor().equalsIgnoreCase( POSTGRESQL_VENDOR)) {
 			return PUBLIC_SCHEMA;
 		}
 		return this.getUserName();
 	}

 	@Override
 	public String getDriverClass() {
 		return this.getProperties().getProperty( IDBDriverDefinitionConstants.DRIVER_CLASS_PROP_ID);
 	}

 	@Override
 	public String getUrl() {
 		return this.getProperties().getProperty( IDBDriverDefinitionConstants.URL_PROP_ID);
 	}

 	@Override
 	public String getInstanceId() {
 		return this.dtpConnectionProfile.getInstanceID();
 	}

 	@Override
 	public String getProviderId() {
 		return this.dtpConnectionProfile.getProviderId();
 	}

 	@Override
 	public String getDriverDefinitionId() {
 		return this.getProperties().getProperty( DTPConnectionProfileWrapper.DRIVER_DEFINITION_PROP_ID);
 	}

 	@Override
 	public String getDriverJarList() {
 		return DriverManager.getInstance().getDriverInstanceByID( this.getDriverDefinitionId()).getJarList();
 	}

 	private Properties getProperties() {
 		return this.dtpConnectionProfile.getBaseProperties();
 	}

 	private Connection getConnection() {

 		if( this.connection == null) {
 			this.connection = this.buildConnection();
 			this.engageConnectionListener();
 		}
 		return this.connection;
 	}
 }
	/*******************************************************************************
	* Copyright (c) 2006, 2007 Oracle. All rights reserved.
	* This program and the accompanying materials are made available under the
	* terms of the Eclipse Public License v1.0, which accompanies this distribution
	* and is available at http://www.eclipse.org/legal/epl-v10.html.
	*
	* Contributors:
	* Oracle - initial API and implementation
	******************************************************************************/
	package org.eclipse.jpt.db.internal;

	import java.util.Properties;

	import org.eclipse.core.runtime.IStatus;
	import org.eclipse.datatools.connectivity.IManagedConnection;
	import org.eclipse.datatools.connectivity.db.generic.IDBDriverDefinitionConstants;
	import org.eclipse.datatools.connectivity.drivers.DriverManager;
	import org.eclipse.datatools.connectivity.sqm.core.connection.ConnectionInfo;
	import org.eclipse.datatools.sqltools.core.DatabaseIdentifier;
	import org.eclipse.datatools.sqltools.core.profile.ProfileUtil;

	/**
	* Wrap a DTP ConnectionProfile
	*/
	public final class DTPConnectionProfileWrapper extends ConnectionProfile {

	final private org.eclipse.datatools.connectivity.IConnectionProfile dtpConnectionProfile;

	private Connection connection; // Lazy initialized

	public static final String LIVE_CONNECTION_TYPE = "java.sql.Connection"; //$NON-NLS-1$
	public static final String OFFLINE_CONNECTION_TYPE = "org.eclipse.datatools.connectivity.sqm.core.connection.ConnectionInfo"; //$NON-NLS-1$

	public static final String CONNECTION_PROFILE_TYPE = "org.eclipse.datatools.connectivity.db.generic.connectionProfile"; //$NON-NLS-1$
	public static final String DATABASE_PRODUCT_PROPERTY = "org.eclipse.datatools.connectivity.server.version"; //$NON-NLS-1$
	/**
	* This property is used in ConnectionProfile creation.
	*/
	public static final String DATABASE_SAVE_PWD_PROP_ID = IDBDriverDefinitionConstants.PROP_PREFIX + "savePWD"; //$NON-NLS-1$
	/**
	* This property is used in ConnectionProfile creation.
	*/
	public static final String DRIVER_DEFINITION_PROP_ID = "org.eclipse.datatools.connectivity.driverDefinitionID"; //$NON-NLS-1$
	/**
	* This property is used in DriverDefinition creation.
	*/
	public static final String DRIVER_DEFINITION_TYPE_PROP_ID = "org.eclipse.datatools.connectivity.drivers.defnType"; //$NON-NLS-1$
	/**
	* This property is used in DriverDefinition creation.
	*/
	public static final String DRIVER_JAR_LIST_PROP_ID = "jarList"; //$NON-NLS-1$

	public static final String POSTGRESQL_VENDOR = "postgres"; //$NON-NLS-1$

	public static final String PUBLIC_SCHEMA = "public"; //$NON-NLS-1$

	// ******** constructors ********

	DTPConnectionProfileWrapper( ConnectionProfileRepository profileRepository, org.eclipse.datatools.connectivity.IConnectionProfile dtpConnectionProfile) {
	super( profileRepository);
	this.dtpConnectionProfile = dtpConnectionProfile;
	}

	// ******** listeners ********

	@Override
	public void addProfileListener( ProfileListener listener) {

	this.getProfileRepository().addProfileListener( listener);
	}

	@Override
	public void removeProfileListener( ProfileListener listener) {

	this.getProfileRepository().removeProfileListener( listener);
	}

	@Override
	public void addConnectionListener( ConnectionListener listener) {

	this.getConnection().addConnectionListener( listener);
	}

	@Override
	public void removeConnectionListener( ConnectionListener listener) {

	this.getConnection().removeConnectionListener( listener);
	}

	// ******** behavior ********

	IManagedConnection getDTPConnection() {
	return this.dtpConnectionProfile.getManagedConnection( LIVE_CONNECTION_TYPE);
	}

	IManagedConnection getDTPOfflineConnection() {
	return this.dtpConnectionProfile.getManagedConnection( OFFLINE_CONNECTION_TYPE);
	}

	/**
	* Connect using this profile.
	*/
	@Override
	public void connect() {
	if( ! this.connectionIsOnline()) {

	IStatus status = this.dtpConnectionProfile.connect();
	if( ! status.isOK()) {
	if( status.isMultiStatus()) {
	IStatus[] statusChildren = status.getChildren();
	throw new RuntimeException( statusChildren[ 0].getMessage(), statusChildren[ 0].getException());
	}
	throw new RuntimeException( status.getMessage(), status.getException());
	}
	}
	}

	@Override
	public void disconnect() {

	IStatus status = this.dtpConnectionProfile.disconnect();
	if( !status.isOK()) {
	if( status.isMultiStatus()) {
	IStatus[] statusChildren = status.getChildren();
	throw new RuntimeException( statusChildren[ 0].getMessage(), statusChildren[ 0].getException());
	}
	throw new RuntimeException( status.getMessage(), status.getException());
	}
	}

	@Override
	protected void dispose() {
	super.dispose();

	this.disposeConnection();
	}

	private void disposeConnection() {
	if( this.getConnection() != null) {
	this.getConnection().dispose();
	}
	}

	@Override
	void databaseChanged( Database database, int eventType) {
	this.getConnection().databaseChanged( database, eventType);
	return;
	}

	@Override
	void catalogChanged( Catalog catalog, Database database, int eventType) {
	//TODO
	// this.getConnection().catalogChanged( catalog, eventType);
	return;
	}

	@Override
	void schemaChanged( Schema schema, Database database, int eventType) {
	this.getConnection().schemaChanged( schema, database, eventType);
	}

	@Override
	void tableChanged( Table table, Schema schema, Database database, int eventType) {
	this.getConnection().tableChanged( table, schema, database, eventType);
	}

	protected Connection buildConnection() {

	Connection connection = Connection.createConnection( this); //$NON-NLS-1$
	return connection;
	}

	@Override
	protected Database buildDatabase() {
	org.eclipse.datatools.modelbase.sql.schema.Database dtpDatabase;
	if( this.isConnected()) {
	if( this.isWorkingOffline()) {
	ConnectionInfo connectionInfo = ( ConnectionInfo) this.getDTPOfflineConnection().getConnection().getRawConnection();
	dtpDatabase = connectionInfo.getSharedDatabase();
	}
	else {
	dtpDatabase = ProfileUtil.getDatabase( new DatabaseIdentifier( this.getName(), this.getDatabaseName()), false);
	}
	return Database.createDatabase( this, dtpDatabase);
	}
	return NullDatabase.instance();
	}

	public IStatus saveWorkOfflineData() {
	return this.dtpConnectionProfile.saveWorkOfflineData();
	}

	public IStatus workOffline() {
	return this.dtpConnectionProfile.workOffline();
	}

	@Override
	boolean wraps( org.eclipse.datatools.connectivity.IConnectionProfile dtpProfile) {
	return this.dtpConnectionProfile == dtpProfile;
	}


	// ******** queries ********

	@Override
	public boolean isConnected() {
	return this.getDTPConnection().isConnected() \|\| this.isWorkingOffline();
	}

	@Override
	public boolean isWorkingOffline() {
	return this.getDTPOfflineConnection().isWorkingOffline();
	}

	public boolean supportsWorkOfflineMode() {
	return this.dtpConnectionProfile.supportsWorkOfflineMode();
	}

	public boolean canWorkOffline() {
	return this.dtpConnectionProfile.canWorkOffline();
	}

	@Override
	public boolean isNull() {
	return false;
	}

	@Override
	public String getName() {

	return this.dtpConnectionProfile.getName();
	}

	@Override
	public String getDatabaseName() {
	return this.getProperties().getProperty( IDBDriverDefinitionConstants.DATABASE_NAME_PROP_ID);
	}

	@Override
	public String getDatabaseProduct() {
	return this.getProperties().getProperty( DATABASE_PRODUCT_PROPERTY);
	}

	@Override
	public String getDatabaseVendor() {
	return this.getProperties().getProperty( IDBDriverDefinitionConstants.DATABASE_VENDOR_PROP_ID);
	}

	@Override
	public String getDatabaseVersion() {
	return this.getProperties().getProperty( IDBDriverDefinitionConstants.DATABASE_VERSION_PROP_ID);
	}

	@Override
	public String getUserName() {
	return this.getProperties().getProperty( IDBDriverDefinitionConstants.USERNAME_PROP_ID);
	}

	@Override
	public String getUserPassword() {
	return this.getProperties().getProperty( IDBDriverDefinitionConstants.PASSWORD_PROP_ID);
	}

	@Override
	public String getDefaultSchema() {
	if( this.getDatabase().getVendor().equalsIgnoreCase( POSTGRESQL_VENDOR)) {
	return PUBLIC_SCHEMA;
	}
	return this.getUserName();
	}

	@Override
	public String getDriverClass() {
	return this.getProperties().getProperty( IDBDriverDefinitionConstants.DRIVER_CLASS_PROP_ID);
	}

	@Override
	public String getUrl() {
	return this.getProperties().getProperty( IDBDriverDefinitionConstants.URL_PROP_ID);
	}

	@Override
	public String getInstanceId() {
	return this.dtpConnectionProfile.getInstanceID();
	}

	@Override
	public String getProviderId() {
	return this.dtpConnectionProfile.getProviderId();
	}

	@Override
	public String getDriverDefinitionId() {
	return this.getProperties().getProperty( DTPConnectionProfileWrapper.DRIVER_DEFINITION_PROP_ID);
	}

	@Override
	public String getDriverJarList() {
	return DriverManager.getInstance().getDriverInstanceByID( this.getDriverDefinitionId()).getJarList();
	}

	private Properties getProperties() {
	return this.dtpConnectionProfile.getBaseProperties();
	}

	private Connection getConnection() {

	if( this.connection == null) {
	this.connection = this.buildConnection();
	this.engageConnectionListener();
	}
	return this.connection;
	}
	}