// This software is released into the Public Domain. See copying.txt for details.
package org.openstreetmap.osmosis.pgsimple.v0_6;
import org.openstreetmap.osmosis.core.container.v0_6.Dataset;
import org.openstreetmap.osmosis.core.container.v0_6.DatasetContext;
import org.openstreetmap.osmosis.core.database.DatabaseLoginCredentials;
import org.openstreetmap.osmosis.core.database.DatabasePreferences;
import org.openstreetmap.osmosis.pgsimple.v0_6.impl.PostgreSqlDatasetContext;
import org.openstreetmap.osmosis.core.task.v0_6.DatasetSink;
import org.openstreetmap.osmosis.core.task.v0_6.RunnableDatasetSource;
/**
* An OSM dataset source exposing generic access to a custom PostgreSQL database.
*
* @author Brett Henderson
*/
public class PostgreSqlDatasetReader implements RunnableDatasetSource, Dataset {
private DatasetSink datasetSink;
private DatabaseLoginCredentials loginCredentials;
private DatabasePreferences preferences;
/**
* Creates a new instance.
*
* @param loginCredentials
* Contains all information required to connect to the database.
* @param preferences
* Contains preferences configuring database behaviour.
*/
public PostgreSqlDatasetReader(DatabaseLoginCredentials loginCredentials, DatabasePreferences preferences) {
this.loginCredentials = loginCredentials;
this.preferences = preferences;
}
/**
* {@inheritDoc}
*/
@Override
public void setDatasetSink(DatasetSink datasetSink) {
this.datasetSink = datasetSink;
}
/**
* {@inheritDoc}
*/
@Override
public void run() {
try {
datasetSink.process(this);
} finally {
datasetSink.release();
}
}
/**
* {@inheritDoc}
*/
@Override
public DatasetContext createReader() {
return new PostgreSqlDatasetContext(loginCredentials, preferences);
}
}