Skip to content

Commit

Permalink
clen up examples
Browse files Browse the repository at this point in the history
  • Loading branch information
giwa committed Aug 21, 2014
1 parent 0d30109 commit 24f95db
Show file tree
Hide file tree
Showing 2 changed files with 6 additions and 12 deletions.
8 changes: 3 additions & 5 deletions examples/src/main/python/streaming/network_wordcount.py
Original file line number Diff line number Diff line change
@@ -1,23 +1,21 @@
import sys
from operator import add

from pyspark.conf import SparkConf
from pyspark.streaming.context import StreamingContext
from pyspark.streaming.duration import *

if __name__ == "__main__":
if len(sys.argv) != 3:
print >> sys.stderr, "Usage: wordcount <hostname> <port>"
exit(-1)
conf = SparkConf()
conf.setAppName("PythonStreamingNetworkWordCount")
ssc = StreamingContext(conf=conf, duration=Seconds(1))
ssc = StreamingContext(appName="PythonStreamingNetworkWordCount",
duration=Seconds(1))

lines = ssc.socketTextStream(sys.argv[1], int(sys.argv[2]))
words = lines.flatMap(lambda line: line.split(" "))
mapped_words = words.map(lambda word: (word, 1))
count = mapped_words.reduceByKey(add)

count.pyprint()

ssc.start()
ssc.awaitTermination()
10 changes: 3 additions & 7 deletions examples/src/main/python/streaming/wordcount.py
Original file line number Diff line number Diff line change
@@ -1,24 +1,20 @@
import sys
from operator import add

from pyspark.conf import SparkConf
from pyspark.streaming.context import StreamingContext
from pyspark.streaming.duration import *

if __name__ == "__main__":
if len(sys.argv) != 2:
print >> sys.stderr, "Usage: wordcount <directory>"
exit(-1)
conf = SparkConf()
conf.setAppName("PythonStreamingWordCount")

ssc = StreamingContext(conf=conf, duration=Seconds(1))
ssc = StreamingContext(appName="PythonStreamingWordCount", duration=Seconds(1))

lines = ssc.textFileStream(sys.argv[1])
words = lines.flatMap(lambda line: line.split(" "))
mapped_words = words.map(lambda x: (x, 1))
count = mapped_words.reduceByKey(add)

count = mapped_words.reduceByKey(lambda a, b: a+b)
count.pyprint()

ssc.start()
ssc.awaitTermination()

0 comments on commit 24f95db

Please sign in to comment.