Search in sources :

Example 6 with QueryStringQueryBuilder

use of org.graylog.shaded.elasticsearch6.org.elasticsearch.index.query.QueryStringQueryBuilder in project graylog2-server by Graylog2.

the class ESMessageList method applyHighlightingIfActivated.

private void applyHighlightingIfActivated(SearchSourceBuilder searchSourceBuilder, SearchJob job, Query query) {
    if (!allowHighlighting) {
        return;
    }
    final QueryStringQueryBuilder highlightQuery = decoratedHighlightQuery(job, query);
    searchSourceBuilder.highlighter(new HighlightBuilder().requireFieldMatch(false).highlightQuery(highlightQuery).field("*").fragmentSize(0).numOfFragments(0));
}
Also used : QueryStringQueryBuilder(org.graylog.shaded.elasticsearch6.org.elasticsearch.index.query.QueryStringQueryBuilder) HighlightBuilder(org.graylog.shaded.elasticsearch6.org.elasticsearch.search.fetch.subphase.highlight.HighlightBuilder)

Example 7 with QueryStringQueryBuilder

use of org.graylog.shaded.elasticsearch6.org.elasticsearch.index.query.QueryStringQueryBuilder in project graylog2-server by Graylog2.

the class Searches method histogram.

public HistogramResult histogram(String query, DateHistogramInterval interval, String filter, TimeRange range) {
    FilterAggregationBuilder builder = AggregationBuilders.filter(AGG_FILTER).subAggregation(AggregationBuilders.dateHistogram(AGG_HISTOGRAM).field("timestamp").interval(interval.toESInterval())).filter(standardAggregationFilters(range, filter));
    QueryStringQueryBuilder qs = queryStringQuery(query);
    qs.allowLeadingWildcard(configuration.isAllowLeadingWildcardSearches());
    final Set<String> affectedIndices = determineAffectedIndices(range, filter);
    final SearchRequestBuilder srb = c.prepareSearch(affectedIndices.toArray(new String[affectedIndices.size()])).setIndicesOptions(IndicesOptions.lenientExpandOpen()).setQuery(qs).addAggregation(builder);
    final SearchRequest request = srb.request();
    SearchResponse r = c.search(request).actionGet();
    recordEsMetrics(r, range);
    final Filter f = r.getAggregations().get(AGG_FILTER);
    return new DateHistogramResult(f.getAggregations().get(AGG_HISTOGRAM), query, request.source(), interval, r.getTook());
}
Also used : SearchRequest(org.elasticsearch.action.search.SearchRequest) DateHistogramResult(org.graylog2.indexer.results.DateHistogramResult) FilterAggregationBuilder(org.elasticsearch.search.aggregations.bucket.filter.FilterAggregationBuilder) SearchRequestBuilder(org.elasticsearch.action.search.SearchRequestBuilder) Filter(org.elasticsearch.search.aggregations.bucket.filter.Filter) QueryStringQueryBuilder(org.elasticsearch.index.query.QueryStringQueryBuilder) SearchResponse(org.elasticsearch.action.search.SearchResponse)

Example 8 with QueryStringQueryBuilder

use of org.graylog.shaded.elasticsearch6.org.elasticsearch.index.query.QueryStringQueryBuilder in project stash-codesearch-plugin by palantir.

the class SearchServlet method doGet.

