Search in sources :

Example 1 with ITaskContext

use of com.qlangtech.tis.fs.ITaskContext in project plugins by qlangtech.

the class HiveFlatTableBuilder method startTask.

@Override
public void startTask(ITableBuildTask dumpTask) {
    final Connection conn = getConnection();
    final DelegatingConnection delegate = new DelegatingConnection(conn) {

        @Override
        public void close() throws SQLException {
            throw new UnsupportedOperationException("in exec phrase close is not supported");
        }
    };
    ITaskContext context = new ITaskContext() {

        @Override
        public Connection getObj() {
            return delegate;
        }
    };
    try {
        dumpTask.process(context);
    } catch (Exception e) {
        throw new RuntimeException(e);
    } finally {
        try {
            conn.close();
        } catch (Exception e) {
        }
    }
}
Also used : ITaskContext(com.qlangtech.tis.fs.ITaskContext) DelegatingConnection(org.apache.commons.dbcp.DelegatingConnection) Connection(java.sql.Connection) DelegatingConnection(org.apache.commons.dbcp.DelegatingConnection) SQLException(java.sql.SQLException)

Example 2 with ITaskContext

use of com.qlangtech.tis.fs.ITaskContext in project tis by qlangtech.

the class MockFlatTableBuilder method startTask.

@Override
public void startTask(ITableBuildTask dumpTask) {
    final Connection conn = null;
    ITaskContext context = new ITaskContext() {

        @Override
        public Connection getObj() {
            return conn;
        }
    };
    try {
        dumpTask.process(context);
    } catch (Exception e) {
        throw new RuntimeException(e);
    }
}
Also used : ITaskContext(com.qlangtech.tis.fs.ITaskContext) Connection(java.sql.Connection)

Aggregations

ITaskContext (com.qlangtech.tis.fs.ITaskContext)2 Connection (java.sql.Connection)2 SQLException (java.sql.SQLException)1 DelegatingConnection (org.apache.commons.dbcp.DelegatingConnection)1