| gvSIG desktop 1

package com.hardcode.gdbms.engine.data;

import com.hardcode.driverManager.Driver;

import com.hardcode.driverManager.DriverLoadException;

import com.hardcode.driverManager.DriverManager;

import com.hardcode.gdbms.engine.Strategy;

import com.hardcode.gdbms.engine.StrategyManager;

import com.hardcode.gdbms.engine.instruction.Adapter;

import com.hardcode.gdbms.engine.instruction.SelectAdapter;

import com.hardcode.gdbms.engine.instruction.SemanticException;

import com.hardcode.gdbms.engine.instruction.TableRefAdapter;

import com.hardcode.gdbms.engine.instruction.UnionAdapter;

import com.hardcode.gdbms.engine.instruction.Utilities;

import com.hardcode.gdbms.parser.SimpleNode;

import java.io.File;

import java.io.IOException;

import java.util.HashMap;

/* Generated by Together */

public class DataSourceFactory {

    /**

     * Asocia los nombres de las tablas con la informaci?n del origen de datos

     */

    private static HashMap tableSource = new HashMap();

    /**

     * Asocia los nombres de los or?genes de datos de base de datos con el

     * nombre  de la tabla en el sistema de gesti?n original

     */

    private static HashMap nameTable = new HashMap();

    private static DriverManager dm;

    static {

        //Setup de las tablas

        /*        addFileDataSource("csv", "persona", "persona.csv", "persona.csv");

           addFileDataSource("csv", "coche", "coche.csv", "coche.csv");

           addFileDataSource("gdbms dbf driver", "vias", "vias.dbf", "vias.dbf");

           addFileDataSource("gdbms dbf driver", "ejes", "ejes.dbf", "ejes.dbf");

           addDBDataSource("mysql", "jdbc:mysql://localhost/prueba?user=root",

               "prueba", "localhost/prueba", "mysql");

         */

    }

    /**

     * A?ade un origen de datos de fichero al sistema

     *

     * @param driverName DOCUMENT ME!

     * @param name Nombre de la tabla con el que se har? referencia en las

     *        instrucciones

     * @param file Fichero con los datos

     * @param dbms Sistema de base de datos capaz de ejecutar instrucciones SQL

     *        en el  que reside el origen de datos. Este par?metro se utiliza

     *        para delegar las instrucciones SQL en dicho sistema, cuando las

     *        tablas que intervienen en una instrucci?n est?n todas en el

     *        mismo dbms. Generalmente los ficheros no residen en ning?n dbms,

     *        por lo que habr? que asignarle un valor distinto a cada

     *        DataSource. El nombre del fichero suele ser una buena opci?n

     */

    public static void addFileDataSource(String driverName, String name,

        String file, String dbms) {

        FileDriverInfo info = new FileDriverInfo();

        info.name = name;

        info.dbms = dbms;

        info.file = file;

        info.driverName = driverName;

        tableSource.put(name, info);

    }

    /**

     * A?ade un origen de datos de base de datos al sistema

     *

     * @param name Nombre de la tabla con el que se har? referencia en las

     *        instrucciones

     * @param host Cadena de conexi?n para conectar con el sgbd donde se

     *        encuentra la tabla

     * @param port Nombre del sgbd capaz de ejecutar SQL en el que reside la

     *        tabla. Generalmente ser? la parte de la cadena de conexi?n

     *        correspondiente a host, puerto y nombre de la base de datos

     * @param user Nombre de usuario. Null para acceso sin usuario

     * @param password Si el usuario es null se ignora el password

     * @param dbName Nombre de la base de datos a la que se accede

     * @param tableName Nombre de la tabla en el sgbd original

     * @param driverInfo Informaci?n para saber qu? driver debe acceder a la

     *        informaci?n. Se escoger? el driver cuyo valor de retorno del

     *        m?todo getType coincida con este valor

     */