@Override
protected void doGet(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
    // Make sure user is logged in
    try {
        validationService.validateAuthenticated();
    } catch (AuthorisationException notLoggedInException) {
        try {
            resp.sendRedirect(propertiesService.getLoginUri(URI.create(req.getRequestURL() + (req.getQueryString() == null ? "" : "?" + req.getQueryString()))).toASCIIString());
        } catch (Exception e) {
            log.error("Unable to redirect unauthenticated user to login page", e);
        }
        return;
    }
    // Query and parse settings
    SearchParams params = SearchParams.getParams(req, DateTimeZone.forTimeZone(propertiesService.getDefaultTimeZone()));
    GlobalSettings globalSettings = settingsManager.getGlobalSettings();
    ImmutableSet.Builder<String> noHighlightBuilder = new ImmutableSet.Builder<String>();
    for (String extension : globalSettings.getNoHighlightExtensions().split(",")) {
        extension = extension.trim().toLowerCase();
        if (!extension.isEmpty()) {
            noHighlightBuilder.add(extension);
        }
    }
    ImmutableSet<String> noHighlight = noHighlightBuilder.build();
    int maxPreviewLines = globalSettings.getMaxPreviewLines();
    int maxMatchLines = globalSettings.getMaxMatchLines();
    int maxFragments = globalSettings.getMaxFragments();
    int pageSize = globalSettings.getPageSize();
    TimeValue searchTimeout = new TimeValue(globalSettings.getSearchTimeout());
    float commitHashBoost = (float) globalSettings.getCommitHashBoost();
    float commitSubjectBoost = (float) globalSettings.getCommitBodyBoost();
    float commitBodyBoost = (float) globalSettings.getCommitBodyBoost();
    float fileNameBoost = (float) globalSettings.getFileNameBoost();
    // Execute ES query
    int pages = 0;
    long totalHits = 0;
    long searchTime = 0;
    SearchHit[] currentHits = {};
    String error = "";
    ArrayList<ImmutableMap<String, Object>> hitArray = new ArrayList<ImmutableMap<String, Object>>(currentHits.length);
    ImmutableMap<String, Object> statistics = ImmutableMap.of();
    if (params.doSearch) {
        // Repo map is null iff user is a system administrator (don't need to validate permissions).
        ImmutableMap<String, Repository> repoMap;
        try {
            validationService.validateForGlobal(Permission.SYS_ADMIN);
            repoMap = null;
        } catch (AuthorisationException e) {
            repoMap = repositoryServiceManager.getRepositoryMap(validationService);
            if (repoMap.isEmpty()) {
                error = "You do not have permissions to access any repositories";
            }
        }
        int startIndex = params.page * pageSize;
        SearchRequestBuilder esReq = es.getClient().prepareSearch(ES_SEARCHALIAS).setFrom(startIndex).setSize(pageSize).setTimeout(searchTimeout).setFetchSource(true);
        if (error != null && !error.isEmpty()) {
            log.warn("Not performing search due to error {}", error);
        } else {
            // Build query source and perform query
            QueryBuilder query = matchAllQuery();
            if (params.searchString != null && !params.searchString.isEmpty()) {
                QueryStringQueryBuilder queryStringQuery = queryString(params.searchString).analyzeWildcard(true).lenient(true).defaultOperator(QueryStringQueryBuilder.Operator.AND);
                if (params.searchCommits) {
                    queryStringQuery.field("commit.subject", commitSubjectBoost).field("commit.hash", commitHashBoost).field("commit.body", commitBodyBoost);
                }
                if (params.searchFilenames) {
                    queryStringQuery.field("file.path", fileNameBoost);
                }
                if (params.searchCode) {
                    queryStringQuery.field("file.contents", 1);
                }
                query = queryStringQuery;
            }
            FilterBuilder filter = andFilter(boolFilter().must(repoMap == null ? matchAllFilter() : sf.aclFilter(repoMap), sf.refFilter(params.refNames.split(",")), sf.projectFilter(params.projectKeys.split(",")), sf.repositoryFilter(params.repoNames.split(",")), sf.extensionFilter(params.extensions.split(",")), sf.authorFilter(params.authorNames.split(","))), sf.dateRangeFilter(params.committedAfter, params.committedBefore));
            FilteredQueryBuilder finalQuery = filteredQuery(query, filter);
            esReq.setQuery(finalQuery).setHighlighterPreTags("\u0001").setHighlighterPostTags("\u0001").addHighlightedField("contents", 1, maxFragments);
            String[] typeArray = {};
            if (params.searchCommits) {
                if (params.searchFilenames || params.searchCode) {
                    typeArray = new String[] { "commit", "file" };
                } else {
                    typeArray = new String[] { "commit" };
                }
            } else if (params.searchFilenames || params.searchCode) {
                typeArray = new String[] { "file" };
            }
            esReq.setTypes(typeArray);
            // Build aggregations if statistics were requested
            if (params.showStatistics) {
                esReq.addAggregation(cardinality("authorCardinality").field("authoremail.untouched").precisionThreshold(1000)).addAggregation(terms("authorRanking").field("authoremail.untouched").size(25)).addAggregation(percentiles("charcountPercentiles").field("charcount").percentiles(PERCENTILES)).addAggregation(extendedStats("charcountStats").field("charcount")).addAggregation(filter("commitCount").filter(typeFilter("commit"))).addAggregation(cardinality("extensionCardinality").field("extension").precisionThreshold(1000)).addAggregation(terms("extensionRanking").field("extension").size(25)).addAggregation(percentiles("linecountPercentiles").field("linecount").percentiles(PERCENTILES)).addAggregation(extendedStats("linecountStats").field("linecount"));
            }
            SearchResponse esResp = null;
            try {
                esResp = esReq.get();
            } catch (SearchPhaseExecutionException e) {
                log.warn("Query failure", e);
                error = "Make sure your query conforms to the Lucene/Elasticsearch query string syntax.";
            }
            if (esResp != null) {
                SearchHits esHits = esResp.getHits();
                totalHits = esHits.getTotalHits();
                pages = (int) Math.min(Integer.MAX_VALUE, (totalHits + pageSize - 1) / pageSize);
                currentHits = esHits.getHits();
                searchTime = esResp.getTookInMillis();
                for (ShardSearchFailure failure : esResp.getShardFailures()) {
                    log.warn("Shard failure {}", failure.reason());
                    if (error == null || error.isEmpty()) {
                        error = "Shard failure: " + failure.reason();
                    }
                }
                Aggregations aggs = esResp.getAggregations();
                if (params.showStatistics && aggs != null && !aggs.asList().isEmpty()) {
                    Cardinality authorCardinality = aggs.get("authorCardinality");
                    Terms authorRanking = aggs.get("authorRanking");
                    Percentiles charcountPercentiles = aggs.get("charcountPercentiles");
                    Filter commitCount = aggs.get("commitCount");
                    ExtendedStats charcountStats = aggs.get("charcountStats");
                    Cardinality extensionCardinality = aggs.get("extensionCardinality");
                    Terms extensionRanking = aggs.get("extensionRanking");
                    Percentiles linecountPercentiles = aggs.get("linecountPercentiles");
                    ExtendedStats linecountStats = aggs.get("linecountStats");
                    statistics = new ImmutableMap.Builder<String, Object>().put("authorCardinality", authorCardinality.getValue()).put("authorRanking", getSoyRankingList(authorRanking, commitCount.getDocCount())).put("charcount", new ImmutableMap.Builder<String, Object>().put("average", charcountStats.getAvg()).put("max", Math.round(charcountStats.getMax())).put("min", Math.round(charcountStats.getMin())).put("percentiles", getSoyPercentileList(charcountPercentiles, PERCENTILES)).put("sum", Math.round(charcountStats.getSum())).build()).put("commitcount", commitCount.getDocCount()).put("extensionCardinality", extensionCardinality.getValue()).put("extensionRanking", getSoyRankingList(extensionRanking, charcountStats.getCount())).put("filecount", charcountStats.getCount()).put("linecount", new ImmutableMap.Builder<String, Object>().put("average", linecountStats.getAvg()).put("max", Math.round(linecountStats.getMax())).put("min", Math.round(linecountStats.getMin())).put("percentiles", getSoyPercentileList(linecountPercentiles, PERCENTILES)).put("sum", Math.round(linecountStats.getSum())).build()).build();
                }
            }
        }
        // Iterate through current page of search hits
        for (SearchHit hit : currentHits) {
            ImmutableMap<String, Object> hitData = searchHitToDataMap(hit, repoMap, maxPreviewLines, maxMatchLines, noHighlight);
            if (hitData != null) {
                hitArray.add(hitData);
            }
        }
    }
    // Render page
    pbs.assembler().resources().requireContext("com.atlassian.auiplugin:aui-date-picker");
    pbs.assembler().resources().requireContext("com.atlassian.auiplugin:aui-experimental-tooltips");
    pbs.assembler().resources().requireWebResource("com.palantir.stash.stash-code-search:scs-resources");
    resp.setContentType("text/html");
    try {
        String queryString = req.getQueryString();
        String fullUri = req.getRequestURI() + "?" + (queryString == null ? "" : queryString.replaceAll("&?page=\\d*", ""));
        ImmutableMap<String, Object> data = new ImmutableMap.Builder<String, Object>().put("pages", pages).put("currentPage", params.page).put("prevParams", params.soyParams).put("doSearch", params.doSearch).put("totalHits", totalHits).put("hitArray", hitArray).put("statistics", statistics).put("error", error).put("fullUri", fullUri).put("baseUrl", propertiesService.getBaseUrl().toASCIIString()).put("resultFrom", Math.min(totalHits, params.page * pageSize + 1)).put("resultTo", Math.min(totalHits, (params.page + 1) * pageSize)).put("searchTime", searchTime).build();
        soyTemplateRenderer.render(resp.getWriter(), "com.palantir.stash.stash-code-search:codesearch-soy", "plugin.page.codesearch.searchPage", data);
    } catch (Exception e) {
        log.error("Error rendering Soy template", e);
    }
}
Also used : SearchHit(org.elasticsearch.search.SearchHit) Aggregations(org.elasticsearch.search.aggregations.Aggregations) FilterBuilder(org.elasticsearch.index.query.FilterBuilder) QueryStringQueryBuilder(org.elasticsearch.index.query.QueryStringQueryBuilder) QueryBuilder(org.elasticsearch.index.query.QueryBuilder) FilteredQueryBuilder(org.elasticsearch.index.query.FilteredQueryBuilder) SearchRequestBuilder(org.elasticsearch.action.search.SearchRequestBuilder) SearchPhaseExecutionException(org.elasticsearch.action.search.SearchPhaseExecutionException) ArrayList(java.util.ArrayList) ExtendedStats(org.elasticsearch.search.aggregations.metrics.stats.extended.ExtendedStats) QueryBuilders.queryString(org.elasticsearch.index.query.QueryBuilders.queryString) QueryStringQueryBuilder(org.elasticsearch.index.query.QueryStringQueryBuilder) QueryBuilder(org.elasticsearch.index.query.QueryBuilder) FilteredQueryBuilder(org.elasticsearch.index.query.FilteredQueryBuilder) ImmutableSet(com.google.common.collect.ImmutableSet) FilterBuilder(org.elasticsearch.index.query.FilterBuilder) SearchHits(org.elasticsearch.search.SearchHits) ShardSearchFailure(org.elasticsearch.action.search.ShardSearchFailure) AuthorisationException(com.atlassian.stash.exception.AuthorisationException) TimeValue(org.elasticsearch.common.unit.TimeValue) SearchRequestBuilder(org.elasticsearch.action.search.SearchRequestBuilder) Cardinality(org.elasticsearch.search.aggregations.metrics.cardinality.Cardinality) Terms(org.elasticsearch.search.aggregations.bucket.terms.Terms) GlobalSettings(com.palantir.stash.codesearch.admin.GlobalSettings) Percentiles(org.elasticsearch.search.aggregations.metrics.percentiles.Percentiles) ServletException(javax.servlet.ServletException) AuthorisationException(com.atlassian.stash.exception.AuthorisationException) SearchPhaseExecutionException(org.elasticsearch.action.search.SearchPhaseExecutionException) IOException(java.io.IOException) ImmutableMap(com.google.common.collect.ImmutableMap) SearchResponse(org.elasticsearch.action.search.SearchResponse) Repository(com.atlassian.stash.repository.Repository) FilterBuilders.matchAllFilter(org.elasticsearch.index.query.FilterBuilders.matchAllFilter) FilterBuilders.boolFilter(org.elasticsearch.index.query.FilterBuilders.boolFilter) Filter(org.elasticsearch.search.aggregations.bucket.filter.Filter) FilterBuilders.andFilter(org.elasticsearch.index.query.FilterBuilders.andFilter) FilterBuilders.typeFilter(org.elasticsearch.index.query.FilterBuilders.typeFilter) FilteredQueryBuilder(org.elasticsearch.index.query.FilteredQueryBuilder) QueryStringQueryBuilder(org.elasticsearch.index.query.QueryStringQueryBuilder)

