elastic.onestep.search.ExpertSearcher Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of info.unclewang.elastic Show documentation
The newest version!
package elastic.onestep.search;


import elastic.onestep.factory.ClientFactory;
import org.elasticsearch.action.search.SearchResponse;
import org.elasticsearch.common.unit.TimeValue;
import org.elasticsearch.index.query.QueryBuilders;
import org.elasticsearch.search.SearchHit;

import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.concurrent.TimeUnit;

public class ExpertSearcher {

    public static Set search(String index, String type) {
        SearchResponse response = ClientFactory.get().prepareSearch(index)
                .setTypes(type)
                .setQuery(QueryBuilders.boolQuery().filter(QueryBuilders.existsQuery("fos"))
                        .must(QueryBuilders.termQuery("lang", "zh_chs")))
                .setFetchSource(new String[]{"id", "keywords", "authors", "fos"}, null)
                .setScroll(new TimeValue(10, TimeUnit.MINUTES))
                .setSize(10)
                .get();
        String scrollId = response.getScrollId();
//        System.out.println("sdfhksj" + scrollId);
        SearchHit[] searchHits = response.getHits().getHits();
        Map source;
        Set docIds = new HashSet<>(searchHits.length);
        for (SearchHit hit : searchHits) {
            System.out.println(hit.getId());
            source = hit.getSourceAsMap();
            System.out.println(source.toString());
            docIds.add(hit.getId());
        }

        List xa = ScrollSearch.searchByScrollId(scrollId);
        System.out.println("前10个打印" + xa.size());
        for (SearchHit hit : xa) {
            docIds.add(hit.getId());
        }
        return docIds;
    }

    public static Set searchName(String index, String type, String name) {
        SearchResponse response = ClientFactory.get().prepareSearch(index)
                .setTypes(type)
                .setQuery(QueryBuilders.boolQuery().filter(QueryBuilders.existsQuery("fos"))
                        .must(QueryBuilders.termQuery("lang", "zh_chs")))
                .setFetchSource(false)
                .setScroll(new TimeValue(10, TimeUnit.MINUTES))
                .setSize(10)
                .get();
        String scrollId = response.getScrollId();
        System.out.println("sdfhksj" + scrollId);
        SearchHit[] searchHits = response.getHits().getHits();
        Set docIds = new HashSet<>(searchHits.length);
        for (SearchHit hit : searchHits) {
            docIds.add(hit.getId());
        }
        System.out.println(docIds.size());
        System.out.println("前10个打印");
        for (SearchHit hit : ScrollSearch.searchByScrollId(scrollId)) {
            docIds.add(hit.getId());
        }
        System.out.println(docIds.size());
        return docIds;
    }

    public static Set searchfoskeywords(String index, String type) {
        SearchResponse response = ClientFactory.get().prepareSearch(index)
                .setTypes(type)
                .setQuery(QueryBuilders.boolQuery()
                        .must(QueryBuilders.boolQuery().filter(QueryBuilders.existsQuery("keywords")))
                        .must(QueryBuilders.matchQuery("lang", "zh_chs"))
                        .must(QueryBuilders.boolQuery().filter(QueryBuilders.existsQuery("fos")))
                        .must(QueryBuilders.matchQuery("year", "2014"))
                        .must(QueryBuilders.matchQuery("issue", "8"))
                        .must(QueryBuilders.matchQuery("page_end", "766")))
                .setFetchSource(true)
                .setScroll(new TimeValue(10, TimeUnit.MINUTES))
                .setSize(10)
                .get();
        String scrollId = response.getScrollId();
        SearchHit[] searchHits = response.getHits().getHits();
        Map source;
        Set docIds = new HashSet<>(searchHits.length);
        for (SearchHit hit : searchHits) {
            System.out.println(hit.getId());
            source = hit.getSourceAsMap();
            System.out.println(source.toString());
            docIds.add(hit.getId());
        }

        List xa = ScrollSearch.searchByScrollId(scrollId);
        System.out.println("前10个打印" + xa.size());
        for (SearchHit hit : xa) {
            docIds.add(hit.getId());
        }
        return docIds;
    }

    public static void main(String[] args) {
        Set result = searchfoskeywords("mag", "mag");
        System.out.println(result.size());
        for (String s : result) {
            System.out.println(s);
        }
    }
}