@PostConstruct

  public void init() {

    super.init();

  }

 

  @Test

  public void highlightUsage() throws IOException, InvalidTokenOffsetsException {

    String text = "my fox jump group org next fox spring health care book fox tape java fox fox shop world fox";

   

    TermQuery query = new TermQuery(new Term("f", "fox"));

    QueryScorer scorer = new QueryScorer(query);

    Highlighter highlighter = new Highlighter(scorer);

    Fragmenter fragmenter = new SimpleFragmenter(5);

    highlighter.setTextFragmenter(fragmenter);

    Analyzer a = new KoreanAnalyzer(false);

    TokenStream tokenStream = a.tokenStream("f", new StringReader(text));

    String result =

            highlighter.getBestFragments(tokenStream, text,2, "...");

    a.close();

   

    System.out.println(result);

   

    Assert.assertEquals(" <B>fox</B>... <B>fox</B>", result);

  }

 

  @Test

  public void fastVectorTest() throws IOException {

    //Directory dir = new RAMDirectory();

        //IndexWriter indexWriter = new IndexWriter(dir, new IndexWriterConfig(Lucene.VERSION, Lucene.STANDARD_ANALYZER));

        //indexWriter.addDocument(doc().add(field("_id", "1")).add(field("content", "the big bad dog", Field.Store.YES, Field.Index.ANALYZED, Field.TermVector.WITH_POSITIONS_OFFSETS)).build());

        //IndexReader reader = IndexReader.open(indexWriter, true);

        //IndexSearcher searcher = new IndexSearcher(reader);

        //TopDocs topDocs = searcher.search(new TermQuery(new Term("_id", "1")), 1);

    SearchRequest searchRequest = new SearchRequest();

    searchRequest.setKeyword("입니다");

    searchRequest.setCollectionName("sample");

    searchRequest.setSearchField("title,dscr");

    //IndexWriterManager indexWriterManager = IndexWriterManager.getIndexWriterManager();

    //IndexWriter indexWriter = indexWriterManager.getIndexWriter("sample");

    //IndexReader indexReader = IndexReader.open(indexWriter, true);

   

    SearcherManager searcherManager = crescentSearcherManager.getSearcherManager("sample");

    IndexSearcher indexSearcher = searcherManager.acquire();

   

    IndexReader indexReader = indexSearcher.getIndexReader();

   

    TopDocs topDocs = indexSearcher.search(new TermQuery(new Term("dscr", "입니다")), 3);

   

    System.out.println("ddd : " + indexReader.document(0));

    System.out.println(topDocs.totalHits);

    FastVectorHighlighter highlighter = new FastVectorHighlighter();

   

    System.out.println("field query : " + highlighter.getFieldQuery(new TermQuery(new Term("dscr", "입니다"))));

   

    //System.out.println(topDocs.scoreDocs[0].doc);

    String fragment = highlighter.getBestFragment(highlighter.getFieldQuery(new TermQuery(new Term("dscr", "텍스트"))),

        indexReader, topDocs.scoreDocs[0].doc, "dscr", 30);

   

    System.out.println(fragment);

   

//

//        FastVectorHighlighter highlighter = new FastVectorHighlighter();

//        String fragment = highlighter.getBestFragment(highlighter.getFieldQuery(new TermQuery(new Term("content", "bad"))),

//                reader, topDocs.scoreDocs[0].doc, "content", 30);

//        assertThat(fragment, notNullValue());

//        System.out.println(fragment);

   

  }

 

  @Test public void testVectorHighlighter() throws Exception {

        Directory dir = new RAMDirectory();

        IndexWriter indexWriter

          = new IndexWriter(dir, new IndexWriterConfig(Version.LUCENE_44, new WhitespaceAnalyzer(Version.LUCENE_44)));

        Document doc = new Document();

        FieldType fieldType = new FieldType();

    fieldType.setIndexed(true);

    fieldType.setStored(true);

    fieldType.setTokenized(true);

    fieldType.setIndexOptions(IndexOptions.DOCS_AND_FREQS_AND_POSITIONS_AND_OFFSETS);

    fieldType.setStoreTermVectors(true);

     

        Field f1 = new Field("_id", "1", fieldType);

        Field f2 = new Field("content", "the big 삼성전자연구원. dog", fieldType);

  

        doc.add(f1);

        doc.add(f2);

       

        indexWriter.addDocument(doc);

       

        indexWriter.commit();

        indexWriter.close();

       

        DirectoryReader reader = DirectoryReader.open(dir);

        IndexSearcher searcher = new IndexSearcher(reader);

        TopDocs topDocs = searcher.search(new TermQuery(new Term("_id", "1")), 1);

        System.out.println(topDocs.totalHits);

        

        FastVectorHighlighter highlighter = new FastVectorHighlighter();

        String fragment = highlighter.getBestFragment(highlighter.getFieldQuery(new TermQuery(new Term("content", "big"))),

            reader, topDocs.scoreDocs[0].doc, "content", 200);

      

        System.out.println(fragment);

    }

}