Best code snippets using com.facebook.presto.spi.RecordPageSource.getCursor(Showing top 6 results out of 315)

protected static void assertPageSourceType(ConnectorPageSource pageSource, HiveStorageFormat hiveStorageFormat)
{
  if (pageSource instanceof RecordPageSource) {
    RecordCursor hiveRecordCursor = ((RecordPageSource) pageSource).getCursor();
    hiveRecordCursor = ((HiveRecordCursor) hiveRecordCursor).getRegularColumnRecordCursor();
    if (hiveRecordCursor instanceof HiveCoercionRecordCursor) {
      hiveRecordCursor = ((HiveCoercionRecordCursor) hiveRecordCursor).getRegularColumnRecordCursor();
    }
    assertInstanceOf(hiveRecordCursor, recordCursorType(hiveStorageFormat), hiveStorageFormat.name());
  }
  else {
    assertInstanceOf(((HivePageSource) pageSource).getPageSource(), pageSourceType(hiveStorageFormat), hiveStorageFormat.name());
  }
}

@Override
public Page getOutput()
{
  if (split == null) {
    return null;
  }
  if (!finishing && pageSource == null && cursor == null) {
    ConnectorPageSource source = pageSourceProvider.createPageSource(operatorContext.getSession(), split, columns);
    if (source instanceof RecordPageSource) {
      cursor = ((RecordPageSource) source).getCursor();
    }
    else {
      pageSource = source;
    }
  }
  if (pageSource != null) {
    return processPageSource();
  }
  else {
    return processColumnSource();
  }
}

    ImmutableMap.of());
RecordCursor cursor = ((RecordPageSource) pageSource.get()).getCursor();

protected static void assertPageSourceType(ConnectorPageSource pageSource, HiveStorageFormat hiveStorageFormat)
{
  if (pageSource instanceof RecordPageSource) {
    RecordCursor hiveRecordCursor = ((RecordPageSource) pageSource).getCursor();
    hiveRecordCursor = ((HiveRecordCursor) hiveRecordCursor).getRegularColumnRecordCursor();
    if (hiveRecordCursor instanceof HiveCoercionRecordCursor) {
      hiveRecordCursor = ((HiveCoercionRecordCursor) hiveRecordCursor).getRegularColumnRecordCursor();
    }
    assertInstanceOf(hiveRecordCursor, recordCursorType(hiveStorageFormat), hiveStorageFormat.name());
  }
  else {
    assertInstanceOf(((HivePageSource) pageSource).getPageSource(), pageSourceType(hiveStorageFormat), hiveStorageFormat.name());
  }
}

@Override
public Page getOutput()
{
  if (split == null) {
    return null;
  }
  if (!finishing && pageSource == null && cursor == null) {
    ConnectorPageSource source = pageSourceProvider.createPageSource(operatorContext.getSession(), split, columns);
    if (source instanceof RecordPageSource) {
      cursor = ((RecordPageSource) source).getCursor();
    }
    else {
      pageSource = source;
    }
  }
  if (pageSource != null) {
    return processPageSource();
  }
  else {
    return processColumnSource();
  }
}

    ImmutableMap.of());
RecordCursor cursor = ((RecordPageSource) pageSource.get()).getCursor();