org.apache.ctakes.dictionary.lookup.lucene.LuceneDocumentMetaDataHitImpl Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of ctakes-dictionary-lookup Show documentation
The newest version!
/**
 * 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.
 */
package org.apache.ctakes.dictionary.lookup.lucene;

import org.apache.ctakes.dictionary.lookup.AbstractBaseMetaDataHit;
import org.apache.lucene.document.Document;
import org.apache.lucene.index.IndexableField;

import javax.annotation.concurrent.Immutable;
import java.util.*;


/**
 * @author Mayo Clinic
 */
@Immutable
public final class LuceneDocumentMetaDataHitImpl extends AbstractBaseMetaDataHit {
   final private Document _luceneDoc;
   final private Set _nameSet;
   final private Collection _valueList;

   public LuceneDocumentMetaDataHitImpl( final Document luceneDoc ) {
      _luceneDoc = luceneDoc;
      final List fieldEnumList = _luceneDoc.getFields();
      final Set nameSet = new HashSet<>( fieldEnumList.size() );
      final List valueList = new ArrayList<>( fieldEnumList.size() );
      for ( IndexableField field : fieldEnumList ) {
         nameSet.add( field.name() );
         valueList.add( field.stringValue() );
      }
      _nameSet = Collections.unmodifiableSet( nameSet );
      _valueList = Collections.unmodifiableList( valueList );
   }

   /**
    * {@inheritDoc}
    */
   @Override
   public String getMetaFieldValue( final String metaFieldName ) {
      return _luceneDoc.get( metaFieldName );
   }

   /**
    * {@inheritDoc}
    */
   @Override
   public Set getMetaFieldNames() {
      return _nameSet;
   }

   /**
    * {@inheritDoc}
    */
   @Override
   public Collection getMetaFieldValues() {
      return _valueList;
   }
}