Application: gvSIG desktop

/*

 * Created on 16-oct-2004

 */

package com.hardcode.gdbms.driver.mysql;

import com.hardcode.driverManager.Driver;

import com.hardcode.gdbms.engine.data.DBDriver;

import com.hardcode.gdbms.engine.data.DriverException;

import com.hardcode.gdbms.engine.values.Value;

import com.hardcode.gdbms.engine.values.ValueFactory;

import java.sql.Connection;

import java.sql.DriverManager;

import java.sql.ResultSet;

import java.sql.SQLException;

import java.sql.Statement;

import java.sql.Types;

/**

 * DOCUMENT ME!

 *

 * @author Fernando Gonz?lez Cort?s

 */

public class MySQLDriver implements DBDriver, Driver {

    private static Exception driverException;

    static {

        try {

            Class.forName("com.mysql.jdbc.Driver").newInstance();

        } catch (Exception ex) {

            driverException = ex;

        }

    }

    private Connection conn;

    private Statement st;

    private ResultSet res;

    private long tableLength;

    /**

     * DOCUMENT ME!

     *

     * @param host DOCUMENT ME!

     * @param port DOCUMENT ME!

     * @param dbName DOCUMENT ME!

     * @param user DOCUMENT ME!

     * @param password DOCUMENT ME!

     *

     * @throws SQLException

     * @throws RuntimeException DOCUMENT ME!

     *

     * @see com.hardcode.gdbms.engine.data.DBDriver#connect(java.lang.String)

     */

    public void connect(String host, int port, String dbName, String user,

        String password) throws SQLException {

        if (driverException != null) {

            throw new RuntimeException(driverException);

        }

        String connectionString = "jdbc:mysql://" + host;

        if (port != -1){

                connectionString += ":" + port;

        }

        connectionString += "/" + dbName;

        if (user != null){

                connectionString += "?user=" + user + "&password=" + password;

        }

        conn = DriverManager.getConnection(connectionString);

    }

    /**

     * DOCUMENT ME!

     *

     * @throws SQLException

     *

     * @see com.hardcode.gdbms.engine.data.DBDriver#closeConnection()

     */

    public void closeConnection() throws SQLException {

        conn.close();

    }

    /**

     * @see com.hardcode.gdbms.engine.data.ReadDriver#getFieldValue(long, int)

     */

    public Value getFieldValue(long rowIndex, int fieldId)

        throws DriverException {

        try {

            fieldId++;

            res.absolute((int) rowIndex + 1);

            int type = res.getMetaData().getColumnType(fieldId);

            switch (type) {

                case Types.BIGINT:

                    return ValueFactory.createValue(res.getLong(fieldId));

                case Types.BIT:

                    return ValueFactory.createValue(res.getBoolean(fieldId));

                case Types.CHAR:

                    return ValueFactory.createValue(res.getString(fieldId));

                case Types.DATE:

                    return ValueFactory.createValue(res.getDate(fieldId));

                case Types.FLOAT:

                case Types.DOUBLE:

                    return ValueFactory.createValue(res.getDouble(fieldId));

                case Types.INTEGER:

                    return ValueFactory.createValue(res.getInt(fieldId));

                case Types.REAL:

                    return ValueFactory.createValue(res.getFloat(fieldId));

                case Types.VARCHAR:

                    return ValueFactory.createValue(res.getString(fieldId));

                case Types.VARBINARY:

                case Types.TINYINT:

                case Types.TIMESTAMP:

                case Types.TIME:

                case Types.SMALLINT:

                case Types.OTHER:

                case Types.NUMERIC:

                case Types.LONGVARCHAR:

                case Types.LONGVARBINARY:

                case Types.DECIMAL:

                case Types.BINARY:default:

                    throw new DriverException("Type not recognized: " + type);

            }

        } catch (SQLException e) {

            throw new DriverException(e);

        }

    }

    /**

     * @see com.hardcode.gdbms.engine.data.ReadDriver#getFieldCount()

     */

    public int getFieldCount() throws DriverException {

        try {

            return res.getMetaData().getColumnCount();

        } catch (SQLException e) {

            throw new DriverException(e);

        }

    }

    /**

     * @see com.hardcode.gdbms.engine.data.ReadDriver#getFieldName(int)

     */

    public String getFieldName(int fieldId) throws DriverException {

        try {

                fieldId++;

            return res.getMetaData().getColumnName(fieldId);

        } catch (SQLException e) {

            throw new DriverException(e);

        }

    }

    /**

     * @see com.hardcode.gdbms.engine.data.ReadDriver#getRowCount()

     */

    public long getRowCount() throws DriverException {

        return tableLength;

    }

    /**

     * @see com.hardcode.driverManager.Driver#getName()

     */

    public String getName() {

        return "mysql";

    }

    /**

     * @see com.hardcode.gdbms.engine.data.DBDriver#openTable(java.lang.String)

     */

    public void openTable(String table) throws SQLException {

        st = conn.createStatement(ResultSet.TYPE_SCROLL_INSENSITIVE,

                ResultSet.CONCUR_READ_ONLY);

        res = st.executeQuery("select * from " + table);

        res.last();

        tableLength = res.getRow();

        res.beforeFirst();

    }

    /**

     * @see com.hardcode.gdbms.engine.data.DBDriver#closeTable()

     */

    public void closeTable() throws SQLException {

        res.close();

        st.close();

    }

    /**

     * @see com.hardcode.gdbms.engine.data.Delegable#executeSQL(java.lang.String)

     */

    public void executeSQL(String sql) throws SQLException {

        st = conn.createStatement(ResultSet.TYPE_SCROLL_INSENSITIVE,

                ResultSet.CONCUR_READ_ONLY);

        res = st.executeQuery(sql);

        res.last();

        tableLength = res.getRow();

        res.beforeFirst();

    }

}