| using System.Diagnostics; |
| |
| namespace Lucene.Net.Codecs.Lucene3x |
| { |
| using System; |
| using System.IO; |
| using BytesRef = Lucene.Net.Util.BytesRef; |
| using CharsRef = Lucene.Net.Util.CharsRef; |
| using Directory = Lucene.Net.Store.Directory; |
| |
| /* |
| * Licensed to the Apache Software Foundation (ASF) under one or more |
| * contributor license agreements. See the NOTICE file distributed with |
| * this work for additional information regarding copyright ownership. |
| * The ASF licenses this file to You under the Apache License, Version 2.0 |
| * (the "License"); you may not use this file except in compliance with |
| * the License. You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| using FieldInfos = Lucene.Net.Index.FieldInfos; |
| using IndexFileNames = Lucene.Net.Index.IndexFileNames; |
| using IndexOutput = Lucene.Net.Store.IndexOutput; |
| using IOContext = Lucene.Net.Store.IOContext; |
| using IOUtils = Lucene.Net.Util.IOUtils; |
| using UnicodeUtil = Lucene.Net.Util.UnicodeUtil; |
| |
| /// <summary> |
| /// this stores a monotonically increasing set of <Term, TermInfo> pairs in a |
| /// Directory. A TermInfos can be written once, in order. |
| /// </summary> |
| |
| internal sealed class TermInfosWriter : IDisposable |
| { |
| /// <summary> |
| /// The file format version, a negative number. </summary> |
| public const int FORMAT = -3; |
| |
| // Changed strings to true utf8 with length-in-bytes not |
| // length-in-chars |
| public const int FORMAT_VERSION_UTF8_LENGTH_IN_BYTES = -4; |
| |
| // NOTE: always change this if you switch to a new format! |
| public const int FORMAT_CURRENT = FORMAT_VERSION_UTF8_LENGTH_IN_BYTES; |
| |
| private FieldInfos FieldInfos; |
| private IndexOutput Output; |
| private TermInfo LastTi = new TermInfo(); |
| private long Size; |
| |
| // TODO: the default values for these two parameters should be settable from |
| // IndexWriter. However, once that's done, folks will start setting them to |
| // ridiculous values and complaining that things don't work well, as with |
| // mergeFactor. So, let's wait until a number of folks find that alternate |
| // values work better. Note that both of these values are stored in the |
| // segment, so that it's safe to change these w/o rebuilding all indexes. |
| |
| /// <summary> |
| /// Expert: The fraction of terms in the "dictionary" which should be stored |
| /// in RAM. Smaller values use more memory, but make searching slightly |
| /// faster, while larger values use less memory and make searching slightly |
| /// slower. Searching is typically not dominated by dictionary lookup, so |
| /// tweaking this is rarely useful. |
| /// </summary> |
| internal int IndexInterval = 128; |
| |
| /// <summary> |
| /// Expert: The fraction of term entries stored in skip tables, |
| /// used to accelerate skipping. Larger values result in |
| /// smaller indexes, greater acceleration, but fewer accelerable cases, while |
| /// smaller values result in bigger indexes, less acceleration and more |
| /// accelerable cases. More detailed experiments would be useful here. |
| /// </summary> |
| internal int SkipInterval = 16; |
| |
| /// <summary> |
| /// Expert: The maximum number of skip levels. Smaller values result in |
| /// slightly smaller indexes, but slower skipping in big posting lists. |
| /// </summary> |
| internal int MaxSkipLevels = 10; |
| |
| private long LastIndexPointer; |
| private bool IsIndex; |
| private readonly BytesRef LastTerm = new BytesRef(); |
| private int LastFieldNumber = -1; |
| |
| private TermInfosWriter Other; |
| |
| internal TermInfosWriter(Directory directory, string segment, FieldInfos fis, int interval) |
| { |
| Initialize(directory, segment, fis, interval, false); |
| bool success = false; |
| try |
| { |
| Other = new TermInfosWriter(directory, segment, fis, interval, true); |
| Other.Other = this; |
| success = true; |
| } |
| finally |
| { |
| if (!success) |
| { |
| IOUtils.CloseWhileHandlingException(Output); |
| |
| try |
| { |
| directory.DeleteFile(IndexFileNames.SegmentFileName(segment, "", (IsIndex ? Lucene3xPostingsFormat.TERMS_INDEX_EXTENSION : Lucene3xPostingsFormat.TERMS_EXTENSION))); |
| } |
| catch (IOException ignored) |
| { |
| } |
| } |
| } |
| } |
| |
| private TermInfosWriter(Directory directory, string segment, FieldInfos fis, int interval, bool isIndex) |
| { |
| Initialize(directory, segment, fis, interval, isIndex); |
| } |
| |
| private void Initialize(Directory directory, string segment, FieldInfos fis, int interval, bool isi) |
| { |
| IndexInterval = interval; |
| FieldInfos = fis; |
| IsIndex = isi; |
| Output = directory.CreateOutput(IndexFileNames.SegmentFileName(segment, "", (IsIndex ? Lucene3xPostingsFormat.TERMS_INDEX_EXTENSION : Lucene3xPostingsFormat.TERMS_EXTENSION)), IOContext.DEFAULT); |
| bool success = false; |
| try |
| { |
| Output.WriteInt(FORMAT_CURRENT); // write format |
| Output.WriteLong(0); // leave space for size |
| Output.WriteInt(IndexInterval); // write indexInterval |
| Output.WriteInt(SkipInterval); // write skipInterval |
| Output.WriteInt(MaxSkipLevels); // write maxSkipLevels |
| Debug.Assert(InitUTF16Results()); |
| success = true; |
| } |
| finally |
| { |
| if (!success) |
| { |
| IOUtils.CloseWhileHandlingException(Output); |
| |
| try |
| { |
| directory.DeleteFile(IndexFileNames.SegmentFileName(segment, "", (IsIndex ? Lucene3xPostingsFormat.TERMS_INDEX_EXTENSION : Lucene3xPostingsFormat.TERMS_EXTENSION))); |
| } |
| catch (IOException ignored) |
| { |
| } |
| } |
| } |
| } |
| |
| // Currently used only by assert statements |
| internal CharsRef Utf16Result1; |
| |
| internal CharsRef Utf16Result2; |
| private readonly BytesRef ScratchBytes = new BytesRef(); |
| |
| // Currently used only by assert statements |
| private bool InitUTF16Results() |
| { |
| Utf16Result1 = new CharsRef(10); |
| Utf16Result2 = new CharsRef(10); |
| return true; |
| } |
| |
| /// <summary> |
| /// note: -1 is the empty field: "" !!!! </summary> |
| internal static string FieldName(FieldInfos infos, int fieldNumber) |
| { |
| if (fieldNumber == -1) |
| { |
| return ""; |
| } |
| else |
| { |
| return infos.FieldInfo(fieldNumber).Name; |
| } |
| } |
| |
| // Currently used only by assert statement |
| private int CompareToLastTerm(int fieldNumber, BytesRef term) |
| { |
| if (LastFieldNumber != fieldNumber) |
| { |
| int cmp = FieldName(FieldInfos, LastFieldNumber).CompareTo(FieldName(FieldInfos, fieldNumber)); |
| // If there is a field named "" (empty string) then we |
| // will get 0 on this comparison, yet, it's "OK". But |
| // it's not OK if two different field numbers map to |
| // the same name. |
| if (cmp != 0 || LastFieldNumber != -1) |
| { |
| return cmp; |
| } |
| } |
| |
| ScratchBytes.CopyBytes(term); |
| Debug.Assert(LastTerm.Offset == 0); |
| UnicodeUtil.UTF8toUTF16(LastTerm.Bytes, 0, LastTerm.Length, Utf16Result1); |
| |
| Debug.Assert(ScratchBytes.Offset == 0); |
| UnicodeUtil.UTF8toUTF16(ScratchBytes.Bytes, 0, ScratchBytes.Length, Utf16Result2); |
| |
| int len; |
| if (Utf16Result1.Length < Utf16Result2.Length) |
| { |
| len = Utf16Result1.Length; |
| } |
| else |
| { |
| len = Utf16Result2.Length; |
| } |
| |
| for (int i = 0; i < len; i++) |
| { |
| char ch1 = Utf16Result1.Chars[i]; |
| char ch2 = Utf16Result2.Chars[i]; |
| if (ch1 != ch2) |
| { |
| return ch1 - ch2; |
| } |
| } |
| if (Utf16Result1.Length == 0 && LastFieldNumber == -1) |
| { |
| // If there is a field named "" (empty string) with a term text of "" (empty string) then we |
| // will get 0 on this comparison, yet, it's "OK". |
| return -1; |
| } |
| return Utf16Result1.Length - Utf16Result2.Length; |
| } |
| |
| /// <summary> |
| /// Adds a new <<fieldNumber, termBytes>, TermInfo> pair to the set. |
| /// Term must be lexicographically greater than all previous Terms added. |
| /// TermInfo pointers must be positive and greater than all previous. |
| /// </summary> |
| public void Add(int fieldNumber, BytesRef term, TermInfo ti) |
| { |
| Debug.Assert(CompareToLastTerm(fieldNumber, term) < 0 || (IsIndex && term.Length == 0 && LastTerm.Length == 0), "Terms are out of order: field=" + FieldName(FieldInfos, fieldNumber) + " (number " + fieldNumber + ")" + " lastField=" + FieldName(FieldInfos, LastFieldNumber) + " (number " + LastFieldNumber + ")" + " text=" + term.Utf8ToString() + " lastText=" + LastTerm.Utf8ToString()); |
| |
| Debug.Assert(ti.FreqPointer >= LastTi.FreqPointer, "freqPointer out of order (" + ti.FreqPointer + " < " + LastTi.FreqPointer + ")"); |
| Debug.Assert(ti.ProxPointer >= LastTi.ProxPointer, "proxPointer out of order (" + ti.ProxPointer + " < " + LastTi.ProxPointer + ")"); |
| |
| if (!IsIndex && Size % IndexInterval == 0) |
| { |
| Other.Add(LastFieldNumber, LastTerm, LastTi); // add an index term |
| } |
| WriteTerm(fieldNumber, term); // write term |
| |
| Output.WriteVInt(ti.DocFreq); // write doc freq |
| Output.WriteVLong(ti.FreqPointer - LastTi.FreqPointer); // write pointers |
| Output.WriteVLong(ti.ProxPointer - LastTi.ProxPointer); |
| |
| if (ti.DocFreq >= SkipInterval) |
| { |
| Output.WriteVInt(ti.SkipOffset); |
| } |
| |
| if (IsIndex) |
| { |
| Output.WriteVLong(Other.Output.FilePointer - LastIndexPointer); |
| LastIndexPointer = Other.Output.FilePointer; // write pointer |
| } |
| |
| LastFieldNumber = fieldNumber; |
| LastTi.Set(ti); |
| Size++; |
| } |
| |
| private void WriteTerm(int fieldNumber, BytesRef term) |
| { |
| //System.out.println(" tiw.write field=" + fieldNumber + " term=" + term.utf8ToString()); |
| |
| // TODO: UTF16toUTF8 could tell us this prefix |
| // Compute prefix in common with last term: |
| int start = 0; |
| int limit = term.Length < LastTerm.Length ? term.Length : LastTerm.Length; |
| while (start < limit) |
| { |
| if (term.Bytes[start + term.Offset] != LastTerm.Bytes[start + LastTerm.Offset]) |
| { |
| break; |
| } |
| start++; |
| } |
| |
| int length = term.Length - start; |
| Output.WriteVInt(start); // write shared prefix length |
| Output.WriteVInt(length); // write delta length |
| Output.WriteBytes(term.Bytes, start + term.Offset, length); // write delta bytes |
| Output.WriteVInt(fieldNumber); // write field num |
| LastTerm.CopyBytes(term); |
| } |
| |
| /// <summary> |
| /// Called to complete TermInfos creation. </summary> |
| public void Dispose() |
| { |
| try |
| { |
| Output.Seek(4); // write size after format |
| Output.WriteLong(Size); |
| } |
| finally |
| { |
| try |
| { |
| Output.Dispose(); |
| } |
| finally |
| { |
| if (!IsIndex) |
| { |
| Other.Dispose(); |
| } |
| } |
| } |
| } |
| } |
| } |