Application: gvSIG desktop

/*

 * To change this license header, choose License Headers in Project Properties.

 * To change this template file, choose Tools | Templates

 * and open the template in the editor.

 */

package org.gvsig.sqlite.dal;

import java.sql.Connection;

import java.sql.PreparedStatement;

import java.sql.SQLException;

import java.sql.Statement;

import java.util.HashMap;

import java.util.Map;

import org.gvsig.fmap.dal.exception.DataException;

import org.gvsig.fmap.dal.spi.DataTransactionServices;

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

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

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

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

import static org.gvsig.sqlite.dal.SQLiteJDBCConnection.LOGGER;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

/**

 *

 * @author fdiaz

 */

public class SQLiteJDBCConnection extends JDBCConnectionBase {

    public static Logger LOGGER = LoggerFactory.getLogger(SQLiteJDBCConnection.class);

    private static Map<String, SQLiteJDBCConnection> connections = new HashMap<>();

//    private static Stack transactions = new Stack();

//    private int references = 0;

    public static JDBCConnection create(SQLiteHelper helper) throws SQLException {

        SQLiteConnectionParameters connectionParameters = helper.getConnectionParameters();

//            

        String key = helper.getConnectionProviderKey(connectionParameters);

//        SQLiteJDBCConnection conn = connections.getOrDefault(key, null);

//        if(conn != null){

//            conn.references++;

//            LOGGER.info("references = "+conn.references+" trans "+conn.isInTransaction());

//            return conn;

//        }

        SQLiteJDBCConnection conn = new SQLiteJDBCConnection(

                helper.getTransaction(),

                helper.getConnectionProvider().getConnection(),

                key

        );

        connections.put(key, conn);

//        LOGGER.info("new connection references = "+conn.references+" trans "+conn.isInTransaction());

        return conn;

    }

    public SQLiteJDBCConnection(DataTransactionServices transaction, Connection conn, String id) {

        super(transaction, conn, id);

        LOGGER.info("["+JDBCUtils.getConnId(conn)+"] createConnection() "+isInDataTransaction());

    }

    public SQLiteJDBCConnection(DataTransactionServices transaction,Connection conn, String id, ConnectionProvider connectionProvider) {

        super(transaction, conn, id, connectionProvider);

        LOGGER.info("["+JDBCUtils.getConnId(conn)+"] createConnection() "+isInDataTransaction());

    }

    @Override

    public void close() throws SQLException {

        if( this.transaction==null ) {

            LOGGER.info("["+JDBCUtils.getConnId(conn)+"] closeConnection() "+isInDataTransaction());

        }

//        this.references--;

//        LOGGER.info("Close references = "+this.references+" trans "+this.isInTransaction());

//        if(this.references <= 0){

//            super.close();

//            connections.remove(this.getId());

//        } else {

//            this.abort();

//        }

        super.close();

    }

    @Override

    public synchronized void begin() throws SQLException {

        if(!isInTransaction()){

            LOGGER.info("["+JDBCUtils.getConnId(conn)+"] begin() "+isInDataTransaction());

        }

//        if(!transactions.isEmpty()){

//            Object mon = new Object();

//            synchronized(mon){

//                try {

//                    transactions.push(mon);

//                    mon.wait();

//                } catch (InterruptedException ex) {

//                    throw new RuntimeException("", ex);

//                }

//            }

//        }

        super.begin();

    }

    @Override

    public synchronized void commit() throws SQLException {

        if( !(this.transaction!=null && this.transaction.isInProgress() )) {

            LOGGER.info("["+JDBCUtils.getConnId(conn)+"] commit() "+isInDataTransaction());

        }

//        if(!transactions.isEmpty()){

//            Object mon = transactions.pop();

//            mon.notify();

//            

//        }

        super.commit();

    }

    @Override

    public void abort() throws DataException {

        LOGGER.info("["+JDBCUtils.getConnId(conn)+"] abort() "+isInDataTransaction());

        super.abort();

    }

    @Override

    public void finish() throws DataException {

        LOGGER.info("["+JDBCUtils.getConnId(conn)+"] finish() "+isInDataTransaction());

        super.finish();

    }

    @Override

    public void execute(String sql) throws SQLException {

        LOGGER.info("["+JDBCUtils.getConnId(conn)+"] execute("+sql+") "+isInDataTransaction());

        super.execute(sql); 

    }

    @Override

    public PreparedStatement prepareStatement(String sql) throws SQLException {

        LOGGER.info("["+JDBCUtils.getConnId(conn)+"] prepareStatement("+sql+") "+isInDataTransaction());

        return super.prepareStatement(sql);

    }

    @Override

    public void rollback() throws SQLException {

        LOGGER.info("["+JDBCUtils.getConnId(conn)+"] rollback() "+isInDataTransaction());

        super.rollback();

    }

    @Override

    public Statement createStatement() throws SQLException {

        LOGGER.info("["+JDBCUtils.getConnId(conn)+"] createStatement() "+isInDataTransaction());

        return super.createStatement();

    }

    @Override

    public Statement createStatement(String sql) throws SQLException {

        LOGGER.info("["+JDBCUtils.getConnId(conn)+"] createStatement("+sql+") "+isInDataTransaction());

        return super.createStatement(sql);

    }

}