com.hp.autonomy.searchcomponents.hod.parametricvalues.HodParametricValuesService Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2015 Hewlett-Packard Development Company, L.P.
 * Licensed under the MIT License (the "License"); you may not use this file except in compliance with the License.
 */

package com.hp.autonomy.searchcomponents.hod.parametricvalues;

import com.hp.autonomy.hod.client.api.resource.ResourceIdentifier;
import com.hp.autonomy.hod.client.api.textindex.query.parametric.FieldNames;
import com.hp.autonomy.hod.client.api.textindex.query.parametric.GetParametricValuesRequestBuilder;
import com.hp.autonomy.hod.client.api.textindex.query.parametric.GetParametricValuesService;
import com.hp.autonomy.hod.client.api.textindex.query.parametric.ParametricSort;
import com.hp.autonomy.hod.client.error.HodErrorException;
import com.hp.autonomy.searchcomponents.core.parametricvalues.ParametricValuesService;
import com.hp.autonomy.types.requests.idol.actions.tags.QueryTagCountInfo;
import com.hp.autonomy.types.requests.idol.actions.tags.QueryTagInfo;

import java.util.ArrayList;
import java.util.Collections;
import java.util.HashSet;
import java.util.Set;

public class HodParametricValuesService implements ParametricValuesService {

    private final GetParametricValuesService getParametricValuesService;

    public HodParametricValuesService(final GetParametricValuesService getParametricValuesService) {
        this.getParametricValuesService = getParametricValuesService;
    }

    @Override
    public Set getAllParametricValues(final HodParametricRequest parametricRequest) throws HodErrorException {
        final GetParametricValuesRequestBuilder parametricParams = new GetParametricValuesRequestBuilder()
                .setQueryProfile(parametricRequest.getQueryProfile())
                .setSort(ParametricSort.document_count)
                .setText(parametricRequest.getQueryRestrictions().getQueryText())
                .setFieldText(parametricRequest.getQueryRestrictions().getFieldText())
                .setMaxValues(5);

        if (parametricRequest.getFieldNames().isEmpty()) {
            return Collections.emptySet();
        }

        final FieldNames fieldNames = getParametricValuesService.getParametricValues(parametricRequest.getFieldNames(),
                new ArrayList<>(parametricRequest.getQueryRestrictions().getDatabases()), parametricParams);
        final Set fieldNamesSet = fieldNames.getFieldNames();
        final Set parametricFieldNames = new HashSet<>();

        for (final String name : fieldNamesSet) {
            final Set values = new HashSet<>(fieldNames.getValuesAndCountsForFieldName(name));
            if (!values.isEmpty()) {
                parametricFieldNames.add(new QueryTagInfo(name, values));
            }
        }

        return parametricFieldNames;
    }

}