lucene-java-3.4.0/lucene/src/java/org/apache/lucene/index/FormatPostingsFieldsWriter.java

   1 package org.apache.lucene.index;
   2
   3 /**
   4  * Licensed to the Apache Software Foundation (ASF) under one or more
   5  * contributor license agreements.  See the NOTICE file distributed with
   6  * this work for additional information regarding copyright ownership.
   7  * The ASF licenses this file to You under the Apache License, Version 2.0
   8  * (the "License"); you may not use this file except in compliance with
   9  * the License.  You may obtain a copy of the License at
  10  *
  11  *     http://www.apache.org/licenses/LICENSE-2.0
  12  *
  13  * Unless required by applicable law or agreed to in writing, software
  14  * distributed under the License is distributed on an "AS IS" BASIS,
  15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  16  * See the License for the specific language governing permissions and
  17  * limitations under the License.
  18  */
  19
  20 import java.io.IOException;
  21
  22 import org.apache.lucene.store.Directory;
  23 import org.apache.lucene.util.IOUtils;
  24
  25 final class FormatPostingsFieldsWriter extends FormatPostingsFieldsConsumer {
  26
  27   final Directory dir;
  28   final String segment;
  29   TermInfosWriter termsOut;
  30   final FieldInfos fieldInfos;
  31   FormatPostingsTermsWriter termsWriter;
  32   final DefaultSkipListWriter skipListWriter;
  33   final int totalNumDocs;
  34
  35   public FormatPostingsFieldsWriter(SegmentWriteState state, FieldInfos fieldInfos) throws IOException {
  36     dir = state.directory;
  37     segment = state.segmentName;
  38     totalNumDocs = state.numDocs;
  39     this.fieldInfos = fieldInfos;
  40     boolean success = false;
  41     try {
  42       termsOut = new TermInfosWriter(dir, segment, fieldInfos, state.termIndexInterval);
  43
  44       // TODO: this is a nasty abstraction violation (that we
  45       // peek down to find freqOut/proxOut) -- we need a
  46       // better abstraction here whereby these child consumers
  47       // can provide skip data or not
  48       skipListWriter = new DefaultSkipListWriter(termsOut.skipInterval,
  49           termsOut.maxSkipLevels, totalNumDocs, null, null);
  50
  51       termsWriter = new FormatPostingsTermsWriter(state, this);
  52       success = true;
  53     } finally {
  54       if (!success) {
  55         IOUtils.closeWhileHandlingException(termsOut, termsWriter);
  56       }
  57     }
  58   }
  59
  60   /** Add a new field */
  61   @Override
  62   FormatPostingsTermsConsumer addField(FieldInfo field) {
  63     termsWriter.setField(field);
  64     return termsWriter;
  65   }
  66
  67   /** Called when we are done adding everything. */
  68   @Override
  69   void finish() throws IOException {
  70     IOUtils.close(termsOut, termsWriter);
  71   }
  72 }