com.ibm.fhir.cql.helpers.DataProviderFactory 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.helpers;

import java.util.HashMap;
import java.util.Map;

import org.opencds.cqf.cql.engine.data.CompositeDataProvider;
import org.opencds.cqf.cql.engine.data.DataProvider;
import org.opencds.cqf.cql.engine.model.ModelResolver;
import org.opencds.cqf.cql.engine.retrieve.RetrieveProvider;

import com.ibm.fhir.cql.Constants;
import com.ibm.fhir.cql.engine.model.FHIRModelResolver;

/**
 * This is a factory that enables creation of a Map of DataProvider objects
 * that support a specified RetrieveProvider and the IBM FHIR Server Model.
 * This is necessary for several reasons, but primarily because the IBM FHIR
 * Server model is spread out across multiple packages and there are hoops to
 * jump through in the CQL Engine to support this.
 */
public class DataProviderFactory {
    public static Map createDataProviders(RetrieveProvider retrieveProvider) {
        String suffix = "";

        // This is a hack to get around the fact that IBM FHIR types are in multiple packages
        // and the CQL engine expects there to be only a single package name
        Map dataProviders = new HashMap<>();
        for (String packageName : FHIRModelResolver.ALL_PACKAGES) {
            ModelResolver modelResolver = new FHIRModelResolver();
            modelResolver.setPackageName(packageName);

            DataProvider provider = new CompositeDataProvider(modelResolver, retrieveProvider);
            dataProviders.put(Constants.FHIR_NS_URI + suffix, provider);
            suffix += "-";
        }
        return dataProviders;
    }
}