程式師世界 >> 編程語言 >> C語言 >> C++ >> 關於C++ >> 創建一個提供搜索功能的搜索類(可運行)

創建一個提供搜索功能的搜索類(可運行)

編輯：關於C++

/*
* 這段代碼的主要功能是對於創建索引的後的文件，
* 創建一個提供搜索功能的搜索類。
* */
package ch2.lucenedemo.process;

import java.io.BufferedReader;
import java.io.File;
import java.io.FileReader;
import java.io.IOException;
import java.util.Date;
import java.util.Iterator;
import java.util.LinkedHashMap;
import java.util.Map;

import org.apache.lucene.index.Term;
import org.apache.lucene.index.TermDocs;
import org.apache.lucene.search.IndexSearcher;
import org.apache.lucene.search.Query;
import org.apache.lucene.search.TermQuery;

public class Search {
//定義索引生成的目錄
private String INDEX_STORE_PATH = "E:\\Lucene項目\\索引目錄";

//利用Lucene的搜索
public void indexSearch(String searchType, String searchKey){
try{

System.out.println("++使用索引方式搜索++");
System.out.println("----------------------------------");

//根據索引位置建立IndexSearcher
IndexSearcher searcher = new IndexSearcher("INDEX_STORE_PATH");

//Term是搜索的最小單元，類似於一個漢字或者詞語
//創建搜索單元，searchType代表要搜索的Field, searchKey代表關鍵字
Term t = new Term(searchType, searchKey);

//由Term生成一個Query
Query q = new TermQuery(t);

//搜索開始的時間
Date beginTime = new Date();

//獲取一個的枚舉對象TermDocs
TermDocs termDocs = searcher.getIndexReader().termDocs(t);
while(termDocs.next()){
//輸出在文檔中出現的關鍵字的次數
System.out.println("find " + termDocs.freq() + "matches in ");

//輸出搜索到關鍵詞的文檔
System.out.println(searcher.getIndexReader().document(termDocs.doc()).getField("fileName").stringValue());
}

//搜索完成時間
Date endTime = new Date();

//搜索所耗使勁
long timeofSearch = endTime.getTime() - beginTime.getTime();
System.out.println("使用索引方式搜索總耗時 : " + timeofSearch + "ms");
}catch(IOException e){
e.printStackTrace();
}
}

//利用String的搜索
public void stringSearch(String keyword, String searchDir){
System.out.println("++使用字符串匹配方式搜索++");
System.out.println("---------------------------------");
File filesDir = new File(searchDir);

//返回目錄文件夾所有文件數組
File[] files = filesDir.listFiles();

//HashMap保存文件名和匹配次數對
Map rs = new LinkedHashMap();

//記錄搜索開始時間
Date beginTime = new Date();

//遍歷所有文件
for(int i = 0; i < files.length; i++){
//初始化匹配次數
int hits = 0;
try{

//讀取文件內容
BufferedReader br = new BufferedReader(new FileReader(files[i]));
StringBuffer sb = new StringBuffer();
String line = br.readLine();
while(line != null){
sb.append(line);
line = br.readLine();
}
br.close();

//將Stringbuffer轉化為String, 以便於搜索
String stringToSearch = sb.toString();

//初始化fromIndex
int fromIndex = -keyword.length();

//逐個匹配關鍵字
while((fromIndex = stringToSearch.indexOf(keyword, fromIndex + keyword.length())) != -1){
hits++;
}

//將文件名和匹配次數加入到hashMap
rs.put(files[i].getName(), new Integer(hits));

}catch(IOException e){
e.printStackTrace();
}

}

//輸出查詢結果
Iterator it = rs.keySet().iterator();
while(it.hasNext()){
String fileName = (String)it.next();
Integer hits = (Integer) rs.get(fileName);
System.out.println("find " + hits.intValue() + "matches in " + fileName);

}

//記錄結束時間
Date endTime = new Date();

//得到搜索耗費時間
long timeOfSearch = endTime.getTime() - beginTime.getTime();
System.out.println("使用字符串匹配方式總耗時 " + timeOfSearch + " ms");

}

public static void main(String[] args){
Search search = new Search();

//通過索引速鎖關鍵字
search.indexSearch("content", "保爾");

//插入一個分隔行
System.out.println("=========================");
System.out.println("=========================");

//通過String的API搜索關鍵詞
search.stringSearch("保爾", "INDEX_STORE_PATH");
}

}