tech.guyi.ipojo.module.h2.datasource.SingleJdbcDataSource Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of easy-h2-jdbc Show documentation

The newest version!

package tech.guyi.ipojo.module.h2.datasource;

import org.h2.jdbcx.JdbcDataSource;

import java.sql.Connection;
import java.sql.SQLException;
import java.util.Properties;

public class SingleJdbcDataSource extends JdbcDataSource {

    private NoCloseConnection connection;

    public void setDBFilePath(String path){
        this.setURL(String.format("jdbc:h2:%s;FILE_LOCK=NO;TRACE_LEVEL_FILE=0;TRACE_LEVEL_SYSTEM_OUT=0",path));
    }

    @Override
    public Connection getConnection() throws SQLException {
        if (connection == null || connection.isClosed()){
            Properties info = new Properties();
            info.setProperty("user", this.getUser());
            info.put("password", this.getPassword());
            connection = new NoCloseConnection(this.getURL(),info);
            connection.setAutoCommit(false);
        }
        return this.connection;
    }

    public void closeConnection() throws SQLException {
        if (this.connection != null){
            this.connection.realClose();
        }
    }
}