    public static void addDBDataSource(String name, String host, int port,

        String user, String password, String dbName, String tableName,

        String driverInfo) {

        DBDriverInfo info = new DBDriverInfo();

        info.name = name;

        info.host = host;

        info.port = port;

        info.user = user;

        info.password = password;

        info.dbName = dbName;

        info.dbms = host + ":" + port + "/" + dbName;

        info.sqlQuery = tableName;

        info.driverInfo = driverInfo;

        tableSource.put(name, info);

        nameTable.put(name, tableName);

    }

    /**

     * Dado el nombre de una tabla, se busca el fichero asociado a dicha tabla

     * y se obtiene un datasource adecuado en funcion del tipo del fichero

     * accediendo al subsistema de drivers

     *

     * @param tableName Nombre de la tabla

     *

     * @return DataSource que accede a dicha tabla

     *

     * @throws NoSuchTableException

     * @throws DriverLoadException

     * @throws RuntimeException Si se encuentra un adaptador no contemplado

     *         (bug)

     */

    public static DataSource createRandomDataSource(String tableName)

        throws NoSuchTableException, DriverLoadException {

        DriverInfo info = (DriverInfo) tableSource.get(tableName);

        if (info == null) {

            throw new NoSuchTableException(tableName);

        }

        if (info instanceof FileDriverInfo) {

            FileDriverInfo fileInfo = (FileDriverInfo) info;

            Driver d = dm.getDriver(fileInfo.driverName);

            FileDataSourceAdapter adapter = new FileDataSourceAdapter();

            adapter.setDriver((FileDriver) d);

            adapter.setFile(new File(fileInfo.file));

            adapter.setName(tableName);

            adapter.setDBMS(fileInfo.dbms);

            return adapter;

        } else if (info instanceof DBDriverInfo) {

            DBDriverInfo dbInfo = (DBDriverInfo) info;

            String driverInfo = dbInfo.driverInfo;

            Driver d = dm.getDriver(driverInfo);

            DBTableDataSourceAdapter adapter = new DBTableDataSourceAdapter();

            adapter.setDriver((DBDriver) d);

            adapter.setConnectionString(dbInfo.host, dbInfo.port, dbInfo.dbName, dbInfo.user, dbInfo.password);

            adapter.setTable(dbInfo.sqlQuery);

            adapter.setName(tableName);

            adapter.setDBMS(dbInfo.dbms);

            return adapter;

        } else {

            throw new RuntimeException();

        }

    }

    /**

     * Dado el nombre de una tabla, se busca el fichero asociado a dicha tabla

     * y se obtiene un datasource adecuado en funcion del tipo del fichero

     * accediendo al subsistema de drivers. Se utiliza internamente como

     * nombre de la tabla el alias que se pasa como par?metro

     *

     * @param tableName Nombre de la tabla que se asocia con un fichero

     * @param tableAlias Alias que tiene el DataSource en una instrucci?n

     *

     * @return DataSource que accede a dicha tabla

     *

     * @throws NoSuchTableException

     * @throws DriverLoadException

     */

    public static DataSource createRandomDataSource(String tableName,

        String tableAlias) throws NoSuchTableException, DriverLoadException {

        DefaultDataSource adapter = (DefaultDataSource) createRandomDataSource(tableName);

        adapter.setName(tableAlias);

        return adapter;

    }

    /**

     * Devuelve true si todas las tablas provienen del mismo data base

     * management system

     *

     * @param tables Array de tablas

     *

     * @return boolean

     */

    private static boolean sameDBMS(DataSource[] tables) {

        String dbms = tables[0].getDBMS();

        for (int i = 1; i < tables.length; i++) {

            if (!dbms.equals(tables[1].getDBMS())) {

                return false;

            }

        }

        return true;

    }

    /**

     * Devuelve true si todas las tablas provienen directamente de un driver de

     * base de datos, no son fruto de ninguna operaci?n

     *

     * @param tables Array de tablas

     *

     * @return boolean

     */

    private static boolean allDrivers(DataSource[] tables) {

        for (int i = 0; i < tables.length; i++) {

            if (tables[i].getDriver() == null) {

                return false;

            }

        }

        return true;

    }

    /**

     * Traduce los nombres en la instrucci?n, sustituyendo los nombres del

     * GDBMS por los nombres de tabla en el sistema original

     *

     * @param adapter

     */

