Skip to content

Instantly share code, notes, and snippets.

View kazabubu's full-sized avatar

Or Gerson kazabubu

View GitHub Profile
public class PoolableRegisteredConnectionFactory extends PoolableConnectionFactory {
private ConnectionRegister connectionRegister;
@Override
public Object makeObject() throws Exception {
PoolableConnection conn = (PoolableConnection) super.makeObject();
if (connectionRegister != null) {
connectionRegister.register(conn);
}
return conn;
@kazabubu
kazabubu / ClosingPoolableConnectionsDataSource.java
Last active December 2, 2019 15:23
Killing Tez Submitted Queries
public class ClosingPoolableConnectionsDataSource extends BasicDataSource implements ConnectionRegister {
private Set<PoolableConnection> connections = new HashSet<>();
@Override
public synchronized void register(Connection conn) {
if (PoolableConnection.class.isAssignableFrom(conn.getClass())) {
connections.add((PoolableConnection) conn);
}
}