【Lucene】挖掘相关搜索词

搜索引擎中往往有一个可选的搜索词的列表，当搜索结果太少时，可以帮助用户扩展搜索内容，或者搜索结果太多的时候可以帮助用户深入定向搜索。一种方法是从搜索日志中挖掘字面相似的词作为相关搜索词列表。另一种方法是把用户共同查询的词作为相关搜索词，需要有搜索日志才能实现。【摘自《Lucene In Action》】

下面使用的是第一种方法：

 

package com.tan.code;



//省略引入



public class RelateWords {

	private static final String TEXT_FIELD = "text";



	/**

	 * 

	 * @param words 候選相関詞列表

	 * @param word 相關搜索詞的種子詞

	 * @return 

	 * @throws IOException

	 * @throws ParseException

	 */

	static public String[] filterRelated(HashSet<String> words, String word)

			throws IOException, ParseException {



		//RAMDirectory ramDirectory = new RAMDirectory();

		Directory directory=new SimpleFSDirectory(new File("E://related"));

		IndexWriter indexWriter = new IndexWriter(directory,

				new IndexWriterConfig(Version.LUCENE_43, new IKAnalyzer(true)));

		for (String text : words) {

			Document document = new Document();

			document.add(new TextField(TEXT_FIELD, text, Store.YES));

			indexWriter.addDocument(document);

		}

		indexWriter.close();

		IndexReader indexReader = DirectoryReader.open(directory);

		IndexSearcher indexSearcher = new IndexSearcher(indexReader);

		QueryParser queryParser = new QueryParser(Version.LUCENE_43,

				TEXT_FIELD, new IKAnalyzer(true));

		Query query = queryParser.parse(word);

		TopDocs td = indexSearcher.search(query, 10);

		ScoreDoc[] sd = td.scoreDocs;

		String relateWords[] = new String[sd.length];

		for (int i = 0; i < sd.length; i++) {

			int z = sd[i].doc;

			Document doc = indexSearcher.doc(z);

			relateWords[i] = doc.get(TEXT_FIELD);

		}

		indexReader.close();

		//ramDirectory.close();

		directory.close();

		return relateWords;



	}



}

测试代码：

 

 

	@Test

	public void test() throws IOException, ParseException {

		// fail("Not yet implemented");

		HashSet<String> words = new HashSet<String>();



		// words.add("Lucene");

		// words.add("Lucene入門資料");

		// words.add("java資料下載");

		// words.add("SQL詳解");

		// words.add("揭祕Lucene原理");

		// words.add("Spring原理解析");

		// words.add("什麽是Lucene?怎麽樣才可以學好Lucene呢？");



		String word = "Spring資料";



		String rewords[] = RelateWords.filterRelated(words, word);

		System.out.println("搜索内容：" + word);

		System.out.println("相關搜索匹配結果：");

		for (int i = 0; i < rewords.length; i++) {

			System.out.println(rewords[i]);

		}

	}

测试结果：

 

 

搜索内容：Spring資料

相關搜索匹配結果：

java資料下載

Lucene入門資料

Spring原理解析