How to use

close

method

in

com.facebook.presto.operator.Driver

  @Override
  public void close()
  {
    driver.close();
    currentPage = null;
  }
}

  @Override
  public void close()
  {
    Driver driver;
    synchronized (this) {
      closed = true;
      driver = this.driver;
    }
    if (driver != null) {
      driver.close();
    }
  }
}

  @Override
  public Boolean call()
  {
    driver.close();
    return true;
  }
});

@Test
public void testBrokenOperatorCloseWhileProcessing()
    throws Exception
{
  BrokenOperator brokenOperator = new BrokenOperator(driverContext.addOperatorContext(0, new PlanNodeId("test"), "source"), false);
  final Driver driver = Driver.createDriver(driverContext, brokenOperator, createSinkOperator(ImmutableList.of()));
  assertSame(driver.getDriverContext(), driverContext);
  // block thread in operator processing
  Future<Boolean> driverProcessFor = executor.submit(new Callable<Boolean>()
  {
    @Override
    public Boolean call()
    {
      return driver.processFor(new Duration(1, TimeUnit.MILLISECONDS)).isDone();
    }
  });
  brokenOperator.waitForLocked();
  driver.close();
  assertTrue(driver.isFinished());
  try {
    driverProcessFor.get(1, TimeUnit.SECONDS);
    fail("Expected InterruptedException");
  }
  catch (ExecutionException e) {
    assertDriverInterrupted(e.getCause());
  }
}

@Test(invocationCount = 1_000, timeOut = 10_000)
public void testConcurrentClose()
{
  List<Type> types = ImmutableList.of(VARCHAR, BIGINT, BIGINT);
  OperatorContext operatorContext = driverContext.addOperatorContext(0, new PlanNodeId("test"), "values");
  ValuesOperator source = new ValuesOperator(operatorContext, rowPagesBuilder(types)
      .addSequencePage(10, 20, 30, 40)
      .build());
  Operator sink = createSinkOperator(types);
  Driver driver = Driver.createDriver(driverContext, source, sink);
  // let these threads race
  scheduledExecutor.submit(() -> driver.processFor(new Duration(1, TimeUnit.NANOSECONDS))); // don't want to call isFinishedInternal in processFor
  scheduledExecutor.submit(() -> driver.close());
  while (!driverContext.isDone()) {
    Uninterruptibles.sleepUninterruptibly(1, TimeUnit.MILLISECONDS);
  }
}

@Test
public void testAbruptFinish()
{
  List<Type> types = ImmutableList.of(VARCHAR, BIGINT, BIGINT);
  ValuesOperator source = new ValuesOperator(driverContext.addOperatorContext(0, new PlanNodeId("test"), "values"), rowPagesBuilder(types)
      .addSequencePage(10, 20, 30, 40)
      .build());
  PageConsumerOperator sink = createSinkOperator(types);
  Driver driver = Driver.createDriver(driverContext, source, sink);
  assertSame(driver.getDriverContext(), driverContext);
  assertFalse(driver.isFinished());
  driver.close();
  assertTrue(driver.isFinished());
  // finish is only called in normal operations
  assertFalse(source.isFinished());
  assertFalse(sink.isFinished());
  // close is always called (values operator doesn't have a closed state)
  assertTrue(sink.isClosed());
}

assertFalse(driver.isFinished());
driver.close();
assertTrue(driver.isFinished());

  @Override
  public void close()
  {
    driver.close();
    currentPage = null;
  }
}

  @Override
  public void close()
  {
    Driver driver;
    synchronized (this) {
      closed = true;
      driver = this.driver;
    }
    if (driver != null) {
      driver.close();
    }
  }
}

  @Override
  public Boolean call()
      throws Exception
  {
    driver.close();
    return true;
  }
});

@Test
public void testBrokenOperatorCloseWhileProcessing()
    throws Exception
{
  BrokenOperator brokenOperator = new BrokenOperator(driverContext.addOperatorContext(0, new PlanNodeId("test"), "source"), false);
  final Driver driver = new Driver(driverContext, brokenOperator, createSinkOperator(brokenOperator));
  assertSame(driver.getDriverContext(), driverContext);
  // block thread in operator processing
  Future<Boolean> driverProcessFor = executor.submit(new Callable<Boolean>()
  {
    @Override
    public Boolean call()
        throws Exception
    {
      return driver.processFor(new Duration(1, TimeUnit.MILLISECONDS)).isDone();
    }
  });
  brokenOperator.waitForLocked();
  driver.close();
  assertTrue(driver.isFinished());
  try {
    driverProcessFor.get(1, TimeUnit.SECONDS);
    fail("Expected InterruptedException");
  }
  catch (ExecutionException e) {
    checkArgument(getRootCause(e) instanceof InterruptedException, "Expected root cause exception to be an instance of InterruptedException");
  }
}

@Test
public void testAbruptFinish()
{
  List<Type> types = ImmutableList.<Type>of(VARCHAR, BIGINT, BIGINT);
  ValuesOperator source = new ValuesOperator(driverContext.addOperatorContext(0, new PlanNodeId("test"), "values"), types, rowPagesBuilder(types)
      .addSequencePage(10, 20, 30, 40)
      .build());
  MaterializingOperator sink = createSinkOperator(source);
  Driver driver = new Driver(driverContext, source, sink);
  assertSame(driver.getDriverContext(), driverContext);
  assertFalse(driver.isFinished());
  driver.close();
  assertTrue(driver.isFinished());
  // finish is only called in normal operations
  assertFalse(source.isFinished());
  assertFalse(sink.isFinished());
  // close is always called (values operator doesn't have a closed state)
  assertTrue(sink.isClosed());
}

assertFalse(driver.isFinished());
driver.close();
assertTrue(driver.isFinished());