spark技术分享FileStreamSource
FileStreamSource is a Source that reads text files from path directory as they appear. It uses LongOffset offsets.
|
Note
|
It is used by DataSource.createSource for FileFormat.
|
You can provide the schema of the data and dataFrameBuilder – the function to build a DataFrame in getBatch at instantiation time.
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
// NOTE The source directory must exist // mkdir text-logs val df = spark.readStream .format("text") .option("maxFilesPerTrigger", 1) .load("text-logs") scala> df.printSchema root |-- value: string (nullable = true) |
Batches are indexed.
It lives in org.apache.spark.sql.execution.streaming package.
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
import org.apache.spark.sql.types._ val schema = StructType( StructField("id", LongType, nullable = false) :: StructField("name", StringType, nullable = false) :: StructField("score", DoubleType, nullable = false) :: Nil) // You should have input-json directory available val in = spark.readStream .format("json") .schema(schema) .load("input-json") val input = in.transform { ds => println("transform executed") // <-- it's going to be executed once only ds } scala> input.isStreaming res9: Boolean = true |
It tracks already-processed files in seenFiles hash map.
|
Tip
|
Enable Add the following line to
Refer to Logging. |
Options
maxFilesPerTrigger
maxFilesPerTrigger option specifies the maximum number of files per trigger (batch). It limits the file stream source to read the maxFilesPerTrigger number of files specified at a time and hence enables rate limiting.
It allows for a static set of files be used like a stream for testing as the file set is processed maxFilesPerTrigger number of files at a time.
schema
If the schema is specified at instantiation time (using optional dataSchema constructor parameter) it is returned.
Otherwise, fetchAllFiles internal method is called to list all the files in a directory.
When there is at least one file the schema is calculated using dataFrameBuilder constructor parameter function. Else, an IllegalArgumentException("No schema specified") is thrown unless it is for text provider (as providerName constructor parameter) where the default schema with a single value column of type StringType is assumed.
|
Note
|
text as the value of providerName constructor parameter denotes text file stream provider.
|
getOffset Method
The maximum offset (getOffset) is calculated by fetching all the files in path excluding files that start with _ (underscore).
When computing the maximum offset using getOffset, you should see the following DEBUG message in the logs:
|
1 2 3 4 5 |
DEBUG Listed ${files.size} in ${(endTime.toDouble - startTime) / 1000000}ms |
When computing the maximum offset using getOffset, it also filters out the files that were already seen (tracked in seenFiles internal registry).
You should see the following DEBUG message in the logs (depending on the status of a file):
|
1 2 3 4 5 6 7 |
new file: $file // or old file: $file |
Generating DataFrame for Streaming Batch — getBatch Method
FileStreamSource.getBatch asks metadataLog for the batch.
You should see the following INFO and DEBUG messages in the logs:
|
1 2 3 4 5 6 |
INFO Processing ${files.length} files from ${startId + 1}:$endId DEBUG Streaming ${files.mkString(", ")} |
The method to create a result batch is given at instantiation time (as dataFrameBuilder constructor parameter).
