Application: gvSIG desktop » gvSIG data provider for PostgreSQL

package org.gvsig.postgresql.dal;

import java.net.URL;

import java.util.Properties;

import org.gvsig.fmap.dal.DALLocator;

import org.gvsig.fmap.dal.DataManager;

import org.gvsig.fmap.dal.store.jdbc2.AbstractTestUtils;

import org.gvsig.fmap.dal.store.jdbc2.JDBCHelper;

import org.gvsig.fmap.dal.store.jdbc2.spi.FakeConnectionProvider;

/**

 *

 * @author jjdelcerro

 */

@SuppressWarnings("UseSpecificCatch")

public class TestUtilsPostgreSQL extends AbstractTestUtils {

    @Override

    public String getProviderName() {

        return PostgreSQLLibrary.NAME;

    }

    @Override

    public String getExpectedsPath() {

        return "/org/gvsig/postgresql/dal";

    }

    @Override

    protected Properties getProperties() {

        if (this.properties == null) {

            try {

                URL url = this.getClass().getResource("postgresql.properties");

                Properties props = new Properties();

                props.load(url.openStream());

                String[] keys = new String[]{

                    "PostgreSQLEnabled",

                    "PostgreSQLHost",

                    "PostgreSQLPort",

                    "PostgreSQLDbname",

                    "PostgreSQLSchema",

                    "PostgreSQLUser",

                    "PostgreSQLPassword"

                };

                for (String key : keys) {

                    if (System.getenv().containsKey(key)) {

                        props.setProperty(key, System.getenv(key));

                    }

                }

                this.properties = props;

            } catch (Exception ex) {

                throw new RuntimeException(ex);

            } 

        }

        return this.properties;

    }

    @Override

    public boolean isTheDatabaseAvailable() {

        return this.getPropertyBoolean("PostgreSQLEnabled");

    }

    @Override

    public JDBCHelper createJDBCHelper() throws Exception {

        PostgreSQLServerExplorerParameters params = this.getServerExplorerParameters("fake");

        PostgreSQLHelper helper = new PostgreSQLHelper(params, new FakeConnectionProvider());

      return helper;

    }

    @Override

    public PostgreSQLServerExplorerParameters getServerExplorerParameters(String dbname) throws Exception {

        DataManager dataManager = DALLocator.getDataManager();

        PostgreSQLServerExplorerParameters conn = (PostgreSQLServerExplorerParameters) dataManager.createServerExplorerParameters(this.getProviderName());

        conn.setHost(getProperty("PostgreSQLHost"));

        conn.setPort(getPropertyInt("PostgreSQLPort", 5432));

        conn.setDBName(getProperty("PostgreSQLDbname"));

        conn.setSchema(getProperty("PostgreSQLSchema"));

        conn.setUser(getProperty("PostgreSQLUser"));

        conn.setPassword(getProperty("PostgreSQLPassword"));

        conn.setShowInformationDBTables(false);

        return conn;

    }

}