Skip to content

Latest commit

 

History

History
 
 

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
 
 
 
 
 
 

README.md

Google Cloud Dataflow SDK for Java Examples

The examples included in this module serve to demonstrate the basic functionality of Google Cloud Dataflow, and act as starting points for the development of more complex pipelines.

In addition to WordCount, further examples are included. They are organized into "Cookbook" and "Complete" subpackages. The "cookbook" examples show how to define common data analysis patterns when you're building a Dataflow pipeline. The "complete" directory contains some end-to-end examples that tell more complete stories than the patterns in the "cookbook" directory.

WordCount

A good starting point for new users is our set of Word Count examples in the top-level examples directory. The canonical 'word count' task runs over input text file(s) and computes how many times each word occurs in the input. These examples, and an accompanying walkthrough, demonstrate a series of four successively more detailed word count example pipelines that perform this task.

The MinimalWordCount example shows a basic word count Pipeline, and introduces how to read from text files; shows how to Count a Pcollection; basic use of a ParDo; and how to write data to Google Cloud Storage as text files.

The WordCount example shows how to execute a Pipeline both locally and using the Dataflow service; how to use command-line arguments to set pipeline options; and introduces some pipeline design concepts: creating custom PTransforms (composite transforms); and using ParDo with static DoFns defined out-of-line.

The DebuggingWordCount example shows how to log to Cloud Logging, so that your log messages can be viewed from the Dataflow Monitoring UI; controlling Dataflow worker log levels; creating a custom aggregator; and testing your Pipeline via DataflowAssert.

Then, the WindowedWordCount example shows how to run over either unbounded or bounded input collections; how to use a PubSub topic as an input source; how to do windowing, and use data element timestamps; and how to write to BigQuery.

'Cookbook' examples

The 'Cookbook' directory, which shows common and useful patterns, includes the following examples:

  • BigQueryTornadoes — An example that reads the public samples of weather data from Google BigQuery, counts the number of tornadoes that occur in each month, and writes the results to BigQuery. Demonstrates reading/writing BigQuery, counting a PCollection, and user-defined PTransforms.
  • CombinePerKeyExamples — An example that reads the public "Shakespeare" data, and for each word in the dataset that exceeds a given length, generates a string containing the list of play names in which that word appears. Demonstrates the Combine.perKey transform, which lets you combine the values in a key-grouped PCollection.
  • DatastoreWordCount — An example that shows you how to read from Google Cloud Datastore.
  • DeDupExample — An example that uses Shakespeare's plays as plain text files, and removes duplicate lines across all the files. Demonstrates the RemoveDuplicates, TextIO.Read, and TextIO.Write transforms, and how to wire transforms together.
  • FilterExamples — An example that shows different approaches to filtering, including selection and projection. It also shows how to dynamically set parameters by defining and using new pipeline options, and use how to use a value derived by a pipeline. Demonstrates the Mean transform, Options configuration, and using pipeline-derived data as a side input.
  • JoinExamples — An example that shows how to join two collections. It uses a sample of the GDELT "world event" data, joining the event action country code against a table that maps country codes to country names. Demonstrates the Join operation, and using multiple input sources.
  • MaxPerKeyExamples — An example that reads the public samples of weather data from BigQuery, and finds the maximum temperature (mean_temp) for each month. Demonstrates the Max statistical combination transform, and how to find the max-per-key group.

'Complete' examples

The 'Complete' directory contains examples that tell more complete end-to-end stories, and are more like the actual pipelines that you would build than are the 'cookbook' examples. It includes the following examples:

  • AutoComplete — An example that computes the most popular hash tags for every prefix, which can be used for auto-completion. Demonstrates how to use the same pipeline in both streaming and batch, combiners, and composite transforms.
  • StreamingWordExtract — A streaming pipeline example that inputs lines of text from a Cloud Pub/Sub topic, splits each line into individual words, capitalizes those words, and writes the output to a BigQuery table.
  • TfIdf — An example that computes a basic TF-IDF search table for a directory or Cloud Storage prefix. Demonstrates joining data, side inputs, and logging.
  • TopWikipediaSessions — An example that reads Wikipedia edit data from Cloud Storage and computes the user with the longest string of edits separated by no more than an hour within each month. Demonstrates using Cloud Dataflow Windowing to perform time-based aggregations of data.
  • TrafficMaxLaneFlow — A streaming Cloud Dataflow example using BigQuery output in the traffic sensor domain. Demonstrates the Cloud Dataflow streaming runner, sliding windows, Cloud Pub/Sub topic ingestion, the use of the AvroCoder to encode a custom class, and custom Combine transforms.
  • TrafficRoutes — A streaming Cloud Dataflow example using BigQuery output in the traffic sensor domain. Demonstrates the Cloud Dataflow streaming runner, GroupByKey, keyed state, sliding windows, and Cloud Pub/Sub topic ingestion.

Running the Examples

After building and installing the SDK and Examples modules, as explained in this README, you can execute the WordCount and other example pipelines using the DirectPipelineRunner on your local machine:

mvn compile exec:java -pl examples \
-Dexec.mainClass=com.google.cloud.dataflow.examples.WordCount \
-Dexec.args="--inputFile=<INPUT FILE PATTERN> --output=<OUTPUT FILE>"

You can use the BlockingDataflowPipelineRunner to execute the WordCount example on Google Cloud Dataflow Service using managed resources in the Google Cloud Platform. Start by following the general Cloud Dataflow Getting Started instructions. You should have a Google Cloud Platform project that has a Cloud Dataflow API enabled, a Google Cloud Storage bucket that will serve as a staging location, and installed and authenticated Google Cloud SDK. In this case, invoke the example as follows:

mvn compile exec:java -pl examples \
-Dexec.mainClass=com.google.cloud.dataflow.examples.WordCount \
-Dexec.args="--project=<YOUR CLOUD PLATFORM PROJECT ID> \
--stagingLocation=<YOUR CLOUD STORAGE LOCATION> \
--runner=BlockingDataflowPipelineRunner"

Your Cloud Storage location should be entered in the form of gs://bucket/path/to/staging/directory. The Cloud Platform project refers to your project id (not the project number or the descriptive name).

Alternatively, you may choose to bundle all dependencies into a single JAR and execute it outside of the Maven environment. For example, after building and installing as usual, you can execute the following commands to create the bundled JAR of the Examples module and execute it both locally and in Cloud Platform:

mvn package

java -cp examples/target/google-cloud-dataflow-java-examples-all-bundled-manual_build.jar \
com.google.cloud.dataflow.examples.WordCount \
--inputFile=<INPUT FILE PATTERN> --output=<OUTPUT FILE>

java -cp examples/target/google-cloud-dataflow-java-examples-all-bundled-manual_build.jar \
com.google.cloud.dataflow.examples.WordCount \
--project=<YOUR CLOUD PLATFORM PROJECT ID> \
--stagingLocation=<YOUR CLOUD STORAGE LOCATION> \
--runner=BlockingDataflowPipelineRunner

Other examples can be run similarly by replacing the WordCount class path with the example classpath, e.g. com.google.cloud.dataflow.examples.cookbook.BigQueryTornadoes, and adjusting runtime options under the Dexec.args parameter, as specified in the example itself. If you are running the streaming pipeline examples, see the additional setup instruction, below.

Note that when running Maven on Microsoft Windows platform, backslashes (\) under the Dexec.args parameter should be escaped with another backslash. For example, input file pattern of c:\*.txt should be entered as c:\\*.txt.