lucene為數據庫數據創建索引並查詢索引庫

1.

package com.home.utils;

import java.io.IOException;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Statement;
import java.util.ArrayList;
import java.util.List;

import org.apache.lucene.analysis.Analyzer;
import org.apache.lucene.analysis.standard.StandardAnalyzer;
import org.apache.lucene.document.Document;
import org.apache.lucene.document.Field;
import org.apache.lucene.document.Field.Store;
import org.apache.lucene.index.IndexWriter;
import org.apache.lucene.index.IndexWriter.MaxFieldLength;
import org.apache.lucene.queryParser.QueryParser;
import org.apache.lucene.search.IndexSearcher;
import org.apache.lucene.search.Query;
import org.apache.lucene.search.ScoreDoc;
import org.apache.lucene.search.TopDocs;
import org.apache.lucene.util.Version;

public class TestLucene {

    static {
        try {
            // 加載MySql的驅動類
            Class.forName("com.mysql.jdbc.Driver");
        } catch (ClassNotFoundException e) {
            System.out.println("找不到驅動程序類 ，加載驅動失敗！");
            e.printStackTrace();
        }
    }

    public static Connection createConnection() {

        String url = "jdbc:mysql://localhost:3306/cool_files";
        String username = "root";
        String password = "123456";
        Connection con = null;
        try {
            con = DriverManager.getConnection(url, username, password);
        } catch (SQLException se) {
            System.out.println("數據庫連接失敗！");
            se.printStackTrace();
        }

        return con;
    }

    public static ResultSet getResult(String sql, Connection conn) {
        try {
            Statement stmt = conn.createStatement();
            ResultSet rs = stmt.executeQuery(sql);
            return rs;
        } catch (SQLException e) {
            System.out.println(e);
        }
        return null;
    }

    public void close(ResultSet rs, Statement stmt, Connection conn) {

        if (rs != null) { // 關閉記錄集
            try {
                rs.close();
            } catch (SQLException e) {
                e.printStackTrace();
            }
        }
        if (stmt != null) { // 關閉聲明
            try {
                stmt.close();
            } catch (SQLException e) {
                e.printStackTrace();
            }
        }
        if (conn != null) { // 關閉連接對象
            try {
                conn.close();
            } catch (SQLException e) {
                e.printStackTrace();
            }
        }
    }

    public static void Index(ResultSet rs) {
        try {
            IndexWriter indexWriter = new IndexWriter(LuceneUtils.directory,LuceneUtils.analyzer,MaxFieldLength.LIMITED);
            while (rs.next()) {
                Document doc = new Document();
                doc.add(new Field("Au_id", rs.getString(1), Store.YES,
                        org.apache.lucene.document.Field.Index.ANALYZED));
                doc.add(new Field("Au_name", rs.getString(2), Store.YES,
                        org.apache.lucene.document.Field.Index.ANALYZED));
                doc.add(new Field("Phone", rs.getString(3), Store.YES,
                        org.apache.lucene.document.Field.Index.ANALYZED));
                doc.add(new Field("Address", rs.getString(4), Store.YES,
                        org.apache.lucene.document.Field.Index.ANALYZED));
                doc.add(new Field("City", rs.getString(5), Store.YES,
                        org.apache.lucene.document.Field.Index.ANALYZED));
                doc.add(new Field("State", rs.getString(6), Store.YES,
                        org.apache.lucene.document.Field.Index.ANALYZED));
                doc.add(new Field("Zip", rs.getString(7), Store.YES,
                        org.apache.lucene.document.Field.Index.ANALYZED));
                doc.add(new Field("contract", rs.getString(8), Store.YES,
                        org.apache.lucene.document.Field.Index.ANALYZED));
                indexWriter.addDocument(doc);
            }
            indexWriter.optimize();
            indexWriter.close();
        } catch (IOException e) {
            System.out.println(e);
        } catch (SQLException e) {
            System.out.println(e);
        }
    }

    public static Analyzer getAnalyzer() {
        return new StandardAnalyzer(Version.LUCENE_30);
    }

    public static List<Authors> seacher(String queryString) {
        List<Authors> authorsList = null;
        try {
            IndexSearcher is = new IndexSearcher(LuceneUtils.directory);
            QueryParser parser = new QueryParser(Version.LUCENE_30, "Au_name",
                    LuceneUtils.analyzer);
            Query query = parser.parse(queryString);
            TopDocs docs = is.search(query, 10);
            ScoreDoc[] scoreDocs = docs.scoreDocs;
            authorsList = new ArrayList<Authors>();

            for (ScoreDoc scoreDoc : scoreDocs) {
                int num = scoreDoc.doc;
                Document document = is.doc(num);
                Authors article = DocumentUtils.document2Authors(document);
                authorsList.add(article);
            }

        } catch (Exception e) {
            System.out.print(e);
        }
        return authorsList;
    }
}

 

 

2.

package com.home.controller;

import java.sql.Connection;
import java.sql.ResultSet;
import java.util.List;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import org.apache.lucene.analysis.Analyzer;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.ResponseBody;

import com.home.utils.Authors;
import com.home.utils.TestLucene;

@Controller
@RequestMapping("/lucene")
public class LuceneController {
    
    @ResponseBody
    @RequestMapping("/search")
    public List<Authors> getAuthors(HttpServletRequest request,HttpServletResponse response,@RequestParam String name){
        
        Connection conn = TestLucene.createConnection();
        Analyzer analyzer = TestLucene.getAnalyzer();
        System.out.println(name);
        ResultSet rs = TestLucene.getResult("select * from Authors where 1=1 and Au_name='"+name+"'", conn);
        TestLucene.Index(rs);
        List<Authors> authorsList = TestLucene.seacher(name);
        
        return authorsList;
    }
    
    @ResponseBody
    @RequestMapping("/searchOne")
    public List<Authors> getAuthorsOne(HttpServletRequest request,HttpServletResponse response,@RequestParam String name){
        
        List<Authors> authorsList = TestLucene.seacher(name);
        
        return authorsList;
    }
}

 

3、

<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Transitional//EN" "http://www.w3.org/TR/xhtml1/DTD/xhtml1-transitional.dtd">
<%@ page language="java" contentType="text/html; charset=UTF-8"
    pageEncoding="UTF-8"%>
<html xmlns="http://www.w3.org/1999/xhtml">
<%
    String path = request.getContextPath();
    String basePath = request.getScheme() + "://"
            + request.getServerName() + ":" + request.getServerPort()
            + path + "/";
%>
<head>
</head>
<body>
    <form name="form2" action="lucene/searchOne" method="post">
            <input type="text" name="name" /> <input type="submit"
                value="搜索" />
    </form>
</body>
</html>