| gvSIG desktop 1

/*

 * Created on 16-ene-2007 by azabala

 *

 */

package com.hardcode.gdbms.driver.mysql;

import java.sql.Connection;

import java.sql.ResultSetMetaData;

import java.sql.SQLException;

import java.sql.Statement;

import java.sql.Types;

import com.iver.cit.gvsig.fmap.core.IRow;

import com.iver.cit.gvsig.fmap.drivers.DBLayerDefinition;

import com.iver.cit.gvsig.fmap.drivers.FieldDescription;

import com.iver.cit.gvsig.fmap.drivers.ITableDefinition;

import com.iver.cit.gvsig.fmap.edition.EditionException;

import com.iver.cit.gvsig.fmap.edition.IRowEdited;

import com.iver.cit.gvsig.fmap.edition.fieldmanagers.JdbcFieldManager;

import com.iver.cit.gvsig.fmap.edition.writers.AbstractWriter;

/**

 * 

 * MySQL alphanumeric writer.

 * It is needed because MySQL JDBC driver doesnt allow updatables recordsets.

 * @author alzabord

 * 

 * @see MySQLSpatialWriter from extJDBC project

 */

public class MySQLWriter extends AbstractWriter{

        private int numRows;

        private Connection conn;

        private Statement st;

        private ResultSetMetaData metaData;

        private boolean bCreateTable;

        private boolean bWriteAll;

        private MySQL mySql = new MySQL();

        private JdbcFieldManager fieldManager;

        public void initialize(Connection conn){

                this.conn = conn;

        }

        public void initialize(ITableDefinition tableDefinition) throws EditionException {

                super.initialize(tableDefinition);

                try {

                        createTableIfNeeded();

                } catch (SQLException e) {

                        throw new EditionException(e);

                }

        }

        private void createTableIfNeeded() throws SQLException{

                st = conn.createStatement();

                if (bCreateTable) {

                        try {

                                st.execute("DROP TABLE " + 

                                                ((DBLayerDefinition)tableDef).getTableName() + ";");

                        } catch (SQLException e1) {

                                e1.printStackTrace();

                        }

                        //In MySQL you can add geometry column in CREATE TABLE statement

                        String sqlCreate = mySql.getSqlCreateSpatialTable((DBLayerDefinition) tableDef,

                                                                                        tableDef.getFieldsDesc(), 

                                                                                        true);

                        st.execute(sqlCreate);

                        conn.commit();

                }//if

                conn.setAutoCommit(false);

                fieldManager = new JdbcFieldManager(conn, 

                                ((DBLayerDefinition) tableDef).getTableName());

}

        public boolean canWriteAttribute(int sqlType) {

                switch (sqlType) {

                case Types.DOUBLE:

                case Types.FLOAT:

                case Types.INTEGER:

                case Types.BIGINT:

                        return true;

                case Types.DATE:

                        return true;

                case Types.BIT:

                case Types.BOOLEAN:

                        return true;

                case Types.VARCHAR:

                case Types.CHAR:

                case Types.LONGVARCHAR:

                        return true; 

                }

                return false;

        }

        public void preProcess() throws EditionException {

                numRows = 0;

                try {

                        conn.setAutoCommit(false);

                        conn.rollback();                        

                        alterTable();

                } catch (SQLException e) {

                        throw new EditionException(e);

                }

        }

        public boolean alterTable() throws EditionException {

                return fieldManager.alterTable();

        }

        public void process(IRowEdited row) throws EditionException {

                String sqlInsert;

                try {

                        switch (row.getStatus()) {

                        case IRowEdited.STATUS_ADDED:

                                IRow record =  row.getLinkedRow();

                                sqlInsert = mySql.getSqlInsertFeature((DBLayerDefinition) tableDef, record);

                                st.execute(sqlInsert);

                                break;

                        case IRowEdited.STATUS_MODIFIED:

                                IRow featM =  row.getLinkedRow();

                                if (bWriteAll) {

                                        sqlInsert = mySql.getSqlInsertFeature((DBLayerDefinition) tableDef, featM);

                                        System.out.println("sql = " + sqlInsert);

                                        st.execute(sqlInsert);

                                } else {

                                        String sqlModify = mySql.getSqlModifyFeature((DBLayerDefinition) tableDef, featM);

                                        st.execute(sqlModify);

                                }

                                break;

                        case IRowEdited.STATUS_ORIGINAL:

                                IRow featO = row.getLinkedRow();

                                if (bWriteAll) {

                                        sqlInsert = mySql.getSqlInsertFeature((DBLayerDefinition) tableDef, featO);

                                        st.execute(sqlInsert);

                                }

                                break;

                        case IRowEdited.STATUS_DELETED:

                                String sqlDelete = mySql.getSqlDeleteFeature((DBLayerDefinition) tableDef, row);

                                System.out.println("sql = " + sqlDelete);

                                st.execute(sqlDelete);

                                break;

                        }

                        numRows++;

                } catch (SQLException e) {

                        e.printStackTrace();

                        throw new EditionException(e);

                }

        }

        public void postProcess() throws EditionException {

                try {

                        conn.setAutoCommit(true);

                } catch (SQLException e) {

                        e.printStackTrace();

                        throw new EditionException(e);

                }

        }

        public boolean canAlterTable() {

                return true;

        }

        public boolean canSaveEdits() {

                try {

                        return !conn.isReadOnly();

                } catch (SQLException e) {

                        e.printStackTrace();

                        return false;

                }

        }

        public String getName() {

                return "MySQL Alphanumeric Writer";

        }

        public void setCreateTable(boolean createTable) {

                bCreateTable = createTable;

        }

        public boolean isCreateTable() {

                return bCreateTable;

        }

        public boolean isWriteAll() {

                return bWriteAll;

        }

        public void setWriteAll(boolean writeAll) {

                bWriteAll = writeAll;

        }

/*

 * TODO 

 * Esta parte no se si es necesaria, o si se usa (revisar)

 * 

 * */

        public FieldDescription[] getFields() {

                return fieldManager.getFields();

        }

        public FieldDescription[] getOriginalFields() {

                return tableDef.getFieldsDesc();

        }

        public void addField(FieldDescription fieldDesc) {

                fieldManager.addField(fieldDesc);

        }

        public FieldDescription removeField(String fieldName) {

                return fieldManager.removeField(fieldName);

        }

        public void renameField(String antName, String newName) {

                fieldManager.renameField(antName, newName);        

        }

}