martinkindall / nyc-taxi-limousine-data-spark Goto Github PK

This is a project where I consume streaming data from Kafka topic A, transform it and write the results to either other topics or to console. I used both Streaming and Structured Streaming API's.

This data was gathered from NYC Taxi Dataflow Codelab.

The available data in topic A is in json format and has the following schema:

• ride_id:string
• point_idx:integer
• latitude:float
• longitude:float
• timestamp:timestamp
• meter_reading:float
• meter_increment:float
• ride_status:string
• passenger_count:integer

There are mainly two ways for working with Streams in Spark:

• Spark Streaming
• Spark Structured Streaming

In this project both options are used, that's why there are two classes that execute operations on the kafka topic: KafkaStreaming.scala and KafkaStructuredStreaming.scala. Some transformations are done in the first class and more complex transformations that require Event-Time Windows and Session Windows are done with the latter.

The transformation is done in TaxiOperations.parseDStreamTaxiCountRides(). The output data has the following schema (JSON):

• key:string
• latitude:float
• longitude:float
• ridesCount:integer

Where ridesCount is the total amount of taxi rides that occurred within cells of 500m² across New York, in a window of time of 1 second. Latitude and Longitude are the center of those cells.

This output is going to be useful for displaying a heatmap of the taxi rides in NY.

The transformation is done in TaxiOperations.parseDStreamTaxiSumIncrements(). The output data has the following schema (JSON):

• dollar_per_minute:float

dollar_per_minute is the sum of each meter_increment field in every taxi ride during the last 60 seconds, computed each 3 seconds.

Previous transformation are done in process-time and are not taking into consideration late arriving data. Briefly explained, remember that our events contain a timestamp which is created before it is sent over the network to the Kafka Topic. Then, there is no guarantee that the events are going to arrive ordered by timestamp! After all, the network is unpredictable. It is common to receive events that occured earlier than previously processed events in our spark stream. This is were event-time processing comes into play. It is clearly explained in the official docs.

Check the function TaxiStructuredOperations.toSumIncrementsEventTime() for the implementation.

Previous output was not considering event time, thus it may not lead to accurrate results. Each taxi ride has a timestamp, which we will consider as the event time. Now we can use accurate time windows and receive data out of order, because Spark Streaming has built-in methods to handle this. We can specify a delayThreshold of 60 seconds for late data and window of 60 seconds with a slide duration of 10 seconds.

The output data has the following schema (JSON):

• window: { start: string, end: string }
• sum(meter_increment): float

Check the function TaxiStructuredOperations.toSessionWindowPerRide() for the implementation.

We can display on a map the busiest pickup locations by considerig only the pickup latitude and longitude of a taxi ride, for as long as it is active. Sessions are very useful in this case, because they can keep track of a computation until a condition is met (usually a timeout). Thus, we will consider a taxi ride to be active until we do not receive any event of that taxi ride for a certain period of time. In this example, ride_id is going to be our session_id.

Here's an example of mapGroupsWithState, which is a key function that allows us to combine different events on a session for calculations.

The output of the data is a dataframe printed to console with the following schema:

sessionId | latitude | longitude | startTimestamp | durationMs | expired