lucene创建索引以及索引文件合并

  1 package test;
  2 
  3 import java.io.File;
  4 import java.io.IOException;
  5 import java.nio.file.Path;
  6 import java.util.ArrayList;
  7 import java.util.List;
  8 import java.util.Map;
  9 
 10 import org.apache.lucene.analysis.standard.StandardAnalyzer;
 11 import org.apache.lucene.document.Document;
 12 import org.apache.lucene.document.Field.Store;
 13 import org.apache.lucene.document.StringField;
 14 import org.apache.lucene.index.DirectoryReader;
 15 import org.apache.lucene.index.IndexReader;
 16 import org.apache.lucene.index.IndexWriter;
 17 import org.apache.lucene.index.IndexWriterConfig;
 18 import org.apache.lucene.store.Directory;
 19 import org.apache.lucene.store.FSDirectory;
 20 
 21 public class IndexCreate {
 22     public static void main(String[] args) {
 23 //        makeIndexs();
 24         mergeIndexFiles();
 25         searchIndexFileSize();
 26         
 27     }
 28 
 29     /**
 30      * 创建索引
 31      */
 32     public static void makeIndexs() {
 33         String sql = " select * from (select rownum r,t.* from TB_MENU t) m where m.r>=15 ";
 34         OracleDBUtils db = new OracleDBUtils();
 35         List<String> list = new ArrayList<String>();
 36         list.add("ID");
 37         list.add("MENUTYPE");
 38         list.add("MENUNAME");
 39         list.add("URL");
 40         list.add("ICONPATH");
 41         List<Map<String, Object>> res = db.exec(sql, list);
 42         /**
 43          * 创建索引文件
 44          */
 45         IndexWriter writer = null;
 46         Directory directory = null;
 47         try {
 48             Path path = new File("D:/temp").toPath();
 49             directory = FSDirectory.open(path);
 50             writer = new IndexWriter(directory, new IndexWriterConfig(
 51                     new StandardAnalyzer()));
 52             for (Map<String, Object> map : res) {
 53                 Document doc = new Document();
 54                 for (String key : list) {
 55                     String value = map.get(key) == null ? "" : map.get(key)
 56                             .toString();
 57                     doc.add(new StringField(key, value, Store.YES));
 58                 }
 59                 writer.addDocument(doc);
 60             }
 61         } catch (IOException e) {
 62             e.printStackTrace();
 63         } finally {
 64             try {
 65                 writer.commit();
 66                 writer.close();
 67                 directory.close();
 68                 System.out.println("索引创建成功!");
 69             } catch (IOException e) {
 70                 e.printStackTrace();
 71             }
 72 
 73         }
 74     }
 75 
 76     /**
 77      * 合并索引文件
 78      */
 79     public static void mergeIndexFiles() {
 80         IndexWriter writer = null;
 81         Directory directory = null;
 82         Directory tempDir = null;
 83         try {
 84             Path path = new File("D:/luceneDir").toPath();
 85             directory = FSDirectory.open(path);
 86             Path temp = new File("D:/temp").toPath();
 87             tempDir = FSDirectory.open(temp);
 88             writer = new IndexWriter(directory,
 89                     new IndexWriterConfig(new StandardAnalyzer()));
 90             writer.addIndexes(tempDir);
 91             writer.commit();
 92         } catch (IOException e) {
 93             e.printStackTrace();
 94         } finally{
 95             try {
 96                 writer.close();
 97                 directory.close();
 98                 tempDir.close();
 99                 System.out.println("索引文件合并成功!");
100             } catch (IOException e) {
101                 e.printStackTrace();
102             }
103         }
104     }
105     
106     public static void searchIndexFileSize() {
107         Directory directory = null;
108         try {
109             Path path = new File("D:/luceneDir").toPath();
110             directory = FSDirectory.open(path);
111             IndexReader reader=DirectoryReader.open(directory);  
112             System.out.println(reader.maxDoc());
113         } catch (IOException e) {
114             e.printStackTrace();
115         } finally{
116             try {
117                 directory.close();
118             } catch (IOException e) {
119                 e.printStackTrace();
120             }
121         }
122     }
123 }