spark MicroBatchStream 源码

  • 2022-10-20
  • 浏览 (269)

spark MicroBatchStream 代码

文件路径:/sql/catalyst/src/main/java/org/apache/spark/sql/connector/read/streaming/MicroBatchStream.java

/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You under the Apache License, Version 2.0
 * (the "License"); you may not use this file except in compliance with
 * the License.  You may obtain a copy of the License at
 *
 *    http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.apache.spark.sql.connector.read.streaming;

import org.apache.spark.annotation.Evolving;
import org.apache.spark.sql.connector.read.InputPartition;
import org.apache.spark.sql.connector.read.PartitionReader;
import org.apache.spark.sql.connector.read.PartitionReaderFactory;
import org.apache.spark.sql.connector.read.Scan;

/**
 * A {@link SparkDataStream} for streaming queries with micro-batch mode.
 *
 * @since 3.0.0
 */
@Evolving
public interface MicroBatchStream extends SparkDataStream {

  /**
   * Returns the most recent offset available.
   */
  Offset latestOffset();

  /**
   * Returns a list of {@link InputPartition input partitions} given the start and end offsets. Each
   * {@link InputPartition} represents a data split that can be processed by one Spark task. The
   * number of input partitions returned here is the same as the number of RDD partitions this scan
   * outputs.
   * <p>
   * If the {@link Scan} supports filter pushdown, this stream is likely configured with a filter
   * and is responsible for creating splits for that filter, which is not a full scan.
   * </p>
   * <p>
   * This method will be called multiple times, to launch one Spark job for each micro-batch in this
   * data stream.
   * </p>
   */
  InputPartition[] planInputPartitions(Offset start, Offset end);

  /**
   * Returns a factory to create a {@link PartitionReader} for each {@link InputPartition}.
   */
  PartitionReaderFactory createReaderFactory();
}

相关信息

spark 源码目录

相关文章

spark CompositeReadLimit 源码

spark ContinuousPartitionReader 源码

spark ContinuousPartitionReaderFactory 源码

spark ContinuousStream 源码

spark Offset 源码

spark PartitionOffset 源码

spark ReadAllAvailable 源码

spark ReadLimit 源码

spark ReadMaxFiles 源码

spark ReadMaxRows 源码

0  赞