All Downloads are FREE. Search and download functionalities are using the official Maven repository.

com.ibm.fhir.cql.engine.rest.retrieve.RestFHIRRetrieveProvider Maven / Gradle / Ivy

Go to download

There is a newer version: 4.11.1
Show newest version
/*
 * (C) Copyright IBM Corp. 2021
 *
 * SPDX-License-Identifier: Apache-2.0
 */
package com.ibm.fhir.cql.engine.rest.retrieve;

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

import javax.ws.rs.core.Response;

import com.ibm.fhir.client.FHIRClient;
import com.ibm.fhir.client.FHIRParameters;
import com.ibm.fhir.client.FHIRResponse;
import com.ibm.fhir.cql.engine.retrieve.SearchParameterFHIRRetrieveProvider;
import com.ibm.fhir.cql.engine.retrieve.SearchParameterMap;
import com.ibm.fhir.cql.engine.searchparam.IQueryParameter;
import com.ibm.fhir.cql.engine.searchparam.SearchParameterResolver;
import com.ibm.fhir.cql.engine.util.FHIRClientUtil;
import com.ibm.fhir.cql.helpers.FHIRBundleCursor;
import com.ibm.fhir.model.resource.Bundle;
import com.ibm.fhir.model.resource.Resource;
import com.ibm.fhir.model.util.ModelSupport;

/**
 * This is an implementation of a retrieve provider for the CQL Engine that uses
 * the IBM FHIR Server REST Client to access data.
 */
public class RestFHIRRetrieveProvider extends SearchParameterFHIRRetrieveProvider {

    private FHIRClient fhirClient;

    public RestFHIRRetrieveProvider(SearchParameterResolver searchParameterResolver, FHIRClient fhirClient) {
        super(searchParameterResolver);
        this.fhirClient = fhirClient;
    }

    @Override
    protected Iterable executeQueries(String dataType, List queries) throws Exception {
        if (queries == null || queries.isEmpty()) {
            return Collections.emptyList();
        }

        List resources = new ArrayList<>();
        for (SearchParameterMap query : queries) {
            Resource resource = executeQuery(dataType, query);
            resources.add(resource);
        }

        List results = new ArrayList<>();
        resources.stream().forEach(resource -> {
            if (resource instanceof Bundle) {
                FHIRBundleCursor cursor = new FHIRBundleCursor(url -> {
                    try {
                        Response response = fhirClient.getWebTarget().path(url).request().get();
                        FHIRClientUtil.handleErrorResponse(response);
                        return response.readEntity(Bundle.class);
                    } catch (Exception ex) {
                        throw new RuntimeException(ex);
                    }
                }, (Bundle) resource);
                
                Iterator it = cursor.iterator();
                while(it.hasNext()) {
                    results.add(it.next());
                }
            } else {
                results.add(resource);
            }
        });

        return results;
    }

    protected Resource executeQuery(String dataType, SearchParameterMap map) throws Exception {
        Resource result = null;

        FHIRParameters query = getFHIRParameters(map);

        if (query.getParameterMap().containsKey("_id")) {
            FHIRResponse response = fhirClient.read(dataType, query.getParameterMap().getFirst("_id"));
            FHIRClientUtil.handleErrorResponse(response);

            Class clazz = ModelSupport.getResourceType(dataType);
            result = response.getResource(clazz);
        } else {
            FHIRResponse response = fhirClient.search(dataType, query);
            FHIRClientUtil.handleErrorResponse(response);
            result = response.getResource(Bundle.class);
        }

        return result;
    }

    protected FHIRParameters getFHIRParameters(SearchParameterMap map) {
        FHIRParameters result = new FHIRParameters();

        Map> parameters = getParameterValueMap(map);

        for (Map.Entry> entry : parameters.entrySet()) {
            result.searchParam(entry.getKey(), entry.getValue().toArray(new String[entry.getValue().size()]));
        }

        if (map.count() != null) {
            result.count(map.count());
        }

        return result;
    }

    private Map> getParameterValueMap(SearchParameterMap map) {
        Map> parameters = new HashMap<>();
        for (Map.Entry>> entry : map.entrySet()) {
            for (List sublist : entry.getValue()) {
                for (IQueryParameter p : sublist) {
                    String name = getModifiedName(entry.getKey(), p);

                    List values = parameters.computeIfAbsent(name, k -> new ArrayList<>());
                    values.add(FHIRClientUtil.urlencode(p.getParameterValue()));
                }
            }
        }
        return parameters;
    }
}
    

    

    

            

    

            



    
    






    
© 2015 - 2024 Weber Informatics LLC | Privacy Policy