test_class.declarator: RAW_SCHEMA =
        "{\"type\": \"record\", \"name\": \"__root__\", \"fields\": [{\"name\": \"f0_\", \"type\":"
            + " [\"null\", \"long\"]}]}"
test_class.modifier: private static final
test_class.original_string: private static final String RAW_SCHEMA =
        "{\"type\": \"record\", \"name\": \"__root__\", \"fields\": [{\"name\": \"f0_\", \"type\":"
            + " [\"null\", \"long\"]}]}";
test_class.type: String
test_class.var_name: RAW_SCHEMA
test_class.declarator: parsedSchema
test_class.modifier: private
test_class.original_string: private Schema parsedSchema;
test_class.type: Schema
test_class.var_name: parsedSchema
test_class.declarator: RESPONSES_123 =
        ImmutableList.of(
            ReadRowsResponse.newBuilder()
                .setAvroRows(
                    AvroRows.newBuilder()
                        .setRowCount(2)
                        .setSerializedBinaryRows(
                            ByteString.copyFrom(new byte[] {2, 2, 2, 4}))) // 1, 2
                .build(),
            ReadRowsResponse.newBuilder()
                .setAvroRows(
                    AvroRows.newBuilder()
                        .setRowCount(1)
                        .setSerializedBinaryRows(ByteString.copyFrom(new byte[] {2, 6}))) // 3
                .build())
test_class.modifier: private static final
test_class.original_string: private static final List<ReadRowsResponse> RESPONSES_123 =
        ImmutableList.of(
            ReadRowsResponse.newBuilder()
                .setAvroRows(
                    AvroRows.newBuilder()
                        .setRowCount(2)
                        .setSerializedBinaryRows(
                            ByteString.copyFrom(new byte[] {2, 2, 2, 4}))) // 1, 2
                .build(),
            ReadRowsResponse.newBuilder()
                .setAvroRows(
                    AvroRows.newBuilder()
                        .setRowCount(1)
                        .setSerializedBinaryRows(ByteString.copyFrom(new byte[] {2, 6}))) // 3
                .build());
test_class.type: List<ReadRowsResponse>
test_class.var_name: RESPONSES_123
test_class.declarator: split = new DirectBigQueryInputSplit("session", RAW_SCHEMA, 5)
test_class.modifier: private
test_class.original_string: private DirectBigQueryInputSplit split = new DirectBigQueryInputSplit("session", RAW_SCHEMA, 5);
test_class.type: DirectBigQueryInputSplit
test_class.var_name: split
test_class.declarator: STREAM = Stream.newBuilder().setName("session").build()
test_class.modifier: private static final
test_class.original_string: private static final Stream STREAM = Stream.newBuilder().setName("session").build();
test_class.type: Stream
test_class.var_name: STREAM
test_class.declarator: bqClient
test_class.modifier: @Mock private
test_class.original_string: @Mock private BigQueryStorageClient bqClient;
test_class.type: BigQueryStorageClient
test_class.var_name: bqClient
test_class.declarator: readRows
test_class.modifier: @Mock private
test_class.original_string: @Mock private ServerStreamingCallable<ReadRowsRequest, ReadRowsResponse> readRows;
test_class.type: ServerStreamingCallable<ReadRowsRequest, ReadRowsResponse>
test_class.var_name: readRows
test_class.declarator: taskContext
test_class.modifier: @Mock private
test_class.original_string: @Mock private TaskAttemptContext taskContext;
test_class.type: TaskAttemptContext
test_class.var_name: taskContext
test_class.declarator: rowsStream
test_class.modifier: @Mock private
test_class.original_string: @Mock private ServerStream<ReadRowsResponse> rowsStream;
test_class.type: ServerStream<ReadRowsResponse>
test_class.var_name: rowsStream
test_class.declarator: reader
test_class.modifier: private
test_class.original_string: private DirectBigQueryRecordReader reader;
test_class.type: DirectBigQueryRecordReader
test_class.var_name: reader
test_class.file: bigquery/src/test/java/com/google/cloud/hadoop/io/bigquery/DirectBigQueryRecordReaderTest.java
test_class.identifier: DirectBigQueryRecordReaderTest
test_case.body: @Test
    public void testInitialize() throws Exception {
      initialize();
    }
