关于 hadoop reduce 阶段遍历 Iterable 的 2 个“坑”

以前有童鞋问到了这样一个问题：为何我在 reduce 阶段遍历了一次 Iterable 以后，再次遍历的时候，数据都没了呢？可能有童鞋想固然的回答：Iterable 只能单向遍历一次，就这样简单的缘由。。。事实果然如此吗？

仍是用代码说话：

package com.test;

import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;

public class T {

	public static void main(String[] args) {

		// 只要实现了Iterable接口的对象均可以使用for-each循环。
		// Iterable接口只由iterator方法构成，
		// iterator()方法是java.lang.Iterable接口，被Collection继承。
		/*public interface Iterable<T> {
			Iterator<T> iterator();
		}*/
		Iterable<String> iter = new Iterable<String>() {
			public Iterator<String> iterator() {
				List<String> l = new ArrayList<String>();
				l.add("aa");
				l.add("bb");
				l.add("cc");
				return l.iterator();
			}
		};
		for(int count : new int[] {1, 2}){
			for (String item : iter) {
				System.out.println(item);
			}
			System.out.println("---------->> " + count + " END.");
		}
	}
}

结果固然是很正常的完整无误的打印了两遍  Iterable 的值。那到底是什么缘由致使了 reduce 阶段的  Iterable 只能被遍历一次呢？

咱们先看一段测试代码：

测试数据：

a 3
a 4
b 50
b 60
a 70
b 8
a 9

import java.io.IOException;
import java.util.ArrayList;
import java.util.List;

import org.apache.hadoop.conf.Configuration;
import org.apache.hadoop.fs.FileSystem;
import org.apache.hadoop.fs.Path;
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;
import org.apache.hadoop.util.GenericOptionsParser;

public class TestIterable {

	public static class M1 extends Mapper<Object, Text, Text, Text> {
		private Text oKey = new Text();
		private Text oVal = new Text();
		String[] lineArr;

		public void map(Object key, Text value, Context context) throws IOException, InterruptedException {
			lineArr = value.toString().split(" ");
			oKey.set(lineArr[0]);
			oVal.set(lineArr[1]);
			context.write(oKey, oVal);
		}
	}

	public static class R1 extends Reducer<Text, Text, Text, Text> {
		List<String> valList = new ArrayList<String>();
		List<Text> textList = new ArrayList<Text>();
		String strAdd;
		public void reduce(Text key, Iterable<Text> values, Context context) throws IOException,
				InterruptedException {
			valList.clear();
			textList.clear();
			strAdd = "";
			for (Text val : values) {
				valList.add(val.toString());
				textList.add(val);
			}
			
			// 坑之 1 ：为神马输出的全是最后一个值？why？
			for(Text text : textList){
				strAdd += text.toString() + ", ";
			}
			System.out.println(key.toString() + "\t" + strAdd);
			System.out.println(".......................");
			
			// 我这样干呢？对了吗？
			strAdd = "";
			for(String val : valList){
				strAdd += val + ", ";
			}
			System.out.println(key.toString() + "\t" + strAdd);
			System.out.println("----------------------");
			
			// 坑之 2 ：第二次遍历的时候为何获得的都是空？why？
			valList.clear();
			strAdd = "";
			for (Text val : values) {
				valList.add(val.toString());
			}
			for(String val : valList){
				strAdd += val + ", ";
			}
			System.out.println(key.toString() + "\t" + strAdd);
			System.out.println(">>>>>>>>>>>>>>>>>>>>>>");
		}
	}

	public static void main(String[] args) throws Exception {
		Configuration conf = new Configuration();
		conf.set("mapred.job.queue.name", "regular");
		String[] otherArgs = new GenericOptionsParser(conf, args).getRemainingArgs();
		if (otherArgs.length != 2) {
			System.err.println("Usage: wordcount <in> <out>");
			System.exit(2);
		}
		System.out.println("------------------------");
		Job job = new Job(conf, "TestIterable");
		job.setJarByClass(TestIterable.class);
		job.setMapperClass(M1.class);
		job.setReducerClass(R1.class);
		job.setOutputKeyClass(Text.class);
		job.setOutputValueClass(Text.class);
		// 输入输出路径
		FileInputFormat.addInputPath(job, new Path(otherArgs[0]));
		FileSystem.get(conf).delete(new Path(otherArgs[1]), true);
		FileOutputFormat.setOutputPath(job, new Path(otherArgs[1]));
		System.exit(job.waitForCompletion(true) ? 0 : 1);
	}
}

在 Eclipse 控制台中的结果以下：

a	9, 9, 9, 9, 
.......................
a	3, 4, 70, 9, 
----------------------
a	
>>>>>>>>>>>>>>>>>>>>>>
b	8, 8, 8, 
.......................
b	50, 60, 8, 
----------------------
b	
>>>>>>>>>>>>>>>>>>>>>>

关于第 1 个坑：对象重用（ objects reuse ）

reduce方法的javadoc中已经说明了会出现的问题： 

The framework calls this method for each <key, (list of values)> pair in the grouped inputs. Output values must be of the same type as input values. Input keys must not be altered. The framework will reuse the key and value objects that are passed into the reduce, therefore the application should clone the objects they want to keep a copy of.

      也就是说虽然reduce方法会反复执行屡次，但key和value相关的对象只有两个，reduce会反复重用这两个对象。因此若是要保存key或者value的结果，只能将其中的值取出另存或者从新clone一个对象（例如Text store = new Text(value) 或者 String a = value.toString()），而不能直接赋引用。由于引用从始至终都是指向同一个对象，你若是直接保存它们，那最后它们都指向最后一个输入记录。会影响最终计算结果而出错。 

看到这里，我想你会恍然大悟：这不是刚毕业找工做，面试官常问的问题：String 是不可变对象但为何能相加呢？为何字符串相加不提倡用 String，而用 StringBuilder ？若是你还不清楚这个问题怎么回答，建议你看看这篇《深刻理解 String, StringBuffer 与 StringBuilder 的区别》http://my.oschina.net/leejun2005/blog/102377

关于第 2 个坑：http://stackoverflow.com/questions/6111248/iterate-twice-on-values

The Iterator you receive from that Iterable's iterator() method is special. The values may not all be in memory; Hadoop may be streaming them from disk. They aren't really backed by a Collection, so it's nontrivial to allow multiple iterations.

最后想说明的是：hadoop 框架的做者们真的是考虑很周全，在 hadoop 框架中，不只有对象重用，还有 JVM 重用等，节约一切能够节约的资源，提升一切能够提升的性能。由于在这种海量数据处理的场景下，性能优化是很是重要的，你可能处理100条数据体现不出性能差异，可是你面对的是千亿、万亿级别的数据呢？

PS：

个人代码是在 Eclipse 中远程调试的，因此 reduce 是没有写 hdfs 的，直接在 eclipse 终端上能够看到结果，很方便，关于怎么在 windows 上远程调试 hadoop，请参考这里 《实战 windows7 下 eclipse 远程调试 linux hadoop》http://my.oschina.net/leejun2005/blog/122775

REF：

hadoop中迭代器的对象重用问题

http://paddy-w.iteye.com/blog/1514595

关于 hadoop 中 JVM 重用和对象重用的介绍

http://wikidoop.com/wiki/Hadoop/MapReduce/Reducer