LuceneUtil.java
上传用户:quxuerui
上传日期:2018-01-08
资源大小:41811k
文件大小:4k
源码类别:

网格计算

开发平台:

Java

  1. /**
  2.  * Licensed to the Apache Software Foundation (ASF) under one
  3.  * or more contributor license agreements.  See the NOTICE file
  4.  * distributed with this work for additional information
  5.  * regarding copyright ownership.  The ASF licenses this file
  6.  * to you under the Apache License, Version 2.0 (the
  7.  * "License"); you may not use this file except in compliance
  8.  * with the License.  You may obtain a copy of the License at
  9.  *
  10.  *     http://www.apache.org/licenses/LICENSE-2.0
  11.  *
  12.  * Unless required by applicable law or agreed to in writing, software
  13.  * distributed under the License is distributed on an "AS IS" BASIS,
  14.  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  15.  * See the License for the specific language governing permissions and
  16.  * limitations under the License.
  17.  */
  18. package org.apache.hadoop.contrib.index.lucene;
  19. import java.io.IOException;
  20. import org.apache.lucene.store.Directory;
  21. /**
  22.  * This class copies some methods from Lucene's SegmentInfos since that class
  23.  * is not public.
  24.  */
  25. public final class LuceneUtil {
  26.   static final class IndexFileNames {
  27.     /** Name of the index segment file */
  28.     static final String SEGMENTS = "segments";
  29.     /** Name of the generation reference file name */
  30.     static final String SEGMENTS_GEN = "segments.gen";
  31.   }
  32.   /**
  33.    * Check if the file is a segments_N file
  34.    * @param name
  35.    * @return true if the file is a segments_N file
  36.    */
  37.   public static boolean isSegmentsFile(String name) {
  38.     return name.startsWith(IndexFileNames.SEGMENTS)
  39.         && !name.equals(IndexFileNames.SEGMENTS_GEN);
  40.   }
  41.   /**
  42.    * Check if the file is the segments.gen file
  43.    * @param name
  44.    * @return true if the file is the segments.gen file
  45.    */
  46.   public static boolean isSegmentsGenFile(String name) {
  47.     return name.equals(IndexFileNames.SEGMENTS_GEN);
  48.   }
  49.   /**
  50.    * Get the generation (N) of the current segments_N file in the directory.
  51.    * 
  52.    * @param directory -- directory to search for the latest segments_N file
  53.    */
  54.   public static long getCurrentSegmentGeneration(Directory directory)
  55.       throws IOException {
  56.     String[] files = directory.list();
  57.     if (files == null)
  58.       throw new IOException("cannot read directory " + directory
  59.           + ": list() returned null");
  60.     return getCurrentSegmentGeneration(files);
  61.   }
  62.   /**
  63.    * Get the generation (N) of the current segments_N file from a list of
  64.    * files.
  65.    * 
  66.    * @param files -- array of file names to check
  67.    */
  68.   public static long getCurrentSegmentGeneration(String[] files) {
  69.     if (files == null) {
  70.       return -1;
  71.     }
  72.     long max = -1;
  73.     for (int i = 0; i < files.length; i++) {
  74.       String file = files[i];
  75.       if (file.startsWith(IndexFileNames.SEGMENTS)
  76.           && !file.equals(IndexFileNames.SEGMENTS_GEN)) {
  77.         long gen = generationFromSegmentsFileName(file);
  78.         if (gen > max) {
  79.           max = gen;
  80.         }
  81.       }
  82.     }
  83.     return max;
  84.   }
  85.   /**
  86.    * Parse the generation off the segments file name and return it.
  87.    */
  88.   public static long generationFromSegmentsFileName(String fileName) {
  89.     if (fileName.equals(IndexFileNames.SEGMENTS)) {
  90.       return 0;
  91.     } else if (fileName.startsWith(IndexFileNames.SEGMENTS)) {
  92.       return Long.parseLong(
  93.           fileName.substring(1 + IndexFileNames.SEGMENTS.length()),
  94.           Character.MAX_RADIX);
  95.     } else {
  96.       throw new IllegalArgumentException("fileName "" + fileName
  97.           + "" is not a segments file");
  98.     }
  99.   }
  100. }