Examples with SearchOptions org.sonar.server.es.SearchOptions used on opensource projects

Example 6 with SearchOptions

use of org.sonar.server.es.SearchOptions in project sonarqube by SonarSource.

the class SearchAction method doContextResponse.

protected void doContextResponse(DbSession dbSession, SearchWsRequest request, SearchResult result, SearchResponse.Builder response, RuleQuery query) {
    SearchOptions contextForResponse = loadCommonContext(request);
    writeRules(response, result, contextForResponse);
    if (contextForResponse.getFields().contains("actives")) {
        activeRuleCompleter.completeSearch(dbSession, query, result.rules, response);
    }
}

Example 7 with SearchOptions

use of org.sonar.server.es.SearchOptions in project sonarqube by SonarSource.

the class SearchAction method handle.

@Override
public void handle(Request request, Response response) throws Exception {
    int page = request.mandatoryParamAsInt(Param.PAGE);
    int pageSize = request.mandatoryParamAsInt(Param.PAGE_SIZE);
    SearchOptions options = new SearchOptions().setPage(page, pageSize);
    String query = defaultIfBlank(request.param(Param.TEXT_QUERY), "");
    Set<String> fields = neededFields(request);
    try (DbSession dbSession = dbClient.openSession(false)) {
        OrganizationDto organization = groupWsSupport.findOrganizationByKey(dbSession, request.param(PARAM_ORGANIZATION_KEY));
        userSession.checkLoggedIn().checkPermission(ADMINISTER, organization);
        int limit = dbClient.groupDao().countByQuery(dbSession, organization.getUuid(), query);
        List<GroupDto> groups = dbClient.groupDao().selectByQuery(dbSession, organization.getUuid(), query, options.getOffset(), pageSize);
        List<Integer> groupIds = groups.stream().map(GroupDto::getId).collect(Collectors.toList(groups.size()));
        Map<String, Integer> userCountByGroup = dbClient.groupMembershipDao().countUsersByGroups(dbSession, groupIds);
        JsonWriter json = response.newJsonWriter().beginObject();
        options.writeJson(json, limit);
        writeGroups(json, groups, userCountByGroup, fields);
        json.endObject().close();
    }
}

Example 8 with SearchOptions

use of org.sonar.server.es.SearchOptions in project sonarqube by SonarSource.

the class SearchAction method handle.

@Override
public void handle(Request request, Response response) throws Exception {
    SearchOptions options = new SearchOptions().setPage(request.mandatoryParamAsInt(Param.PAGE), request.mandatoryParamAsInt(Param.PAGE_SIZE));
    List<String> fields = request.paramAsStrings(Param.FIELDS);
    SearchResult<UserDoc> result = userIndex.search(request.param(Param.TEXT_QUERY), options);
    try (DbSession dbSession = dbClient.openSession(false)) {
        List<String> logins = Lists.transform(result.getDocs(), UserDocToLogin.INSTANCE);
        Multimap<String, String> groupsByLogin = dbClient.groupMembershipDao().selectGroupsByLogins(dbSession, logins);
        Map<String, Integer> tokenCountsByLogin = dbClient.userTokenDao().countTokensByLogins(dbSession, logins);
        JsonWriter json = response.newJsonWriter().beginObject();
        options.writeJson(json, result.getTotal());
        List<UserDto> userDtos = dbClient.userDao().selectByOrderedLogins(dbSession, logins);
        writeUsers(json, userDtos, groupsByLogin, tokenCountsByLogin, fields);
        json.endObject().close();
    }
}

Example 9 with SearchOptions

use of org.sonar.server.es.SearchOptions in project sonarqube by SonarSource.

the class IssueIndexDebtTest method facets_on_languages.

@Test
public void facets_on_languages() {
    ComponentDto project = ComponentTesting.newProjectDto(newOrganizationDto());
    ComponentDto file = ComponentTesting.newFileDto(project, null);
    RuleKey ruleKey = RuleKey.of("repo", "X1");
    indexIssues(IssueDocTesting.newDoc("ISSUE1", file).setRuleKey(ruleKey.toString()).setLanguage("xoo").setEffort(10L));
    SearchResult<IssueDoc> result = index.search(newQueryBuilder().build(), new SearchOptions().addFacets(newArrayList("languages")));
    assertThat(result.getFacets().getNames()).containsOnly("languages", FACET_MODE_EFFORT);
    assertThat(result.getFacets().get("languages")).containsOnly(entry("xoo", 10L));
    assertThat(result.getFacets().get(FACET_MODE_EFFORT)).containsOnly(entry("total", 10L));
}

Example 10 with SearchOptions

use of org.sonar.server.es.SearchOptions in project sonarqube by SonarSource.

the class IssueIndexTest method facets_on_severities.

@Test
public void facets_on_severities() {
    ComponentDto project = newProjectDto(newOrganizationDto());
    ComponentDto file = newFileDto(project, null);
    indexIssues(IssueDocTesting.newDoc("ISSUE1", file).setSeverity(Severity.INFO), IssueDocTesting.newDoc("ISSUE2", file).setSeverity(Severity.INFO), IssueDocTesting.newDoc("ISSUE3", file).setSeverity(Severity.MAJOR));
    SearchResult<IssueDoc> result = underTest.search(IssueQuery.builder().build(), new SearchOptions().addFacets(newArrayList("severities")));
    assertThat(result.getFacets().getNames()).containsOnly("severities");
    assertThat(result.getFacets().get("severities")).containsOnly(entry("INFO", 2L), entry("MAJOR", 1L));
}