Method addRowBatch

java/core/src/java/org/apache/orc/impl/WriterImpl.java:731–800

(VectorizedRowBatch batch)

  }

  @Override
  public void addRowBatch(VectorizedRowBatch batch) throws IOException {
    try {
      // If this is the first set of rows in this stripe, tell the tree writers
      // to prepare the stripe.
      if (batch.size != 0 && rowsInStripe == 0) {
        treeWriter.prepareStripe(stripes.size() + 1);
      }
      if (buildIndex) {
        // Batch the writes up to the rowIndexStride so that we can get the
        // right size indexes.
        int posn = 0;
        while (posn < batch.size) {
          int chunkSize = Math.min(batch.size - posn,
              rowIndexStride - rowsInIndex);
          if (batch.isSelectedInUse()) {
            // find the longest chunk that is continuously selected from posn
            for (int len = 1; len < chunkSize; ++len) {
              if (batch.selected[posn + len] - batch.selected[posn] != len) {
                chunkSize = len;
                break;
              }
            }
            treeWriter.writeRootBatch(batch, batch.selected[posn], chunkSize);
          } else {
            treeWriter.writeRootBatch(batch, posn, chunkSize);
          }
          posn += chunkSize;
          rowsInIndex += chunkSize;
          rowsInStripe += chunkSize;
          if (rowsInIndex >= rowIndexStride) {
            createRowIndexEntry();
          }
        }
      } else {
        if (batch.isSelectedInUse()) {
          int posn = 0;
          while (posn < batch.size) {
            int chunkSize = 1;
            while (posn + chunkSize < batch.size) {
              // find the longest chunk that is continuously selected from posn
              if (batch.selected[posn + chunkSize] - batch.selected[posn] != chunkSize) {
                break;
              }
              ++chunkSize;
            }
            treeWriter.writeRootBatch(batch, batch.selected[posn], chunkSize);
            posn += chunkSize;
          }
        } else {
          treeWriter.writeRootBatch(batch, 0, batch.size);
        }
        rowsInStripe += batch.size;
      }
      rowsSinceCheck += batch.size;
      previousAllocation = memoryManager.checkMemory(previousAllocation, this);
      checkMemory();
    } catch (Throwable t) {