Application: gvSIG desktop

/**

 * gvSIG. Desktop Geographic Information System.

 *

 * Copyright (C) 2007-2020 gvSIG Association.

 *

 * This program is free software; you can redistribute it and/or

 * modify it under the terms of the GNU General Public License

 * as published by the Free Software Foundation; either version 3

 * of the License, or (at your option) any later version.

 *

 * This program is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 * GNU General Public License for more details.

 *

 * You should have received a copy of the GNU General Public License

 * along with this program; if not, write to the Free Software

 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,

 * MA  02110-1301, USA.

 *

 * For any additional information, do not hesitate to contact us

 * at info AT gvsig.com, or visit our website www.gvsig.com.

 */

package org.gvsig.fmap.dal.store.jdbc2.spi.operations;

import java.sql.Connection;

import java.sql.SQLException;

import org.gvsig.fmap.dal.resource.exception.AccessResourceException;

import org.gvsig.fmap.dal.store.jdbc.exception.JDBCTransactionRollbackException;

import org.gvsig.fmap.dal.store.jdbc.exception.SQLRuntimeException;

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

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

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

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

@SuppressWarnings("UseSpecificCatch")

public abstract class AbstractConnectionOperation implements ConnectionOperation {

    final static protected Logger LOGGER = LoggerFactory.getLogger(AbstractConnectionOperation.class);

    final protected JDBCHelper helper;

    protected JDBCConnection conn = null;

    public AbstractConnectionOperation(JDBCHelper helper) {

        this.helper = helper;

    }

    @Override

    public boolean continueTransactionAllowed() {

        return this.helper.getTransaction()!=null;

    }

    @Override

    public boolean needTransaction() {

        return true;

    }

    protected JDBCSQLBuilderBase createSQLBuilder() {

        return this.helper.createSQLBuilder();

    }

    @Override

    public Object perform()  {

        try {

            if( this.helper.isThreadSafe() ) {

                return this.perform_operation();

            } 

            synchronized(AbstractConnectionOperation.class) {

                return this.perform_operation();

            }

        } catch (RuntimeException ex) {

            throw ex;

        } catch (Exception ex) {

            throw new RuntimeException("Can't perform operation '"+this.getClass().getSimpleName()+"'.",ex);

        }

    }

    protected JDBCConnection getConnection() throws AccessResourceException {

        if( this.conn == null ) {

            this.conn = this.helper.getConnection();

        }

        return this.conn;

    }

    protected Object perform_operation() throws Exception {

        Object result = null;

        LOGGER.trace("preparing execution of "+this.getClass().getSimpleName()+".");

        if( needTransaction() ) {

            try {

                if( this.helper.allowNestedOperations() ) {

                    boolean commitOnFinish = true;

                    if (!this.getConnection().isInTransaction()) {

                        commitOnFinish = false;

                    } 

                    this.getConnection().begin();

                    try {

                        LOGGER.trace("Excuting operation {}.", this.getClass().getSimpleName());

                        result = perform(this.getConnection());

                    } catch (Exception ex) {

                        try {

                            this.getConnection().rollback();

                        } catch (Exception e1) {

                            throw new JDBCTransactionRollbackException(e1, ex);

                        }

                        throw new RuntimeException("Can't perform operation '"+this.getClass().getSimpleName()+"'.",ex);

                    }

                    if( commitOnFinish ) {

                        this.getConnection().commit();

                    }

                } else {

                    if (this.getConnection().isInTransaction()) {

                        if (!continueTransactionAllowed()) {

                            throw new SQLException("nested operations not allowed.");

                        }

                    }

                    this.getConnection().begin();

                    try {

                        LOGGER.trace("Excuting operation {}.", this.getClass().getSimpleName());

                        result = perform(this.getConnection());

                    } catch (SQLRuntimeException ex) {

                        try {

                            this.getConnection().rollback();

                        } catch (Exception e1) {

                            throw new JDBCTransactionRollbackException(e1, ex);

                        }

                        throw new SQLRuntimeException("Can't perform operation '"+this.getClass().getSimpleName()+"'.",ex);

                    } catch (Exception ex) {

                        try {

                            this.getConnection().rollback();

                        } catch (Exception e1) {

                            throw new JDBCTransactionRollbackException(e1, ex);

                        }

                        throw new RuntimeException("Can't perform operation '"+this.getClass().getSimpleName()+"'.",ex);

                    }

                    this.getConnection().commit();

                }

            } finally {

                closeConnection(result);

            }

        } else {

            try {

                LOGGER.trace("Excuting operation {}.", this.getClass().getSimpleName());

                result = perform(this.getConnection());

            } catch (SQLRuntimeException ex) {

                throw new SQLRuntimeException("Can't perform operation '"+this.getClass().getSimpleName()+"'.",ex);

            } catch (Exception ex) {

                throw new RuntimeException("Can't perform operation '"+this.getClass().getSimpleName()+"'.",ex);

            } finally {

                closeConnection(result);

            }

        }

        return result;

    }

    /**

     * 

     * @param result the result of the operation recently executed

     * @throws Exception

     */

    protected void closeConnection(Object result) throws Exception {

        this.getConnection().close();

        conn = null;

    }

}