MergePolicy.cs

Go to the documentation of this file.

/* 
 * 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 System;
using System.Collections.Generic;
using Directory = Lucene.Net.Store.Directory;

namespace Lucene.Net.Index
{
    
    
    public abstract class MergePolicy : IDisposable
    {
        
        
        public class OneMerge
        {
            
            internal SegmentInfo info;              // used by IndexWriter
            internal bool mergeDocStores;           // used by IndexWriter
            internal bool optimize;                 // used by IndexWriter
            internal bool registerDone;             // used by IndexWriter
            internal long mergeGen;                 // used by IndexWriter
            internal bool isExternal;               // used by IndexWriter
            internal int maxNumSegmentsOptimize;    // used by IndexWriter
            internal SegmentReader[] readers;       // used by IndexWriter
            internal SegmentReader[] readersClone;  // used by IndexWriter
            internal SegmentInfos segments;
            internal bool useCompoundFile;
            internal bool aborted;
            internal System.Exception error;
            
            public OneMerge(SegmentInfos segments, bool useCompoundFile)
            {
                if (0 == segments.Count)
                    throw new ArgumentException("segments must include at least one segment", "segments");
                this.segments = segments;
                this.useCompoundFile = useCompoundFile;
            }
            
            internal virtual void  SetException(System.Exception error)
            {
                lock (this)
                {
                    this.error = error;
                }
            }
            
            internal virtual System.Exception GetException()
            {
                lock (this)
                {
                    return error;
                }
            }
            
            internal virtual void  Abort()
            {
                lock (this)
                {
                    aborted = true;
                }
            }
            
            internal virtual bool IsAborted()
            {
                lock (this)
                {
                    return aborted;
                }
            }
            
            internal virtual void  CheckAborted(Directory dir)
            {
                lock (this)
                {
                    if (aborted)
                        throw new MergeAbortedException("merge is aborted: " + SegString(dir));
                }
            }
            
            internal virtual String SegString(Directory dir)
            {
                var b = new System.Text.StringBuilder();
                int numSegments = segments.Count;
                for (int i = 0; i < numSegments; i++)
                {
                    if (i > 0)
                        b.Append(' ');
                    b.Append(segments.Info(i).SegString(dir));
                }
                if (info != null)
                    b.Append(" into ").Append(info.name);
                if (optimize)
                    b.Append(" [optimize]");
                if (mergeDocStores)
                {
                    b.Append(" [mergeDocStores]");
                }
                return b.ToString();
            }

            public SegmentInfos segments_ForNUnit
            {
                get { return segments; }
            }
        }
        
        
        public class MergeSpecification
        {
            
            
            public IList<OneMerge> merges = new List<OneMerge>();
            
            public virtual void  Add(OneMerge merge)
            {
                merges.Add(merge);
            }
            
            public virtual String SegString(Directory dir)
            {
                var b = new System.Text.StringBuilder();
                b.Append("MergeSpec:\n");
                int count = merges.Count;
                for (int i = 0; i < count; i++)
                    b.Append("  ").Append(1 + i).Append(": ").Append(merges[i].SegString(dir));
                return b.ToString();
            }
        }
        
        [Serializable]
        public class MergeException:System.SystemException
        {
            private readonly Directory dir;

            public MergeException(System.String message, Directory dir):base(message)
            {
                this.dir = dir;
            }

            public MergeException(System.Exception exc, Directory dir):base(null, exc)
            {
                this.dir = dir;
            }

            public virtual Directory Directory
            {
                get { return dir; }
            }
        }
        
        [Serializable]
        public class MergeAbortedException:System.IO.IOException
        {
            public MergeAbortedException():base("merge is aborted")
            {
            }
            public MergeAbortedException(System.String message):base(message)
            {
            }
        }
        
        protected internal IndexWriter writer;

        protected MergePolicy(IndexWriter writer)
        {
            this.writer = writer;
        }
        
        public abstract MergeSpecification FindMerges(SegmentInfos segmentInfos);

        public abstract MergeSpecification FindMergesForOptimize(SegmentInfos segmentInfos, int maxSegmentCount,
                                                                 ISet<SegmentInfo> segmentsToOptimize);
        
        public abstract MergeSpecification FindMergesToExpungeDeletes(SegmentInfos segmentInfos);

        [Obsolete("Use Dispose() instead")]
        public void Close()
        {
            Dispose();
        }

        public void Dispose()
        {
            Dispose(true);
        }

        protected abstract void Dispose(bool disposing);
        
        public abstract bool UseCompoundFile(SegmentInfos segments, SegmentInfo newSegment);
        
        public abstract bool UseCompoundDocStore(SegmentInfos segments);
    }
}