Example 9 with QueryStringQueryBuilder

use of org.graylog.shaded.elasticsearch6.org.elasticsearch.index.query.QueryStringQueryBuilder in project Anserini by castorini.

the class SearchElastic method search.

public <K> ScoredDocuments search(String queryString) {
    SearchHits results = null;
    String specials = "+-=&|><!(){}[]^\"~*?:\\/";
    for (int i = 0; i < specials.length(); i++) {
        char c = specials.charAt(i);
        queryString = queryString.replace(String.valueOf(c), " ");
    }
    QueryStringQueryBuilder query = QueryBuilders.queryStringQuery(queryString).defaultField("contents").analyzer("english");
    SearchRequest searchRequest = new SearchRequest(args.esIndex);
    SearchSourceBuilder sourceBuilder = new SearchSourceBuilder();
    sourceBuilder.query(query);
    sourceBuilder.size(args.hits);
    sourceBuilder.sort(new ScoreSortBuilder().order(SortOrder.DESC));
    sourceBuilder.sort(new FieldSortBuilder(IndexArgs.ID).order(SortOrder.ASC));
    searchRequest.source(sourceBuilder);
    try {
        SearchResponse searchResponse = client.search(searchRequest, COMMON_OPTIONS);
        results = searchResponse.getHits();
    } catch (Exception e) {
        LOG.error("Exception during ES query: ", e);
    }
    ScoreTiesAdjusterReranker reranker = new ScoreTiesAdjusterReranker();
    return reranker.rerank(ScoredDocuments.fromESDocs(results), null);
}
Also used : SearchRequest(org.elasticsearch.action.search.SearchRequest) ScoreSortBuilder(org.elasticsearch.search.sort.ScoreSortBuilder) ScoreTiesAdjusterReranker(io.anserini.rerank.lib.ScoreTiesAdjusterReranker) FieldSortBuilder(org.elasticsearch.search.sort.FieldSortBuilder) SearchHits(org.elasticsearch.search.SearchHits) QueryStringQueryBuilder(org.elasticsearch.index.query.QueryStringQueryBuilder) IOException(java.io.IOException) CmdLineException(org.kohsuke.args4j.CmdLineException) SearchSourceBuilder(org.elasticsearch.search.builder.SearchSourceBuilder) SearchResponse(org.elasticsearch.action.search.SearchResponse)