test_case.class_method_signature: DirectBigQueryRecordReaderTest.testInitialize()
test_case.full_signature: @Test public void testInitialize()
test_case.identifier: testInitialize
test_case.invocations: initialize
test_case.modifiers: @Test public
test_case.parameters: ()
test_case.return: void
test_case.signature: void testInitialize()
focal_class.declarator: schema
focal_class.modifier: private
focal_class.original_string: private Schema schema;
focal_class.type: Schema
focal_class.var_name: schema
focal_class.declarator: stream
focal_class.modifier: private
focal_class.original_string: private Stream stream;
focal_class.type: Stream
focal_class.var_name: stream
focal_class.declarator: parser = new Parser()
focal_class.modifier: private
focal_class.original_string: private Parser parser = new Parser();
focal_class.type: Parser
focal_class.var_name: parser
focal_class.declarator: current
focal_class.modifier: private
focal_class.original_string: private GenericRecord current;
focal_class.type: GenericRecord
focal_class.var_name: current
focal_class.declarator: finalized
focal_class.modifier: private
focal_class.original_string: private boolean finalized;
focal_class.type: boolean
focal_class.var_name: finalized
focal_class.declarator: limit
focal_class.modifier: private
focal_class.original_string: private long limit;
focal_class.type: long
focal_class.var_name: limit
focal_class.declarator: idx
focal_class.modifier: private
focal_class.original_string: private long idx;
focal_class.type: long
focal_class.var_name: idx
focal_class.declarator: client
focal_class.modifier: private
focal_class.original_string: private BigQueryStorageClient client;
focal_class.type: BigQueryStorageClient
focal_class.var_name: client
focal_class.declarator: responseIterator
focal_class.modifier: private
focal_class.original_string: private Iterator<ReadRowsResponse> responseIterator;
focal_class.type: Iterator<ReadRowsResponse>
focal_class.var_name: responseIterator
focal_class.declarator: recordIterator
focal_class.modifier: private
focal_class.original_string: private Iterator<GenericRecord> recordIterator;
focal_class.type: Iterator<GenericRecord>
focal_class.var_name: recordIterator
focal_class.file: bigquery/src/main/java/com/google/cloud/hadoop/io/bigquery/DirectBigQueryRecordReader.java
focal_class.identifier: DirectBigQueryRecordReader
focal_class.class_method_signature: DirectBigQueryRecordReader.initialize(InputSplit genericSplit, TaskAttemptContext context)
focal_class.full_signature: @Override public void initialize(InputSplit genericSplit, TaskAttemptContext context)
focal_class.identifier: initialize
focal_class.modifiers: @Override public
focal_class.parameters: (InputSplit genericSplit, TaskAttemptContext context)
focal_class.return: void
focal_class.signature: void initialize(InputSplit genericSplit, TaskAttemptContext context)
focal_class.class_method_signature: DirectBigQueryRecordReader.nextKeyValue()
focal_class.full_signature: @Override public boolean nextKeyValue()
focal_class.identifier: nextKeyValue
focal_class.modifiers: @Override public
focal_class.parameters: ()
focal_class.return: boolean
focal_class.signature: boolean nextKeyValue()
focal_class.class_method_signature: DirectBigQueryRecordReader.getCurrentKey()
focal_class.full_signature: @Override public NullWritable getCurrentKey()
focal_class.identifier: getCurrentKey
focal_class.modifiers: @Override public
focal_class.parameters: ()
focal_class.return: NullWritable
focal_class.signature: NullWritable getCurrentKey()
focal_class.class_method_signature: DirectBigQueryRecordReader.getCurrentValue()
focal_class.full_signature: @Override public GenericRecord getCurrentValue()
focal_class.identifier: getCurrentValue
focal_class.modifiers: @Override public
focal_class.parameters: ()
focal_class.return: GenericRecord
focal_class.signature: GenericRecord getCurrentValue()
focal_class.class_method_signature: DirectBigQueryRecordReader.getProgress()
focal_class.full_signature: @Override public float getProgress()
focal_class.identifier: getProgress
focal_class.modifiers: @Override public
focal_class.parameters: ()
focal_class.return: float
focal_class.signature: float getProgress()
focal_class.class_method_signature: DirectBigQueryRecordReader.close()
focal_class.full_signature: @Override public void close()
focal_class.identifier: close
focal_class.modifiers: @Override public
focal_class.parameters: ()
focal_class.return: void
focal_class.signature: void close()
focal_class.class_method_signature: DirectBigQueryRecordReader.getClient(Configuration conf)
focal_class.full_signature: protected BigQueryStorageClient getClient(Configuration conf)
focal_class.identifier: getClient
focal_class.modifiers: protected
focal_class.parameters: (Configuration conf)
focal_class.return: BigQueryStorageClient
focal_class.signature: BigQueryStorageClient getClient(Configuration conf)
focal_class.superclass: extends RecordReader<NullWritable, GenericRecord>
focal_method.body: @Override
    public void initialize(InputSplit genericSplit, TaskAttemptContext context) throws IOException {
      DirectBigQueryInputSplit split = (DirectBigQueryInputSplit) genericSplit;
      schema = parser.parse(checkNotNull(split.getSchema(), "schema"));
  
      stream = Stream.newBuilder().setName(checkNotNull(split.getName(), "name")).build();
      ReadRowsRequest request =
          ReadRowsRequest.newBuilder()
              .setReadPosition(StreamPosition.newBuilder().setStream(stream).build())
              .build();
  
      client = getClient(context.getConfiguration());
      responseIterator = client.readRowsCallable().call(request).iterator();
      recordIterator = Collections.emptyIterator();
  
      limit = split.getLimit();
      idx = 0;
      finalized = false;
    }
focal_method.class_method_signature: DirectBigQueryRecordReader.initialize(InputSplit genericSplit, TaskAttemptContext context)
focal_method.full_signature: @Override public void initialize(InputSplit genericSplit, TaskAttemptContext context)
focal_method.identifier: initialize
focal_method.invocations: parse checkNotNull getSchema build setName newBuilder checkNotNull getName build setReadPosition newBuilder build setStream newBuilder getClient getConfiguration iterator call readRowsCallable emptyIterator getLimit
focal_method.modifiers: @Override public
focal_method.parameters: (InputSplit genericSplit, TaskAttemptContext context)
focal_method.return: void
focal_method.signature: void initialize(InputSplit genericSplit, TaskAttemptContext context)