com.dottydingo.hyperion.client.builder.QueryRequestBuilder Maven / Gradle / Ivy

Go to download
package com.dottydingo.hyperion.client.builder;

import com.dottydingo.hyperion.api.ApiObject;
import com.dottydingo.hyperion.api.EntityResponse;
import com.dottydingo.hyperion.client.*;

import java.io.Serializable;
import java.util.ArrayList;
import java.util.List;

/**
 */
public class QueryRequestBuilder,ID extends Serializable> extends RequestBuilder
{
    private List sorts = new ArrayList();

    public QueryRequestBuilder(int version, Class objectType, String entityName,String query)
    {
        this(version, objectType, entityName);
        setParameter("query",query);
    }

    public QueryRequestBuilder(int version, Class objectType, String entityName)
    {
        super(version, objectType, entityName);
    }

    public QueryRequestBuilder returnFields(String... fields)
    {
        setParameter("fields",join(fields));
        return this;
    }

    public QueryRequestBuilder start(long start)
    {
        setParameter("start",Long.toString(start));
        return this;
    }

    public QueryRequestBuilder limit(long limit)
    {
        setParameter("limit",Long.toString(limit));
        return this;
    }

    public QueryRequestBuilder addAscendingSort(String field)
    {
        sorts.add(field);
        return this;
    }

    public QueryRequestBuilder addDescendingSort(String field)
    {
        sorts.add(field + ":desc");
        return this;
    }

    @Override
    public QueryRequestBuilder addParameter(String name, String value)
    {
        super.addParameter(name, value);
        return this;
    }

    @Override
    public QueryRequestBuilder setParameter(String name, String value)
    {
        super.setParameter(name, value);
        return this;
    }

    @Override
    public QueryRequestBuilder addHeader(String name, String value)
    {
        super.addHeader(name, value);
        return this;
    }

    @Override
    public QueryRequestBuilder setHeader(String name, String value)
    {
        super.setHeader(name, value);
        return this;
    }

    @Override
    public QueryRequestBuilder withHeaderFactory(HeaderFactory headerFactory)
    {
        super.withHeaderFactory(headerFactory);
        return this;
    }

    @Override
    public QueryRequestBuilder withParameterFactory(ParameterFactory parameterFactory)
    {
        super.withParameterFactory(parameterFactory);
        return this;
    }

    @Override
    public Request build()
    {
        setParameter("sort",join(sorts));

        Request request = super.build();
        request.setRequestMethod(RequestMethod.GET);

        return request;
    }

    public EntityResponse execute(HyperionClient client)
    {
        return client.query(build());
    }
}