sdmx.gateway.util.CodelistUtil Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of sdmx-gateway Show documentation
experimental sdmx gateway
The newest version!
/*
 * To change this license header, choose License Headers in Project Properties.
 * To change this template file, choose Tools | Templates
 * and open the template in the editor.
 */
package sdmx.gateway.util;

import java.util.ArrayList;
import java.util.List;
import javax.persistence.EntityManager;
import javax.persistence.Query;
import sdmx.commonreferences.IDType;
import sdmx.commonreferences.NCNameID;
import sdmx.commonreferences.NestedNCNameID;
import sdmx.commonreferences.Version;
import sdmx.gateway.entities.Codelist;
import sdmx.structure.base.ItemType;
import sdmx.structure.codelist.CodelistType;
import sdmx.structure.datastructure.DataStructureType;

/**
 *
 * @author James
 */
public class CodelistUtil {

    public static sdmx.gateway.entities.Codelist createDatabaseCodelist(EntityManager em, CodelistType c) {
        sdmx.gateway.entities.Codelist cl = new sdmx.gateway.entities.Codelist();
        sdmx.gateway.entities.CodelistPK pk = new sdmx.gateway.entities.CodelistPK();
        pk.setAgencyid(c.getAgencyID().toString());
        pk.setId(c.getId().toString());
        pk.setVersion(c.getVersion().toString());
        cl.setAnnotations(AnnotationsUtil.toDatabaseAnnotations(em,c.getAnnotations()));
        cl.setCodelistPK(pk);
        List codes = new ArrayList<>();
        for (int i = 0; i < c.size(); i++) {
            codes.add(CodeUtil.createDatabaseCode(em, cl, c.getCode(i)));
        }
        NameUtil.setName(em, cl, c);
        cl.setCodeList(codes);
        return cl;
    }

    public static Codelist findDatabaseCodelist(EntityManager em, String agency, String id, String version) {
        try {
            Query q = em.createQuery("select c from Codelist c where c.codelistPK.agencyid=:agency and c.codelistPK.id=:id and c.codelistPK.version=:version");
            q.setParameter("agency", agency);
            q.setParameter("id", id);
            q.setParameter("version", version);
            return (sdmx.gateway.entities.Codelist) q.getSingleResult();
        } catch (Exception ex) {
            return null;
        }
    }

    public static List searchCodelist(EntityManager em, String agency, String id, String version) {
        if ("*".equals(version) && "all".equals(id) && "all".equals(agency)) {
            Query q = em.createQuery("select c from Codelist c");
            return (List) q.getResultList();
        } else if ("all".equals(id) && "all".equals(agency)) {
            Query q = em.createQuery("select c from Codelist c where c.codelistPK.version=:version");
            q.setParameter("version", version);
            return (List) q.getResultList();
        } else if ("*".equals(version) && "all".equals(id)) {
            Query q = em.createQuery("select c from Codelist c where c.codelistPK.agencyID=:agency");
            q.setParameter("agency", agency);
            return (List) q.getResultList();
        } else if ("*".equals(version) && "all".equals(agency)) {
            Query q = em.createQuery("select c from Codelist c where c.codelistPK.id=:id");
            q.setParameter("id", id);
            return (List) q.getResultList();
        } else if ("*".equals(version)) {
            Query q = em.createQuery("select c from Codelist c where c.codelistPK.agencyid=:agency and c.codelistPK.id=:id");
            q.setParameter("agency", agency);
            q.setParameter("id", id);
            return (List) q.getResultList();
        } else if ("all".equals(id)) {
            Query q = em.createQuery("select c from Codelist c where c.codelistPK.agencyid=:agency and c.codelistPK.version=:version");
            q.setParameter("agency", agency);
            q.setParameter("version", version);
            return (List) q.getResultList();
        } else if ("all".equals(agency)) {
            Query q = em.createQuery("select c from Codelist c where c.codelistPK.id=:id and c.codelistPK.version=:version");
            q.setParameter("id", id);
            q.setParameter("version", version);
            return (List) q.getResultList();
        } else {
            Query q = em.createQuery("select c from Codelist c where c.codelistPK.agencyid=:agency and c.codelistPK.id=:id and c.codelistPK.version=:version");
            q.setParameter("agency", agency);
            q.setParameter("id", id);
            q.setParameter("version", version);
            return (List) q.getResultList();
        }
    }

    public static CodelistType toSDMXCodelist(sdmx.gateway.entities.Codelist c) {
        CodelistType cl = new CodelistType();
        cl.setNames(NameUtil.toSDMXName(c.getName()));
        cl.setAgencyID(new NestedNCNameID(c.getCodelistPK().getAgencyid()));
        cl.setId(new NCNameID(c.getCodelistPK().getId()));
        cl.setVersion(new Version(c.getCodelistPK().getVersion()));
        List codes = c.getCodeList();
        List items = new ArrayList();
        for (sdmx.gateway.entities.Code cd : codes) {
            items.add(CodeUtil.toSDMXCode(cd));
        }
        cl.setItems(items);
        return cl;
    }

    public static List toSDMXCodelist(List cls) {
        List result = new ArrayList<>();
        for (sdmx.gateway.entities.Codelist c : cls) {
            result.add(toSDMXCodelist(c));
        }
        return result;
    }
    public static void stub(CodelistType cl) {
        cl.setItems(null);
    
    }
}