Killing TEZ Queries

PoolableRegisteredConnectionFactory.java

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;
}

@Override
public void destroyObject(Object obj) throws Exception {
  if (obj instanceof PoolableConnection) {
    ((PoolableConnection) obj).reallyClose();

    if (connectionRegister != null) {
      connectionRegister.unregister((PoolableConnection) obj);
    }
  }
}