Example 10 with QueryStringQueryBuilder

use of org.graylog.shaded.elasticsearch6.org.elasticsearch.index.query.QueryStringQueryBuilder in project benchmark by seelunzi.

the class TestContreller method testSearch.

@RequestMapping("/query")
public void testSearch() {
    // 搜索关键字
    String queryString = "springboot";
    QueryStringQueryBuilder builder = new QueryStringQueryBuilder(queryString);
    Iterable<Article> searchResult = articleSearchRepository.search(builder);
    Iterator<Article> iterator = searchResult.iterator();
    while (iterator.hasNext()) {
        System.out.println(iterator.next());
    }
}
Also used : Article(com.tangtang.elasticsearchexample.domain.Article) QueryStringQueryBuilder(org.elasticsearch.index.query.QueryStringQueryBuilder) RequestMapping(org.springframework.web.bind.annotation.RequestMapping)

Aggregations

QueryStringQueryBuilder (org.elasticsearch.index.query.QueryStringQueryBuilder)8 SearchResponse (org.elasticsearch.action.search.SearchResponse)5 SearchRequest (org.elasticsearch.action.search.SearchRequest)4 IOException (java.io.IOException)3 SearchRequestBuilder (org.elasticsearch.action.search.SearchRequestBuilder)3 SearchHits (org.elasticsearch.search.SearchHits)3 Filter (org.elasticsearch.search.aggregations.bucket.filter.Filter)3 ScoreTiesAdjusterReranker (io.anserini.rerank.lib.ScoreTiesAdjusterReranker)2 ArrayList (java.util.ArrayList)2 BoolQueryBuilder (org.elasticsearch.index.query.BoolQueryBuilder)2 FilterBuilder (org.elasticsearch.index.query.FilterBuilder)2 QueryBuilder (org.elasticsearch.index.query.QueryBuilder)2 RangeQueryBuilder (org.elasticsearch.index.query.RangeQueryBuilder)2 FilterAggregationBuilder (org.elasticsearch.search.aggregations.bucket.filter.FilterAggregationBuilder)2 SearchSourceBuilder (org.elasticsearch.search.builder.SearchSourceBuilder)2 FieldSortBuilder (org.elasticsearch.search.sort.FieldSortBuilder)2 ScoreSortBuilder (org.elasticsearch.search.sort.ScoreSortBuilder)2 CmdLineException (org.kohsuke.args4j.CmdLineException)2 AuthorisationException (com.atlassian.stash.exception.AuthorisationException)1 Repository (com.atlassian.stash.repository.Repository)1