我是如何利用Hadoop做大规模日志压缩的
背景
刚毕业那几年有幸进入了当时非常热门的某社交网站，在数据平台部从事大数据开发相关的工作。从日志收集、存储、数据仓库建设、数据统计、数据展示都接触了一遍，比较早的赶上了大数据热这波浪潮。虽然今天的人工智能的热度要远高于大数据，但是大家还是不能否定大数据在人工智能中不可取代的地位。
话回正题，当时遇到了一个需要解决的问题就是如何快速对日志进行压缩。那时一天的日志量大概是3TB左右，共100+种日志，最大的一个日志一天要1TB，最小的日志只有几十M。统计需求大部分是用HIVE完成，HIVE中的表每天建立一个分区，每个分区对应一种日志的压缩文件（有天级和小时级）。
当时日志压缩方式是一个日志一个日志进行压缩，利用crontab进行任务并行，效率非常低。经常出现的情况是到了第二天中午12点钟，前一天的日志还没有压缩完，统计需求就没法用hive去做，报表数据就出不来，给我们的压力很大。
也许有小伙伴说，hive可以利用前一天不经过压缩的日志进行统计，后台慢慢进行日志压缩，压缩完成后在重新load一下分区不就ok了吗？这个方案确实可行。但是当时的实际情况是，有好多的表load的都是压缩后的数据，修改成本比较高（几百张表）。不得已还是得尽量缩短压缩时间，这个问题经过我的一番折腾，终于把日志压缩完成时间提前到凌晨1点钟之前，各种报表数据的统计可以在早晨八点钟之前完成。接下来我就把详细做法介绍给大家。
Hadoop并行压缩
压缩格式
首先我们就要讨论压缩格式，我们选择的压缩格式是bz2，原因是bz2算法支持分片压缩合并：即每个小bz2文件头尾相连拼接到一起就是一个大的bz2文件。map/reduce也支持对bz2文件的分块：即利用多个map同时对压缩文件的不同部分进行处理。当时也试过gzip算法，但是gzip没法分片，hive生成的任务只会有一个map，统计效率低下。
压缩方案
如图所示，有两个日志文件A.log和B.log需要压缩，利用map/reduce并行处理这两个日志。假设map/reduce自动为A.log和B.log分别生成3个map任务同时进行压缩，每个map任务读取日志文件的一部分并用bz2算法进行压缩后写入到集群的HDFS中。A.log通过map端压缩生成了3个压缩文件：A.log.1.bz2，A.log.2.bz2，A.log.3.bz2，之后map通过k-v对把<源文件名称，压缩文件名称>发送给reduce，这样相同日志就会分配到同一个reduce上。reduce做的事情很简单，首先根据压缩文件编号从小到大排序，然后从hadoop上读取压缩文件并merge到一起，最后在HDFS上生成一个新的压缩文件。
注：这里每个日志分成1,2,3三个块是为了描述方便，实际上使用的是map处理文件块时文件的偏移量。
存在的问题
- reduce性能瓶颈 这么做之后reduce就成为性能瓶颈了，因为一个日志最终都交给一个reduce进行合并，还是比较慢。解决方案是压缩前的日志不能按天存放，需要按小时存放，这样大日志可以分批次压缩合并到天级别的压缩文件中。由于我们只是保证在第二天及时产生前一天的压缩文件，我们在前一天就可以对已存在的部分日志进行分批压缩，而只在每天零点对前一天最后面几个小时的日志进行压缩合并，缩短延迟。当时我采用每6个小时压缩一次，这样一天的日志分四次压缩完成，每天凌晨只对前一天最后6小时日志压缩，延迟保证在一小时之内。
- 集群流量风暴 这个方案会大量的从HDFS上读写数据，非常容易造成集群流量风暴，导致集群上其它计算任务失败。解决方案是每次读写一定大小的数据后sleep几秒。
- map端读数据优化 我们知道map/reduce默认是按行读取数据并处理，这对于我们来说效率很低。比如一个大的日志可能有几亿条日志，那么就要调用map几亿次，而我们的map只对数据进行压缩，不要求按行传递，最好的方式是按块。解决方案是重写RecordReader类，实现自己的读数据方案。
- 如何让一个Reduce只merge一种日志 如果只按文件名进行reduce路由，就会出现有两种日志都分配到一个reduce上merge的情况。因为选择reduce的时候，默认行为是根据key计算哈希值后对reduce数取模得到编号，这样就有可能两个不同的key的哈希值是相同的。如果两个日志都分配到同一个reduce上，那么排在后面的日志必须等前面的日志merge完之后才能merge，效率不高。解决方案是：设置reduce数为日志种类数，覆写Partitioner类，把日志种类与reduce编号一一对应，这样就能达到所有日志不用排队同时merge的效果。
具体实现
在这里把需要实现的代码简要的列出来，这里面以java版本为例。
- FileInputFormat类：必须自己写一个类继承该类，覆写其createRecordReader方法。这个方法是一个工厂方法，告诉map/reduce需要一个什么样的RecordReader，RecordReader就是map读取数据所用到的类。
-
public class CompressMergeInputFormat extends FileInputFormat<FileAndPos, ByteBuffer> { @Override public RecordReader<FileAndPos, ByteBuffer> createRecordReader( InputSplit split, TaskAttemptContext context) throws IOException, InterruptedException { return new CompressMergeRecordReader(); } /** * 由于数据量较大，默认以8个blockSize作为一个Split分配给一个map。hadoop默认一个blockSize是64M，当日志量太大时会产生很多小压缩文件。 * */ @Override protected long computeSplitSize(long blockSize, long minSize, long maxSize) { return 8 * blockSize; } }
- RecordReader<KEY, VALUE>类：必须自己写一个类继承该类，该类是一个模板类，模板参数分别由Key和Value类型指定，实际上是kv对。map/reduce默认key是当前读取数据在文件中的偏移量，value是内容。我们必须覆写其initialize，nextKeyValue，getCurrentKey，getCurrentValue方法，并且实现KEY和VALUE。
-
public class CompressMergeRecordReader extends RecordReader<FileAndPos, ByteBuffer> { private Path file; private long pos; private long readed = 0; private long length; private FileSystem fs; private FSDataInputStream in; private Configuration config = null; private FileAndPos currentKey = new FileAndPos(); private ByteBuffer currentValue = new ByteBuffer(); @Override public void initialize(InputSplit split, TaskAttemptContext context){ FileSplit filesplit = (FileSplit) split; file = filesplit.getPath(); //获取文件路径 pos = filesplit.getStart(); //获取split块偏移，每个split都会被map压缩成1个单独的文件 length = filesplit.getLength(); currentKey.setFile(file.toString()); currentKey.setPos(pos); config = context.getConfiguration(); fs = file.getFileSystem(context.getConfiguration()); in = fs.open(file); in.seek(pos); } @Override public boolean nextKeyValue(){ //读取下一个k-v if (readed >= length) { return false; } int once = in.read(currentValue.buffer); currentValue.length = once; if (once == -1) { return false; } if(readed + once > length){ //如果大于本文件块则要少读一些 currentValue.length = (int)(length - readed); readed = length; } else { readed += once; }
return true; } @Override public FileAndPos getCurrentKey() throws IOException, InterruptedException { return currentKey; } @Override public ByteBuffer getCurrentValue() throws IOException, InterruptedException { return currentValue; } //... 其它省略
} - 实现KEY：必须是一个Writable对象，实现readFields和write和compareTo方法。我们的key中记录文件路径以及当前数据偏移量。
-
public class FileAndPos implements WritableComparable<FileAndPos> { private String file; private long pos; @Override public void write(DataOutput out) throws IOException { out.writeUTF(file); out.writeLong(pos); } @Override public void readFields(DataInput in) throws IOException { file = in.readUTF(); pos = in.readLong(); } @Override public int compareTo(FileAndPos o) { //需要自己实现比较函数，map会把读到的一批key-value安装这个顺序排序 int filecompare = file.compareTo(o.file); if (filecompare == 0) { if (pos < o.pos) { return -1; } else if (pos > o.pos) { return 1; } else { return 0; } } else { return filecompare; } } //... 省略 }
- 实现VALUE：可以实现成一个固定大小的数组（1M），代表调用一次map函数传递多少数据。
-
public class ByteBuffer { public byte[] buffer = new byte[1024 * 1024]; public int length = 0; }
- 实现Partitioner：当map完成时，在map的cleanup函数中向reduce发送一条kv。Partioner的默认行为是对key计算hash值，根据hash值对reduce数取模得到reduce编号。但是由于我们的key带有文件路径以及偏移信息，直接使用hadoop默认行为会把本应分到同一个reduce上的kv对分配到多个reduce上，造成多个reduce同时写一个文件的问题，所以我们必须重写Partitioner类，对同一个日志的kv对产生固定的reduce编号，这样达到所有种类日志同时merge，并且每个reduce只merge一种日志，不存在一个日志分配到多个reduce上的效果。
-
public class CompressMergePartitioner extends Partitioner<Text, Text> { @Override public int getPartition(Text key, Text value, int numPartitions) { String str = key.toString(); //查表得到日志对应的reduce编号，需要提前做好这张表 return reduceIndex; } }
- 实现OutputFormat 由于我们的reduce是自己写文件，必须阻止reduce自己的默认行为（把value写入一个part_xxx文件），需要覆写OutputFormat类，使其不产生任何输出文件
-
public class CompressMergeOutputFormat<K, V> extends OutputFormat<K, V> { @Override public RecordWriter<K, V> getRecordWriter(TaskAttemptContext context) { return new RecordWriter<K, V>() { public void write(K key, V value) { //啥都不做 } public void close(TaskAttemptContext context) { } }; } //...其它代码忽略
- map实现
-
public class CompressMergeMap extends Mapper<FileAndPos, ByteBuffer, Text, Text> { //...省略 private CompressionOutputStream out; @Override protected void setup(Context context) throws IOException, InterruptedException { String buf = null; Configuration conf = context.getConfiguration(); long pos = ((FileSplit) context.getInputSplit()).getStart(); String path = ((FileSplit) context.getInputSplit()).getPath().toString(); //....省略 } @Override protected void cleanup(Context context) throws IOException, InterruptedException { //根据path和pos生成你的key-value，key中包含日志名以及偏移量，value是这个Split的压缩文件路径 context.write(key,value); } @Override protected void map( FileAndPos key, ByteBuffer value, org.apache.hadoop.mapreduce.Mapper<FileAndPos, ByteBuffer, Text, Text>.Context context) throws IOException, InterruptedException { out.write(value.buffer, 0, value.length); delta += value.length; if(delta >= 5 * 1024 * 1024){ //写5M休息一会 Thread.sleep(10); delta = 0; } } }
- reduce实现
-
public class CompressMergeReduce extends Reducer<Text, Text, NullWritable, NullWritable> { private FileSystem fs; //...省略 @Override protected void reduce( Text key, java.lang.Iterable<Text> values, org.apache.hadoop.mapreduce.Reducer<Text, Text, NullWritable, NullWritable>.Context context) throws IOException, InterruptedException { //根据key中的offset排序，然后merge到最终的bz2文件中 } }
