com.ibm.fhir.cql.engine.retrieve.SearchParameterMap Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of fhir-cql Show documentation
The newest version!
/*
 * (C) Copyright IBM Corp. 2021
 *
 * SPDX-License-Identifier: Apache-2.0
 */
package com.ibm.fhir.cql.engine.retrieve;

import java.io.Serializable;
import java.util.ArrayList;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import java.util.Set;

import com.ibm.fhir.cql.engine.searchparam.IQueryParameter;
import com.ibm.fhir.cql.engine.searchparam.IQueryParameterAnd;
import com.ibm.fhir.cql.engine.searchparam.IQueryParameterOr;

/**
 * This is a helper class that encapsulates a set of query parameters for a FHIR
 * REST API call including common parameters such as _count.
 */
public class SearchParameterMap implements Serializable {

    private static final long serialVersionUID = 2302049000744568377L;

    private Map>> parameters = new LinkedHashMap<>();
    private Integer count;

    public void count(Integer count) {
        this.count = count;
    }

    public Integer count() {
        return this.count;
    }

    public void put(String key, IQueryParameterOr orParam) {
        List> list = parameters.computeIfAbsent(key, k -> new ArrayList<>());
        list.add(orParam.getParameterValues());
    }

    public void put(String key, IQueryParameterAnd> andParam) {
        andParam.getParameterValues().forEach(p -> {
            put(key, p);
        });
    }

    public void put(String key, IQueryParameter param) {
        List> list = parameters.computeIfAbsent(key, k -> new ArrayList<>());
        List sublist = new ArrayList<>();
        sublist.add(param);
        list.add(sublist);
    }

    public boolean containsKey(String key) {
        return this.parameters.containsKey(key);
    }

    public Set keySet() {
        return this.parameters.keySet();
    }

    public boolean isEmpty() {
        return this.parameters.isEmpty();
    }

    public Set>>> entrySet() {
        return this.parameters.entrySet();
    }

    public List> remove(String theName) {
        return this.parameters.remove(theName);
    }
    
    public String toString() {
        return parameters.toString();
    }
}