refactor

apache · Jan 8, 2015 · eea16a7 · eea16a7
1 parent f6ce899
commit eea16a7
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/examples/src/main/python/streaming/kafka_wordcount.py b/examples/src/main/python/streaming/kafka_wordcount.py
@@ -45,8 +45,9 @@
     ssc = StreamingContext(sc, 1)
 
     zkQuorum, topic = sys.argv[1:]
-    lines = KafkaUtils.createStream(ssc, zkQuorum, "spark-streaming-consumer", {topic: 1})
-    counts = lines.map(lambda x: x[1]).flatMap(lambda line: line.split(" ")) \
+    kvs = KafkaUtils.createStream(ssc, zkQuorum, "spark-streaming-consumer", {topic: 1})
+    lines = kvs.map(lambda x: x[1])
+    counts = lines.flatMap(lambda line: line.split(" ")) \
         .map(lambda word: (word, 1)) \
         .reduceByKey(lambda a, b: a+b)
     counts.pprint()