Example usage for org.apache.solr.client.solrj SolrQuery setFields

List of usage examples for org.apache.solr.client.solrj SolrQuery setFields

Introduction

In this page you can find the example usage for org.apache.solr.client.solrj SolrQuery setFields.

Prototype

public SolrQuery setFields(String... fields) 

Source Link

Usage

From source file:org.zenoss.zep.index.impl.solr.SolrEventIndexBackend.java

License:Open Source License

@Override
public SavedSearchProcessor<SolrSavedSearch> savedSearchUuidsProcessor() {
    assertReady();/*from  w  w  w  .  ja  v a2  s . co m*/
    return new SavedSearchProcessor<SolrSavedSearch>() {
        @Override
        public EventSummaryResult result(SolrSavedSearch search, int offset, int limit) throws ZepException {
            SolrQuery query = search.getSolrQuery().getCopy();
            query.setRows(limit);
            query.setStart(offset);
            query.setFields(IndexConstants.FIELD_UUID);
            return execute(query, true);
        }
    };
}

From source file:org.zenoss.zep.index.impl.solr.SolrEventIndexBackend.java

License:Open Source License

private SolrQuery buildSolrQuery(EventFilter filter, EventFilter exclusionFilter, Integer limit, Integer offset,
        List<EventSort> sortList, SolrFieldFilter fieldFilter) throws ZepException {
    final String query = buildQuery(filter, exclusionFilter);
    SolrQuery solrQuery = new SolrQuery().setQuery(query);

    if (limit != null && limit < MAX_RESULTS && limit > 0)
        solrQuery.setRows(limit);/*  w  w  w. j  a  v  a 2s. c o m*/
    else
        solrQuery.setRows(MAX_RESULTS);

    if (offset != null)
        solrQuery.setStart(offset);

    if (sortList == null)
        solrQuery.clearSorts();
    else if (sortList.isEmpty())
        solrQuery.addSort(SortClause.desc(IndexConstants.FIELD_LAST_SEEN_TIME));
    else
        for (EventSort sort : sortList)
            for (SortClause clause : createSortClauses(sort))
                solrQuery.addSort(clause);

    if (fieldFilter != null) {
        switch (fieldFilter) {
        case DEFAULTS:
            break;
        case JUST_UUID:
            solrQuery.setFields(IndexConstants.FIELD_UUID);
            break;
        case UUID_LAST_SEEN_AND_PROTOBUF:
            solrQuery.setFields(IndexConstants.FIELD_UUID, IndexConstants.FIELD_LAST_SEEN_TIME,
                    IndexConstants.FIELD_PROTOBUF);
            break;
        case SEARCH_EVENT_TAG_SEVERITIES:
            solrQuery.setFields(IndexConstants.FIELD_ELEMENT_IDENTIFIER,
                    IndexConstants.FIELD_ELEMENT_SUB_IDENTIFIER, IndexConstants.FIELD_SEVERITY,
                    IndexConstants.FIELD_STATUS, IndexConstants.FIELD_TAGS, IndexConstants.FIELD_COUNT);
            break;
        default:
            throw new IllegalStateException("Unexpected fieldFilter: " + fieldFilter);
        }
    }
    solrQuery.setIncludeScore(false);
    solrQuery.setHighlight(false);
    solrQuery.setTerms(false);
    solrQuery.setFacet(false);

    return solrQuery;
}

From source file:se.nrm.dina.naturarv.portal.Main.java

public static void main(String[] args) {
    try {/*w w w  .  j a v  a 2  s  .  c o  m*/
        HttpSolrClient client = new HttpSolrClient.Builder("http://localhost:8983/solr").build();
        SolrQuery query = new SolrQuery();

        query.setQuery("*");
        query.setFields("clnm");
        query.setStart(0);

        QueryResponse response = client.query(query);
        SolrDocumentList results = response.getResults();
        for (int i = 0; i < results.size(); ++i) {
            System.out.println(results.get(i));
        }
    } catch (SolrServerException | IOException ex) {

    }
}

From source file:uk.ac.ebi.atlas.solr.query.builders.BioentityIdentifierQueryBuilder.java

License:Apache License

private SolrQuery buildQueryObject(String queryString) {

    SolrQuery solrQuery = new SolrQuery(queryString);
    solrQuery.setFields(BIOENTITY_IDENTIFIER_FIELD);
    solrQuery.setParam("group", true);
    solrQuery.setParam("group.field", BIOENTITY_IDENTIFIER_FIELD);
    solrQuery.setParam("group.main", true);
    solrQuery.setRows(MAX_GENE_IDS_TO_FETCH);

    LOGGER.trace("<buildQueryObject> solr query: " + solrQuery.toString());

    return solrQuery;
}

From source file:uk.ac.ebi.atlas.solr.query.SpeciesLookupService.java

License:Apache License

Optional<String> fetchFirstSpecies(String fieldName, String queryToken) {
    LOGGER.debug("fetch first species for " + fieldName + ":" + queryToken);

    SolrQuery query = new SolrQuery(fieldName + ":" + queryToken);

    //fields to be returned, ie: fl=species
    query.setFields(SPECIES_FIELD);
    query.setRows(1);//from  w w w . ja v  a  2  s.c  om

    Collection<String> species = solrServer.query(query, SPECIES_FIELD, false);

    return species.isEmpty() ? Optional.<String>absent() : Optional.of(species.iterator().next());
}

From source file:uk.ac.ebi.intact.dataexchange.psimi.solr.ontology.OntologySearcher.java

License:Apache License

