public SpimiIndexer(ILexer lexer)
{
this.lexer = lexer;
this.blockReader = new SpimiBlockReader();
this.blockWriter = new SpimiBlockWriter();
this.fileIndexWriter = new FileIndexWriter();
}
public void TestMerge()
{
List<IEnumerator<PostingList>> postingLists
= new List<IEnumerator<PostingList>>();
SpimiBlockReader reader = new SpimiBlockReader();
SpimiBlockWriter writer = new SpimiBlockWriter();
writer.AddPosting("aTerm", "aDoc");
writer.AddPosting("aTerm", "bDoc");
writer.AddPosting("bTerm", "aDoc");
postingLists.Add(reader.Read(writer.FlushToFile()).GetEnumerator());
writer = new SpimiBlockWriter();
writer.AddPosting("cTerm", "dDoc");
writer.AddPosting("aTerm", "zDoc");
writer.AddPosting("bTerm", "aDoc");
postingLists.Add(reader.Read(writer.FlushToFile()).GetEnumerator());
using (IEnumerator<PostingList> merged = reader.BeginBlockMerge(postingLists).GetEnumerator())
{
Assert.IsTrue(merged.MoveNext());
PostingList postingList = merged.Current;
Assert.AreEqual("aTerm", postingList.Term);
Assert.AreEqual(3, postingList.Postings.Count);
Assert.AreEqual("aDoc", postingList.Postings[0]);
Assert.AreEqual("bDoc", postingList.Postings[1]);
Assert.AreEqual("zDoc", postingList.Postings[2]);
Assert.IsTrue(merged.MoveNext());
postingList = merged.Current;
Assert.AreEqual("bTerm", postingList.Term);
Assert.AreEqual(1, postingList.Postings.Count);
Assert.AreEqual("aDoc", postingList.Postings[0]);
Assert.IsTrue(merged.MoveNext());
postingList = merged.Current;
Assert.AreEqual("cTerm", postingList.Term);
Assert.AreEqual(1, postingList.Postings.Count);
Assert.AreEqual("dDoc", postingList.Postings[0]);
Assert.IsFalse(merged.MoveNext());
}
}
public void TestRead()
{
SpimiBlockWriter writer = new SpimiBlockWriter();
writer.AddPosting("aTerm", "aDoc");
writer.AddPosting("aTerm", "bDoc");
writer.AddPosting("bTerm", "aDoc");
string filePath = writer.FlushToFile();
SpimiBlockReader reader = new SpimiBlockReader();
IEnumerable<PostingList> postingLists = reader.Read(filePath);
IEnumerator<PostingList> enumerator = postingLists.GetEnumerator();
Assert.AreEqual(true, enumerator.MoveNext());
Assert.AreEqual("aTerm", enumerator.Current.Term);
Assert.AreEqual("aDoc", enumerator.Current.Postings[0]);
Assert.AreEqual("bDoc", enumerator.Current.Postings[1]);
Assert.AreEqual(true, enumerator.MoveNext());
Assert.AreEqual("bTerm", enumerator.Current.Term);
Assert.AreEqual("aDoc", enumerator.Current.Postings[0]);
}
public void TestRead()
{
SpimiBlockWriter writer = new SpimiBlockWriter();
writer.AddPosting("aTerm", DocA);
writer.AddPosting("aTerm", DocB);
writer.AddPosting("bTerm", DocA);
string filePath = writer.FlushToFile();
SpimiBlockReader reader = new SpimiBlockReader();
IEnumerator<PostingList> postingLists = reader.Read(filePath).GetEnumerator();
Assert.AreEqual(true, postingLists.MoveNext());
Assert.AreEqual("aTerm", postingLists.Current.Term);
Assert.AreEqual(2, postingLists.Current.Postings.Count);
Assert.AreEqual(new Posting(DocA, 1), postingLists.Current.Postings[0]);
Assert.AreEqual(new Posting(DocB, 1), postingLists.Current.Postings[1]);
Assert.AreEqual(true, postingLists.MoveNext());
Assert.AreEqual("bTerm", postingLists.Current.Term);
Assert.AreEqual(new Posting(DocA, 1), postingLists.Current.Postings[0]);
}
private void MergeBlocks()
{
if (termIndexBlockWriter.Postings > 0)
FlushBlockWriter();
using (FileIndexWriter<string, IList<Posting>> writer = new FileIndexWriter<string, IList<Posting>>(
new StringEncoder(),
new PostingListEncoder(), indexStream))
{
SpimiBlockReader blockReader = new SpimiBlockReader();
List<IEnumerator<PostingList>> openedBlocks = blockReader.OpenBlocks(this.termIndexBlockFilePaths);
foreach (PostingList postingList in blockReader.BeginBlockMerge(openedBlocks))
{
writer.Add(postingList.Term, postingList.Postings);
}
writer.WriteOut();
}
}
