WordCount: WordCount.java

File WordCount.java, 1.9 KB (added by jazz, 14 years ago)
Line 
1import java.io.IOException;
2import java.util.*;
3       
4import org.apache.hadoop.fs.Path;
5import org.apache.hadoop.conf.*;
6import org.apache.hadoop.io.*;
7import org.apache.hadoop.mapred.*;
8import org.apache.hadoop.util.*;
9       
10public class WordCount {
11       
12 public static class Map extends MapReduceBase implements Mapper<LongWritable, Text, Text, IntWritable> {
13    private final static IntWritable one = new IntWritable(1);
14    private Text word = new Text();
15       
16    public void map(LongWritable key, Text value, OutputCollector<Text, IntWritable> output, Reporter reporter) throws IOException {
17        String line = value.toString();
18        StringTokenizer tokenizer = new StringTokenizer(line);
19        while (tokenizer.hasMoreTokens()) {
20            word.set(tokenizer.nextToken());
21            output.collect(word, one);
22        }
23    }
24 } 
25       
26 public static class Reduce extends MapReduceBase implements Reducer<Text, IntWritable, Text, IntWritable> {
27
28    public void reduce(Text key, Iterator<IntWritable> values, OutputCollector<Text, IntWritable> output, Reporter reporter) throws IOException {
29        int sum = 0;
30        while (values.hasNext()) {
31            sum += values.next().get();
32        }
33        output.collect(key, new IntWritable(sum));
34    }
35 }
36       
37 public static void main(String[] args) throws Exception {
38    JobConf conf = new JobConf(WordCount.class);
39    conf.setJobName("wordcount");
40       
41    conf.setOutputKeyClass(Text.class);
42    conf.setOutputValueClass(IntWritable.class);
43       
44    conf.setMapperClass(Map.class);
45    conf.setCombinerClass(Reduce.class);
46    conf.setReducerClass(Reduce.class);
47       
48    conf.setInputFormat(TextInputFormat.class);
49    conf.setOutputFormat(TextOutputFormat.class);
50       
51    FileInputFormat.setInputPaths(conf, new Path(args[0]));
52    FileOutputFormat.setOutputPath(conf, new Path(args[1]));
53       
54    JobClient.runJob(conf);
55 }
56       
57}