com.despegar.integration.mongo.query.ProjectQuery Maven / Gradle / Ivy

Go to download
package com.despegar.integration.mongo.query;

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

import com.despegar.integration.mongo.query.Expression.Arithmetical;
import com.despegar.integration.mongo.query.Expression.Array;
import com.despegar.integration.mongo.query.Expression.Date;

public class ProjectQuery {

    private Map operators = new HashMap();
    private List showProperties = new ArrayList();
    private Boolean showId = Boolean.TRUE;

    public ProjectQuery put(String name, Arithmetical expression) {
        this.addExpression(name, expression);
        return this;
    }

    public ProjectQuery put(String name, Array expression) {
        this.addExpression(name, expression);
        return this;
    }

    public ProjectQuery put(String name, Date expression) {
        this.addExpression(name, expression);
        return this;
    }

    private void addExpression(String name, Expression expression) {
        if (name == null || expression == null) {
            return;
        }
        this.operators.put(name, expression);
    }

    public ProjectQuery put(String name, String property) {
        if (name == null || property == null) {
            return this;
        }
        this.operators.put(name, property);

        return this;
    }

    public ProjectQuery show(String property) {
        if (property == null) {
            return this;
        }
        this.showProperties.add(property);

        return this;
    }

    public ProjectQuery hideId() {
        this.showId = Boolean.FALSE;

        return this;
    }

    public Map getOperators() {
        return this.operators;
    }

    public List getShowProperties() {
        return this.showProperties;
    }

    public Boolean getShowId() {
        return this.showId;
    }
}