[java] 아파치 루신(Apache Lucene)을 사용한 자동완성 기능 구현 방법

28 Nov 2023

java

이번 포스트에서는 아파치 루신(Apache Lucene)을 사용하여 자동완성 기능을 구현하는 방법을 알아보겠습니다.

1. 아파치 루신(Lucene) 소개

아파치 루신은 자바로 작성된 검색 라이브러리로, 텍스트 기반의 검색 기능을 제공합니다. 아파치 루신은 인덱스 구조를 활용하여 효율적인 검색을 가능하게 합니다.

2. 자동완성 기능의 구현 방법

자동완성 기능은 입력한 텍스트에 대한 맞춤 검색을 수행하여 매칭되는 결과를 제공하는 기능입니다. 아래는 아파치 루신을 사용하여 자동완성 기능을 구현하는 단계별 방법입니다.

2.1. 검색 인덱스 생성하기

먼저, 검색을 수행할 데이터를 바탕으로 검색 인덱스를 생성해야 합니다. 검색 인덱스는 텍스트 데이터를 색인화하여 검색을 빠르게 수행할 수 있도록 합니다.

// 인덱스 생성 코드 예시
import org.apache.lucene.analysis.standard.StandardAnalyzer;
import org.apache.lucene.document.Document;
import org.apache.lucene.document.Field;
import org.apache.lucene.index.IndexWriter;
import org.apache.lucene.store.Directory;
import org.apache.lucene.store.RAMDirectory;

public class IndexBuilder {
    public static void main(String[] args) {
        StandardAnalyzer analyzer = new StandardAnalyzer();
        Directory index = new RAMDirectory();

        IndexWriter writer = new IndexWriter(index, new IndexWriterConfig(analyzer));

        // 문서 생성 및 인덱싱
        Document doc = new Document();
        doc.add(new Field("content", "example content", TextField.TYPE_STORED));
        writer.addDocument(doc);

        writer.close();
    }
}

위 코드에서는 아파치 루신의 StandardAnalyzer를 사용하여 텍스트를 토큰화하고 인덱싱합니다. 인덱싱될 필드는 “content”라는 필드로 정의되었습니다.

2.2. 입력 텍스트에 대한 자동완성 결과 가져오기

자동완성 기능을 구현하기 위해서는 입력한 텍스트에 대한 맞춤 검색을 수행하여 결과를 가져와야 합니다. 아래는 아파치 루신을 사용하여 입력 텍스트에 대한 자동완성 결과를 가져오는 코드 예시입니다.

// 검색 코드 예시
import org.apache.lucene.analysis.standard.StandardAnalyzer;
import org.apache.lucene.index.DirectoryReader;
import org.apache.lucene.index.IndexReader;
import org.apache.lucene.index.Term;
import org.apache.lucene.search.IndexSearcher;
import org.apache.lucene.search.PrefixQuery;
import org.apache.lucene.search.Query;
import org.apache.lucene.store.Directory;
import org.apache.lucene.store.RAMDirectory;
import org.apache.lucene.search.ScoreDoc;
import org.apache.lucene.search.TopDocs;

public class Searcher {
    public static void main(String[] args) {
        StandardAnalyzer analyzer = new StandardAnalyzer();
     
        Directory index = new RAMDirectory();
        IndexReader reader = DirectoryReader.open(index);
        IndexSearcher searcher = new IndexSearcher(reader);

        String queryStr = "ex"; // 사용자 입력 텍스트
        Term term = new Term("content", queryStr);
        Query query = new PrefixQuery(term);

        int numResults = 10; // 가져올 결과 개수
        TopDocs topDocs = searcher.search(query, numResults);
        ScoreDoc[] scoreDocs = topDocs.scoreDocs;

        // 결과 출력
        for (ScoreDoc scoreDoc : scoreDocs) {
            Document doc = searcher.doc(scoreDoc.doc);
            System.out.println(doc.get("content"));
        }

        reader.close();
    }
}

위 코드에서는 입력된 텍스트를 PrefixQuery를 사용하여 검색하고, 일치하는 결과를 가져옵니다. 결과는 TopDocs 객체로 반환되며, 각 결과의 Document에서 필드 값을 가져와 출력합니다.

3. 결론

아파치 루신을 활용하여 자동완성 기능을 구현하는 방법에 대해 알아보았습니다. 아파치 루신은 강력한 검색 라이브러리로 다양한 검색 기능을 구현할 수 있으며, 자동완성 기능도 그 중 하나입니다.

더 자세한 내용은 아파치 루신 공식 문서를 참고하시기 바랍니다.

아파치 루신 공식 홈페이지: https://lucene.apache.org/