    public static void translateNames(Adapter adapter) {

        //Se itera recursivamente sustituyendo el contenido de los tableRef

        if (adapter instanceof TableRefAdapter) {

            SimpleNode n = adapter.getEntity();

            n.first_token.image = (String) nameTable.get(n.first_token.image);

        } else {

            Adapter[] hijos = adapter.getChilds();

            for (int i = 0; i < hijos.length; i++) {

                translateNames(hijos[i]);

            }

        }

    }

    /**

     * A partir de una instrucci?n select se encarga de obtener el DataSource

     * resultado de la ejecuci?n de dicha instrucci?n

     *

     * @param instr Instrucci?n select origen del datasource

     *

     * @return DataSource que accede a los datos resultado de ejecutar la

     *         select

     *

     * @throws DriverLoadException

     * @throws DriverException Si fallo la lectura de la fuente de datos por

     *         parte del driver

     * @throws SemanticException Si la instrucci?n tiene errores sem?nticos

     * @throws IOException

     */

    public static DataSource createRandomDataSource(SelectAdapter instr)

        throws DriverLoadException, DriverException, SemanticException, 

            IOException {

        DataSource[] tables = instr.getTables();

        //Estrategia de delegaci?n de la instrucci?n en el sgbd original de la tabla

        if (sameDBMS(tables) && allDrivers(tables)) {

            ReadDriver driver = tables[0].getDriver();

            if (driver instanceof DBDriver) {

                DBDriver d = (DBDriver) driver;

                //Obtiene un driver del mismo tipo

                Driver newDriver = dm.getDriver(((Driver) d).getName());

                DBQueryDataSourceAdapter adapter = new DBQueryDataSourceAdapter();

                adapter.setDriver((DBDriver) newDriver);

                DBDataSourceAdapter dbAdapter = ((DBDataSourceAdapter) tables[0]);

                adapter.setConnectionString(dbAdapter.getHost(), dbAdapter.getPort(), dbAdapter.getDbName(), dbAdapter.getUser(), dbAdapter.getPassword());

                translateNames(instr);

                adapter.setSQL(Utilities.getText(instr.getEntity()));

                adapter.setName("");

                adapter.setDBMS(null);

                return adapter;

            }

        }

        //Estrategia normal

        Strategy strategy = StrategyManager.getStrategy(instr);

        return strategy.select(instr);

    }

    /**

     * Obtiene el DataSource resultado de ejecutar la instrucci?n de union

     *

     * @param instr instrucci?n de union

     *

     * @return DataSource

     *

     * @throws DriverException Si fallo la lectura de la fuente de datos por

     *         parte del driver

     * @throws IOException Si se produce un error de entrada/salida

     * @throws SemanticException Si la instrucci?n tiene errores sem?nticos

     */

    public static DataSource createRandomDataSource(UnionAdapter instr)

        throws DriverException, IOException, SemanticException {

        Strategy strategy = StrategyManager.getStrategy(instr);

        return strategy.union(instr);

    }

    /**

     * Establece el DriverManager que se usar? para instanciar DataSource's.

     * Este metodo debe ser invocado antes que ning?n otro

     *

     * @param dm El manager que se encarga de cargar los drivers

     */

    public static void setDriverManager(DriverManager dm) {

        DataSourceFactory.dm = dm;

    }

    /**

     * Informaci?n del driver

     *

     * @author Fernando Gonz?lez Cort?s

     */

    public static class DriverInfo {

        public String name;

        public String dbms;

        public String driverName;

    }

    /**

     * Informaci?n del driver de fichero

     *

     * @author Fernando Gonz?lez Cort?s

     */

    public static class FileDriverInfo extends DriverInfo {

        public String file;

    }

    /**

     * Informaci?n del driver de base de datos

     *

     * @author Fernando Gonz?lez Cort?s

     */

    public static class DBDriverInfo extends DriverInfo {

        public String host;

        public int port;

        public String dbName;

        public String user;

        public String password;

        public String sqlQuery;

        public String driverInfo;

    }

}