MapReduceの例

2284 ワード

テストしやすいので、ここに置いておきます.
バージョン:1.0.4
新しいapiを使用
package test;


import java.io.IOException;
import java.util.Iterator;
import java.util.StringTokenizer;


import org.apache.hadoop.conf.Configuration;
import org.apache.hadoop.fs.Path;
import org.apache.hadoop.io.IntWritable;
import org.apache.hadoop.io.LongWritable;
import org.apache.hadoop.io.Text;
import org.apache.hadoop.mapreduce.Job;
import org.apache.hadoop.mapreduce.Mapper;
import org.apache.hadoop.mapreduce.Reducer;
import org.apache.hadoop.mapreduce.lib.input.FileInputFormat;
import org.apache.hadoop.mapreduce.lib.output.FileOutputFormat;


public class WordCount {
	// mapper  
	public static class WordCountMapper extends
			Mapper<LongWritable, Text, Text, IntWritable> {
		//    
		private final static IntWritable one = new IntWritable(1);
		// key  
		private Text word = new Text();


		protected void map(LongWritable key, Text value, Context context)
				throws java.io.IOException, InterruptedException {
			String line = value.toString();
			StringTokenizer st = new StringTokenizer(line);


			while (st.hasMoreTokens()) {
				word.set(st.nextToken());
				context.write(word, one);
			}
		};
	}


	// reduce  
	public static class WordCountReduce extends
			Reducer<Text, IntWritable, Text, IntWritable> {
		@Override
		protected void reduce(Text key, java.lang.Iterable<IntWritable> arg1,
				Context context) throws IOException, InterruptedException {
			Iterator<IntWritable> iterator = arg1.iterator();
			int sum = 0;
			while (iterator.hasNext()) {
				sum += iterator.next().get();
			}
			context.write(key, new IntWritable(sum));
		};
	}


	public static void main(String[] args) throws Exception {
		//     job
		Configuration conf = new Configuration();
		Job job = new Job(conf, "WordCount");
		job.setJarByClass(WordCount.class);


		//         
		job.setOutputKeyClass(Text.class);
		job.setOutputValueClass(IntWritable.class);


		//   map reduce 
		job.setMapperClass(WordCountMapper.class);
		job.setReducerClass(WordCountReduce.class);


		//        
		FileInputFormat.addInputPath(job, new Path("/tmp/a.txt"));
		FileOutputFormat.setOutputPath(job, new Path("/tmp/output"));


		job.submit();
	}


}