SPEQueryResponse.java

package com.constellio.model.services.search;

import java.util.ArrayList;
import java.util.Collections;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import com.constellio.data.dao.dto.records.FacetValue;
import com.constellio.model.entities.records.Record;
import com.constellio.model.entities.schemas.DataStoreField;

public class SPEQueryResponse {

	private final Map<String, List<FacetValue>> fieldFacetValues;
	private final Map<String, Map<String, Object>> statisticsValues;

	private final Map<String, Integer> queryFacetsValues;

	private final Map<String, Map<String, List<String>>> highlights;

	private final long qtime;

	private final long numFound;

	private final List<Record> records;

	private final Map<Record, Map<Record, Double>> recordsWithMoreLikeThis;

	private final boolean correctlySpelt;
	private final List<String> spellcheckerSuggestions;

	public SPEQueryResponse(List<Record> records, Map<Record, Map<Record, Double>> recordsWithMoreLikeThis) {
		this.fieldFacetValues = new HashMap<>();
		this.statisticsValues = new HashMap<>();
		this.queryFacetsValues = new HashMap<>();
		this.qtime = -1;
		this.numFound = records.size();
		this.records = records;
		this.highlights = new HashMap<>();
		this.correctlySpelt = true;
		this.spellcheckerSuggestions = new ArrayList<>();
		this.recordsWithMoreLikeThis = recordsWithMoreLikeThis;
	}

	public SPEQueryResponse(
			Map<String, List<FacetValue>> fieldFacetValues, Map<String, Map<String, Object>> statisticsValues,
			Map<String, Integer> queryFacetsValues, long qtime,
			long numFound, List<Record> records, Map<String, Map<String, List<String>>> highlights, boolean correctlySpelt,
			List<String> spellcheckerSuggestions, Map<Record, Map<Record, Double>> recordsWithMoreLikeThis) {
		this.fieldFacetValues = fieldFacetValues;
		this.statisticsValues = statisticsValues;
		this.queryFacetsValues = queryFacetsValues;
		this.qtime = qtime;
		this.numFound = numFound;
		this.records = records;
		this.highlights = highlights;
		this.correctlySpelt = correctlySpelt;
		this.spellcheckerSuggestions = spellcheckerSuggestions;
		this.recordsWithMoreLikeThis = recordsWithMoreLikeThis;
	}

	public List<FacetValue> getFieldFacetValues(String metadata) {
		if (fieldFacetValues.containsKey(metadata)) {
			return fieldFacetValues.get(metadata);
		} else {
			return Collections.emptyList();
		}
	}

	public Map<String, Object> getStatValues(DataStoreField field) {
		return this.statisticsValues.get(field.getDataStoreCode());
	}

	public boolean hasQueryFacetResults(String query) {
		Integer result = getQueryFacetCount(query);
		return result != null && result > 0;
	}

	public Integer getQueryFacetCount(String query) {
		return queryFacetsValues.get(query);
	}

	public List<String> getFieldFacetValuesWithResults(String field) {
		List<String> values = new ArrayList<>();

		for (FacetValue facetValue : getFieldFacetValues(field)) {
			if (facetValue.getQuantity() > 0) {
				values.add(facetValue.getValue());
			}
		}

		return values;
	}

	public List<Record> getRecords() {
		return records;
	}

	public long getQtime() {
		return qtime;
	}

	public long getNumFound() {
		return numFound;
	}

	public FacetValue getQueryFacetValue(String value) {
		int count = !queryFacetsValues.containsKey(value) ? 0 : queryFacetsValues.get(value);
		return new FacetValue(value, count);
	}

	public FacetValue getFieldFacetValue(String datastoreCode, String value) {
		if (fieldFacetValues.containsKey(datastoreCode)) {
			for (FacetValue facetValue : getFieldFacetValues(datastoreCode)) {
				if (facetValue.getValue().equals(value)) {
					return facetValue;
				}
			}
		}
		return null;
	}

	public Map<String, List<FacetValue>> getFieldFacetValues() {
		return fieldFacetValues;
	}

	public Map<String, Integer> getQueryFacetsValues() {
		return queryFacetsValues;
	}

	public SPEQueryResponse withModifiedRecordList(List<Record> records) {
		return new SPEQueryResponse(fieldFacetValues, statisticsValues, queryFacetsValues, qtime, numFound, records, null,
				correctlySpelt,
				spellcheckerSuggestions, recordsWithMoreLikeThis);
	}

	public SPEQueryResponse withNumFound(int numFound) {
		return new SPEQueryResponse(fieldFacetValues, statisticsValues, queryFacetsValues, qtime, numFound, records, null,
				correctlySpelt,
				spellcheckerSuggestions, recordsWithMoreLikeThis);
	}

	public Map<String, Map<String, List<String>>> getHighlights() {
		return highlights;
	}

	public boolean isCorrectlySpelt() {
		return correctlySpelt;
	}

	public List<String> getSpellCheckerSuggestions() {
		return spellcheckerSuggestions;
	}

	public Map<String, List<String>> getHighlighting(String recordId) {
		if (highlights == null) {
			return Collections.emptyMap();
		} else {
			return highlights.get(recordId);
		}
	}

	public Map<Record, Map<Record, Double>> getRecordsWithMoreLikeThis() {
		return recordsWithMoreLikeThis;
	}
}