源码
\hadoop-1.0.0\src\contrib\index
主要涉及的类
org.apache.hadoop.contrib.index.main.UpdateIndex
org.apache.hadoop.contrib.index.mapred.IndexUpdater
org.apache.hadoop.contrib.index.mapred.IndexUpdateMapper
org.apache.hadoop.contrib.index.mapred.IndexUpdatePartitioner
org.apache.hadoop.contrib.index.mapred.IndexUpdateCombiner
org.apache.hadoop.contrib.index.mapred.IndexUpdateReducer
org.apache.hadoop.contrib.index.example.LineDocInputFormat
org.apache.hadoop.contrib.index.mapred.IndexUpdateOutputFormat
org.apache.hadoop.contrib.index.mapred.IndexUpdateConfiguration
解析
1、UpdateIndex类(main实现读取控制台参数)
UpdateIndex是整个程序的入口,提供参数给用户配置。
- -inputPaths -->mapred.input.dir (V)
- -outputPath -->mapred.output.dir (V)
- -shards -->最终每个reduct建立的索引存放hdfs位置 (V)
- -indexPath -->sea.index.path
- -numShards -->sea.num.shards
- -numMapTasks -->mapred.map.tasks (V)
- -conf 添加Configuration xml文件形式的配置方式
shards 通过转换最终保存到Configuration的sea.index.shards
shards参数 与 indexPath和numShards参数 设置一种就可以了。如果没有设置shards,则在main方法中会通过indexPath和numShards生成得到 shards 参数。
从Configuration中获得Updater类: IndexUpdateConfiguration.getIndexUpdaterClass()
public Class<? extends IIndexUpdater> getIndexUpdaterClass() {
return conf.getClass("sea.index.updater", IndexUpdater.class,
IIndexUpdater.class);
}
在UpdateIndex最终调用IndexUpdater
updater.run(conf, inputPaths, outputPath, numMapTasks, shards);
2、IndexUpdater类(配置MapReduce Job)
public void run(Configuration conf, Path[] inputPaths, Path outputPath, int numMapTasks, Shard[] shards) throws IOException { JobConf jobConf = createJob(conf, inputPaths, outputPath, numMapTasks, shards); JobClient.runJob(jobConf); }
JobConf createJob(Configuration conf, Path[] inputPaths, Path outputPath, int numMapTasks, Shard[] shards) throws IOException { setShardGeneration(conf, shards); // iconf.set sets properties in conf IndexUpdateConfiguration iconf = new IndexUpdateConfiguration(conf); Shard.setIndexShards(iconf, shards); // MapTask.MapOutputBuffer uses "io.sort.mb" to decide its max buffer size // (max buffer size = 1/2 * "io.sort.mb"). // Here we half-en "io.sort.mb" because we use the other half memory to // build an intermediate form/index in Combiner. iconf.setIOSortMB(iconf.getIOSortMB() / 2);
setShardGeneration(conf, shards);方法从shard path路径获得segment_N来获得generation,并更新shard实例的gen属性。
Shard.setIndexShards(iconf, shards);则是序列化shards为String写入到Configuration中。
接下来的代码,就是设置InputFormat,OutputFormat, Map, Reduce,Partitioner, KeyValue Class, Combine等一些列的Job必备的参数。
// create the job configuration
JobConf jobConf = new JobConf(conf, IndexUpdater.class);
jobConf.setJobName(this.getClass().getName() + "_"
+ System.currentTimeMillis());
// provided by application
FileInputFormat.setInputPaths(jobConf, inputPaths);
FileOutputFormat.setOutputPath(jobConf, outputPath);
jobConf.setNumMapTasks(numMapTasks);
// already set shards
jobConf.setNumReduceTasks(shards.length);
jobConf.setInputFormat(iconf.getIndexInputFormatClass());
// set by the system
jobConf.setMapOutputKeyClass(IndexUpdateMapper.getMapOutputKeyClass());
jobConf.setMapOutputValueClass(IndexUpdateMapper.getMapOutputValueClass());
jobConf.setOutputKeyClass(IndexUpdateReducer.getOutputKeyClass());
jobConf.setOutputValueClass(IndexUpdateReducer.getOutputValueClass());
jobConf.setMapperClass(IndexUpdateMapper.class);
jobConf.setPartitionerClass(IndexUpdatePartitioner.class);
jobConf.setCombinerClass(IndexUpdateCombiner.class);
jobConf.setReducerClass(IndexUpdateReducer.class);
jobConf.setOutputFormat(IndexUpdateOutputFormat.class);
return jobConf;
3、LineDocInputFormat(数据输入类)
从IndexUpdateConfiguration.getIndexInputFormatClass()可以获得当前默认使用的数据输入类org.apache.hadoop.contrib.index.example.LineDocInputFormat。
这里定义了org.apache.hadoop.contrib.index.example.LineDocRecordReader来解析数据,获取建立索引的命令(Insert, Update, Delete)和对应的数据。
public synchronized boolean next(DocumentID key, LineDocTextAndOp value) throws IOException { if (pos >= end) { return false; } // key is document id, which are bytes until first space if (!readInto(key.getText(), SPACE)) { // 把读到的第一个数据写入Key return false; } // read operation: i/d/u, or ins/del/upd, or insert/delete/update Text opText = new Text(); if (!readInto(opText, SPACE)) { return false; } String opStr = opText.toString(); DocumentAndOp.Op op; if (opStr.equals("i") || opStr.equals("ins") || opStr.equals("insert")) { op = DocumentAndOp.Op.INSERT; } else if (opStr.equals("d") || opStr.equals("del") || opStr.equals("delete")) { op = DocumentAndOp.Op.DELETE; } else if (opStr.equals("u") || opStr.equals("upd") || opStr.equals("update")) { op = DocumentAndOp.Op.UPDATE; } else { // default is insert op = DocumentAndOp.Op.INSERT; } value.setOp(op); if (op == DocumentAndOp.Op.DELETE) { return true; } else { // read rest of the line return readInto(value.getText(), EOL); } }
// 把InputStream offset到分隔符的数据写入到text
private boolean readInto(Text text, char delimiter) throws IOException
2 ins apache
3 ins apache
4 ins apache
0 del
1 upd hadoop
2 del
3 upd hadoop
4、IndexUpdateMapper(最终形成IntermediateForm传递给Combiner和Reducer处理)
· ILocalAnalysis -->org.apache.hadoop.contrib.index.example.LineDocLocalAnalysis
把Text组成为Lucene需要的Document。
· DocumentAndOp
public class DocumentAndOp implements Writable {
private Op op; //
private Document doc; // INSERT UPDATE
private Term term; // DELETE UPDATE
· IntermediateForm
public class IntermediateForm implements Writable {
private final Collection<Term> deleteList;
private RAMDirectory dir;
private IndexWriter writer;
private int numDocs;
public void process(DocumentAndOp doc, Analyzer analyzer) throws IOException {
if (doc.getOp() == DocumentAndOp.Op.DELETE
|| doc.getOp() == DocumentAndOp.Op.UPDATE) {
deleteList.add(doc.getTerm());
}
if (doc.getOp() == DocumentAndOp.Op.INSERT
|| doc.getOp() == DocumentAndOp.Op.UPDATE) {
if (writer == null) {
// analyzer is null because we specify an analyzer with addDocument
writer = createWriter();
}
writer.addDocument(doc.getDocument(), analyzer);
numDocs++;
}
}
public void process(IntermediateForm form) throws IOException {
if (form.deleteList.size() > 0) {
deleteList.addAll(form.deleteList);
}
if (form.dir.sizeInBytes() > 0) {
if (writer == null) {
writer = createWriter();
}
writer.addIndexesNoOptimize(new Directory[] { form.dir });
numDocs++;
}
}
· HashingDistributionPolicy
public class HashingDistributionPolicy implements IDistributionPolicy {
private int numShards;
public void init(Shard[] shards) {
numShards = shards.length;
}
public int chooseShardForInsert(DocumentID key) {
int hashCode = key.hashCode();
return hashCode >= 0 ? hashCode % numShards : (-hashCode) % numShards;
}
public int chooseShardForDelete(DocumentID key) {
int hashCode = key.hashCode();
return hashCode >= 0 ? hashCode % numShards : (-hashCode) % numShards;
}
}
IndexUpdateMapper.map(K, V, OutputCollector<Shard, IntermediateForm>, Reporter)
根据DocumentAndOp形成IntermediateForm,以及DistributionPolicy选择Shard,传入Combiner。
评注: 这里每条数据都要开启和关闭Writer,消耗应该不小。
5、IndexUpdateCombiner
把该节点的IntermediateForm合并成一个大的IntermediateForm集合。
评注: 感觉作用不是很大,合并减少的数据量有限; 但是却又要初始化和关闭一次Writer。
6、IndexUpdateReducer
把从Mapper传递来的IntermediateForm写入到ShardWriter。最终把Lucene索引写入到IndexPath下。
public ShardWriter(FileSystem fs, Shard shard, String tempDir, IndexUpdateConfiguration iconf) throws IOException
在Reducer最后关闭ShardWriter时,由于要等待比较长的时间。这里使用一个新的线程来发送心跳。 使用hadoop/编写mapreduce程序 注意点(! 不要关闭mapreduce的超时)
new Closeable() {
volatile boolean closed = false;
public void close() throws IOException {
// spawn a thread to give progress heartbeats
Thread prog = new Thread() {
public void run() {
while (!closed) {
try {
fReporter.setStatus("closing");
Thread.sleep(1000);
} catch (InterruptedException e) {
continue;
} catch (Throwable e) {
return;
}
}
}
};
try {
prog.start();
if (writer != null) {
writer.close();
}
} finally {
closed = true;
}
}
}.close();
· ShardWriter
把索引写到本地,等处理完了数据后,关闭IndexWriter后把索引库转入拷贝到HDFS持久化。
// move the files created in the temp dir into the perm dir
// and then delete the temp dir from the local FS
private void moveFromTempToPerm() throws IOException {
try {
FileStatus[] fileStatus =
localFs.listStatus(temp, LuceneIndexFileNameFilter.getFilter());
Path segmentsPath = null;
Path segmentsGenPath = null;
// move the files created in temp dir except segments_N and segments.gen
for (int i = 0; i < fileStatus.length; i++) {
Path path = fileStatus[i].getPath();
String name = path.getName();
if (LuceneUtil.isSegmentsGenFile(name)) {
assert (segmentsGenPath == null);
segmentsGenPath = path;
} else if (LuceneUtil.isSegmentsFile(name)) {
assert (segmentsPath == null);
segmentsPath = path;
} else {
fs.completeLocalOutput(new Path(perm, name), path);
}
}
// move the segments_N file
if (segmentsPath != null) {
fs.completeLocalOutput(new Path(perm, segmentsPath.getName()),
segmentsPath);
}
// move the segments.gen file
if (segmentsGenPath != null) {
fs.completeLocalOutput(new Path(perm, segmentsGenPath.getName()),
segmentsGenPath);
}
} finally {
// finally delete the temp dir (files should have been deleted)
localFs.delete(temp);
}
}