我们还是看下这个图,mapper处理后的中间数据经过shuffle阶段再由reducer处理。在shuffle阶段会进行分区,分组,排序,二次排序。这是个比较复杂的过程,但是我们理解以下这些东西对于工作中常见业务的开发就够用了:
select class,count(student) from tablename group by class
最终就是翻译成了这里要说的分组,而tablename表中总的数据就可以理解成这个partition的数据。可以说分区是为了并发处理提高效率,分组则是为了服务业务的开发。我在新浪处理微博数据的时候有个需求很简单:
两个数据来源:
需要得到在全量imei中存在的当天活跃imei
如果用hive很简单:select a.imei from all_imei a join active_imei b where a.imei=b,imei
就是一个join操作,下面说下用mapreduce(reducer端join)来实现:
两个数据来源需要对每个来源做个标记,所以key值不能单单是imei,需要是imei+标记,定义他们的封装类:
public class TextPair implements WritableComparable
{
private Text first;
private Text second;
public TextPair()
{
set(new Text(), new Text());
}
public TextPair(String first, String second)
{
this.set(new Text(first), new Text(second));
}
public void set(Text first, Text second)
{
this.first = first;
this.second = second;
}
public void readFields(DataInput in) throws IOException
{
first.readFields(in);
second.readFields(in);
}
public void write(DataOutput out) throws IOException
{
first.write(out);
second.write(out);
}
@Override
public int hashCode()
{
return first.hashCode() * 163 + second.hashCode();
}
@Override
public String toString()
{
return first + "\t" + second;
}
public Text getFirst()
{
return first;
}
public Text getSecond()
{
return second;
}
}
first是imei,second是标记
处理全量imei的map方法:
map(key,value,context){
context.write(new TextPair(imei,"0"),new Text("WEIBO"))
}
标记是0
处理当天活跃imei的map方法:
map(key,value,context){
context.write(new TextPair(imei,"1"),new Text(""))
}
标记是1
分区:就根据imei进行分区,也就是TextPair中的first
public class TextPairKeyPartitioner extends Partitioner
public int getPartition(TextPair key, Text value, int numPartitions){
return (key.getFirst().hashCode() & Integer.MAX_VALUE) % numPartitions;
}
}
分组:注意分组也要根据imei,而不是根据整个TextPair分组,这样两个数据来源也就是两个map来的两部分数据相同的imei分到一个组
` public static class FirstComparator extends WritableComparator
{
private static final Text.Comparator TEXT_COMPARATOR = new Text.Comparator();
protected FirstComparator()
{
super(TextPair.class);
}
@Override
public int compare(byte[] b1, int s1, int l1, byte[] b2, int s2, int l2)
{
try
{
int firstL1 = WritableUtils.decodeVIntSize(b1[s1]) + readVInt(b1, s1);
int firstL2 = WritableUtils.decodeVIntSize(b2[s2]) + readVInt(b2, s2);
return TEXT_COMPARATOR.compareBytes(b1, s1, firstL1, b2, s2, firstL2);
} catch (IOException e)
{
e.printStackTrace();
}
return 0;
}
@Override
public int compare(WritableComparable a, WritableComparable b)
{
if (a instanceof TextPair && b instanceof TextPair)
{
return ((TextPair) a).getFirst().compareTo(((TextPair) b).getFirst());
}
return super.compare(a, b);
}
}`
重点是两个map的数据到reducer的处理:
public void reduce(TextPair key, Iterable values, Context context) throws IOException, InterruptedException{
String imeiKey = key.getFirst().toString();
String second = key.getSecond().toString();
if(second.equals("1")){
return;
}
String lines = "";
String weekInfo = "";
java.util.Iterator it = values.iterator();
while(it.hasNext()){
lines = it.next().toString();
second = key.getSecond().toString();
if(second.equals("0")){
weekInfo = lines;
}else if(second.equals("1")){
//context.write(new Text(imeiKey+"\t"+weekInfo), NullWritable.get());
String [] line = new String[1];
line[0]=imeiKey+"\t"+weekInfo;
context.write(NullWritable.get(),Tools.rcfileDeal(line));
return;
}
}
}
驱动函数中的方法:
Job job = new Job(conf);
job.setJarByClass(CalcImeiWeekKeepMainV2.class);
//设置分组
job.setGroupingComparatorClass(TextPair.FirstComparator.class);
//设置分区
job.setPartitionerClass(TextPairKeyPartitioner.class);
//全量imei处理mapper
MultipleInputs.addInputPath(job, new Path(input_imei_info), RCFileInputFormat.class, ImeiWeekKeepMapperV2.class);
//当天活跃imei处理mapper
MultipleInputs.addInputPath(job,new Path(input_bhv),RCFileInputFormat.class,ImeiBhvWeekKeepMapper.class);
job.setMapOutputKeyClass(TextPair.class);
job.setMapOutputValueClass(Text.class);
**job.setReducerClass(CalcImeiWeekKeepDetailReducer.class);
RCFileOutputFormat.setOutputPath(job, new** Path(output_mid_path_day));
job.setOutputFormatClass(RCFileOutputFormat.class);
job.setNumReduceTasks(reduceNum[0]);
code = job.waitForCompletion(true) ? 0 : 1;
比如:
全量的数据有两个字段:
imei wm
a weino
b weibo
c weibo
mapper输出:
(<”a”,”0”>,”weibo”)
(<”b”,”0”>,”weibo”)
(<”c”,”0”>,”weibo”)
当天活跃imei两个个字段:
imei
a
b
mapper输出:
(<”a”,”1”>,”“)
(<”b”,”1”>,”“)
(<”d”,”1”>,”“)
分区根据imei来分区,加入上面mapper输出的5条数据全部到了一个分区:
(<”a”,”0”>,”weibo”)
(<”b”,”0”>,”weibo”)
(<”c”,”0”>,”weibo”)
(<”a”,”1”>,”“)
(<”b”,”1”>,”“)
因为也是根据imei分组,这5条数据到reduce方法时候分为了4组:
(<”a”,”0”>,”weibo”)
(<”a”,”1”>,”“)
(<”b”,”0”>,”weibo”)
(<”b”,”1”>,”“)
(<”c”,”0”>,”weibo”)
(<”d”,”1”>,”“)
第一组在reduce方法中:key是(<”a”,”0”>,<”a”,”1”>) 值value是{“weibo” ,”“},值是个列表很好理解,其实key也可以理解为一个列表
当value第一次执行value.next()方法获取列表中第一个值时,key的getSecond方法返回的是0,第二次执行value.next()方法时,key的getSecond方法返回的是1。当然我们也可以重写排序方法,1在前面0在后面。
最终利用分组和二次排序实现reducer端的join操作