public QueryResponse search(SolrQuery query, String[] fieldNames) throws SolrServerException {

    if (fieldNames != null && fieldNames.length > 0) {
        query.setFields(fieldNames);
    }/*from   w  w  w  . j  a v a2s  . co m*/

    // order by unique id
    query.addSortField(OntologyFieldNames.ID, SolrQuery.ORDER.asc);

    return solrServer.query(query);
}

From source file:uk.ac.ebi.phenotype.imaging.springrest.images.dao.ImagesSolrJ.java

License:Apache License

private SolrDocumentList runQuery(String query, int start, int length) throws SolrServerException {

    SolrQuery solrQuery = new SolrQuery();

    log.debug("solr query=" + query);
    solrQuery.setQuery(query);//from w w w.j  a va 2  s.  co  m
    solrQuery.setStart(start);
    solrQuery.setRows(length);
    solrQuery.setFields("id");

    QueryResponse rsp = null;
    rsp = server.query(solrQuery);

    return rsp.getResults();
}

From source file:uk.ac.ebi.phenotype.service.DiseaseService.java

License:Apache License

/**
 * @return all diseases in the specified <code>diseaseSource</code> (see
 * public string definitions) from the disease core.
 * @param diseaseSource the desired disease source (e.g. DiseaseService.OMIM,
 * DiseaseSource.ORPHANET, etc.)/*from w w  w  . j a  v a  2s.  com*/
 * 
 * @throws SolrServerException
 */
public Set<String> getAllDiseasesInDiseaseSource(String diseaseSource) throws SolrServerException {

    SolrQuery solrQuery = new SolrQuery();
    solrQuery.setQuery("disease_source:\"" + diseaseSource + "\"");
    solrQuery.setFields("disease_id");
    solrQuery.setRows(1000000);
    QueryResponse rsp = solr.query(solrQuery);
    SolrDocumentList res = rsp.getResults();
    HashSet<String> allDiseases = new HashSet();
    for (SolrDocument doc : res) {
        allDiseases.add((String) doc.getFieldValue(DiseaseField.DISEASE_ID));
    }
    return allDiseases;
}

From source file:uk.ac.ebi.phenotype.service.GeneService.java

License:Apache License

/**
 * Return all genes in the gene core matching latestPhenotypeStatus and
 * latestProductionCentre.//from   www.j  a  va 2 s .  c o  m
 * 
 * @param latestPhenotypeStatus
 *            latest phenotype status (i.e. most advanced along the pipeline)
 * @param latestProductionCentre
 *            latest production centre (i.e. most advanced along the pipeline)
 * @return all genes in the gene core matching phenotypeStatus and
 *         productionCentre.
 * @throws SolrServerException
 */
public Set<String> getGenesByLatestPhenotypeStatusAndProductionCentre(String latestPhenotypeStatus,
        String latestProductionCentre) throws SolrServerException {

    SolrQuery solrQuery = new SolrQuery();
    String queryString = "(" + GeneDTO.LATEST_PHENOTYPE_STATUS + ":\"" + latestPhenotypeStatus + "\") AND ("
            + GeneDTO.LATEST_PRODUCTION_CENTRE + ":\"" + latestProductionCentre + "\")";
    solrQuery.setQuery(queryString);
    solrQuery.setRows(1000000);
    solrQuery.setFields(GeneDTO.MGI_ACCESSION_ID);
    QueryResponse rsp = null;
    rsp = solr.query(solrQuery);
    SolrDocumentList res = rsp.getResults();
    HashSet<String> allGenes = new HashSet<String>();
    for (SolrDocument doc : res) {
        allGenes.add((String) doc.getFieldValue(GeneDTO.MGI_ACCESSION_ID));
    }

    log.debug("getGenesByLatestPhenotypeStatusAndProductionCentre: solrQuery = " + queryString);
    return allGenes;
}

From source file:uk.ac.ebi.phenotype.service.GeneService.java

License:Apache License

/**
 * Return all genes in the gene core matching latestPhenotypeStatus and
 * latestPhenotypeCentre.//from   w w w  .  java 2s .c  om
 * 
 * @param latestPhenotypeStatus
 *            latest phenotype status (i.e. most advanced along the pipeline)
 * @param latestPhenotypeCentre
 *            latest phenotype centre (i.e. most advanced along the pipeline)
 * @return all genes in the gene core matching phenotypeStatus and
 *         productionCentre.
 * @throws SolrServerException
 */
public Set<String> getGenesByLatestPhenotypeStatusAndPhenotypeCentre(String latestPhenotypeStatus,
        String latestPhenotypeCentre) throws SolrServerException {

    SolrQuery solrQuery = new SolrQuery();
    String queryString = "(" + GeneDTO.LATEST_PHENOTYPE_STATUS + ":\"" + latestPhenotypeStatus + "\") AND ("
            + GeneDTO.LATEST_PHENOTYPING_CENTRE + ":\"" + latestPhenotypeCentre + "\")";
    solrQuery.setQuery(queryString);
    solrQuery.setRows(1000000);
    solrQuery.setFields(GeneDTO.MGI_ACCESSION_ID);
    QueryResponse rsp = null;
    rsp = solr.query(solrQuery);
    SolrDocumentList res = rsp.getResults();
    HashSet<String> allGenes = new HashSet<String>();
    for (SolrDocument doc : res) {
        allGenes.add((String) doc.getFieldValue(GeneDTO.MGI_ACCESSION_ID));
    }

    log.debug("getGenesByLatestPhenotypeStatusAndPhenotypeCentre: solrQuery = " + queryString);
    return allGenes;
}