lucene-java-3.5.0/lucene/src/java/org/apache/lucene/index/StoredFieldsWriterPerThread.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 import org.apache.lucene.store.IndexOutput;
  22 import org.apache.lucene.document.Fieldable;
  23
  24 final class StoredFieldsWriterPerThread {
  25
  26   final FieldsWriter localFieldsWriter;
  27   final StoredFieldsWriter storedFieldsWriter;
  28   final DocumentsWriter.DocState docState;
  29
  30   StoredFieldsWriter.PerDoc doc;
  31
  32   public StoredFieldsWriterPerThread(DocumentsWriter.DocState docState, StoredFieldsWriter storedFieldsWriter) throws IOException {
  33     this.storedFieldsWriter = storedFieldsWriter;
  34     this.docState = docState;
  35     localFieldsWriter = new FieldsWriter((IndexOutput) null, (IndexOutput) null, storedFieldsWriter.fieldInfos);
  36   }
  37
  38   public void startDocument() {
  39     if (doc != null) {
  40       // Only happens if previous document hit non-aborting
  41       // exception while writing stored fields into
  42       // localFieldsWriter:
  43       doc.reset();
  44       doc.docID = docState.docID;
  45     }
  46   }
  47
  48   public void addField(Fieldable field, FieldInfo fieldInfo) throws IOException {
  49     if (doc == null) {
  50       doc = storedFieldsWriter.getPerDoc();
  51       doc.docID = docState.docID;
  52       localFieldsWriter.setFieldsStream(doc.fdt);
  53       assert doc.numStoredFields == 0: "doc.numStoredFields=" + doc.numStoredFields;
  54       assert 0 == doc.fdt.length();
  55       assert 0 == doc.fdt.getFilePointer();
  56     }
  57
  58     localFieldsWriter.writeField(fieldInfo, field);
  59     assert docState.testPoint("StoredFieldsWriterPerThread.processFields.writeField");
  60     doc.numStoredFields++;
  61   }
  62
  63   public DocumentsWriter.DocWriter finishDocument() {
  64     // If there were any stored fields in this doc, doc will
  65     // be non-null; else it's null.
  66     try {
  67       return doc;
  68     } finally {
  69       doc = null;
  70     }
  71   }
  72
  73   public void abort() {
  74     if (doc != null) {
  75       doc.abort();
  76       doc = null;
  77     }
  78   }
  79 }