Search in sources :

Example 1 with OrderParams

use of org.hisp.dhis.dxf2.common.OrderParams in project dhis2-core by dhis2.

the class DefaultMetadataExportService method getParamsFromMap.

@Override
@SuppressWarnings("unchecked")
public MetadataExportParams getParamsFromMap(Map<String, List<String>> parameters) {
    MetadataExportParams params = new MetadataExportParams();
    Map<Class<? extends IdentifiableObject>, Map<String, List<String>>> map = new HashMap<>();
    if (parameters.containsKey("fields")) {
        params.setDefaultFields(parameters.get("fields"));
        parameters.remove("fields");
    }
    if (parameters.containsKey("filter")) {
        params.setDefaultFilter(parameters.get("filter"));
        parameters.remove("filter");
    }
    if (parameters.containsKey("order")) {
        params.setDefaultOrder(parameters.get("order"));
        parameters.remove("order");
    }
    for (String parameterKey : parameters.keySet()) {
        String[] parameter = parameterKey.split(":");
        Schema schema = schemaService.getSchemaByPluralName(parameter[0]);
        if (schema == null || !schema.isIdentifiableObject()) {
            continue;
        }
        Class<? extends IdentifiableObject> klass = (Class<? extends IdentifiableObject>) schema.getKlass();
        // class is enabled if value = true, or fields/filter/order is present
        if ("true".equalsIgnoreCase(parameters.get(parameterKey).get(0)) || (parameter.length > 1 && ("fields".equalsIgnoreCase(parameter[1]) || "filter".equalsIgnoreCase(parameter[1]) || "order".equalsIgnoreCase(parameter[1])))) {
            if (!map.containsKey(klass))
                map.put(klass, new HashMap<>());
        } else {
            continue;
        }
        if (parameter.length > 1) {
            if ("fields".equalsIgnoreCase(parameter[1])) {
                if (!map.get(klass).containsKey("fields"))
                    map.get(klass).put("fields", new ArrayList<>());
                map.get(klass).get("fields").addAll(parameters.get(parameterKey));
            }
            if ("filter".equalsIgnoreCase(parameter[1])) {
                if (!map.get(klass).containsKey("filter"))
                    map.get(klass).put("filter", new ArrayList<>());
                map.get(klass).get("filter").addAll(parameters.get(parameterKey));
            }
            if ("order".equalsIgnoreCase(parameter[1])) {
                if (!map.get(klass).containsKey("order"))
                    map.get(klass).put("order", new ArrayList<>());
                map.get(klass).get("order").addAll(parameters.get(parameterKey));
            }
        }
    }
    map.keySet().forEach(params::addClass);
    for (Class<? extends IdentifiableObject> klass : map.keySet()) {
        Map<String, List<String>> classMap = map.get(klass);
        Schema schema = schemaService.getDynamicSchema(klass);
        if (classMap.containsKey("fields"))
            params.addFields(klass, classMap.get("fields"));
        if (classMap.containsKey("filter") && classMap.containsKey("order")) {
            OrderParams orderParams = new OrderParams(Sets.newHashSet(classMap.get("order")));
            Query query = queryService.getQueryFromUrl(klass, classMap.get("filter"), orderParams.getOrders(schema));
            query.setDefaultOrder();
            params.addQuery(query);
        } else if (classMap.containsKey("filter")) {
            Query query = queryService.getQueryFromUrl(klass, classMap.get("filter"), new ArrayList<>());
            query.setDefaultOrder();
            params.addQuery(query);
        } else if (classMap.containsKey("order")) {
            OrderParams orderParams = new OrderParams();
            orderParams.setOrder(Sets.newHashSet(classMap.get("order")));
            Query query = queryService.getQueryFromUrl(klass, new ArrayList<>(), orderParams.getOrders(schema));
            query.setDefaultOrder();
            params.addQuery(query);
        }
    }
    return params;
}
Also used : Query(org.hisp.dhis.query.Query) HashMap(java.util.HashMap) Schema(org.hisp.dhis.schema.Schema) ArrayList(java.util.ArrayList) IdentifiableObject(org.hisp.dhis.common.IdentifiableObject) List(java.util.List) ArrayList(java.util.ArrayList) OrderParams(org.hisp.dhis.dxf2.common.OrderParams) Map(java.util.Map) SetMap(org.hisp.dhis.common.SetMap) HashMap(java.util.HashMap)

Example 2 with OrderParams

use of org.hisp.dhis.dxf2.common.OrderParams in project dhis2-core by dhis2.

the class DefaultMetadataExportService method getMetadata.

@Override
@SuppressWarnings("unchecked")
public Map<Class<? extends IdentifiableObject>, List<? extends IdentifiableObject>> getMetadata(MetadataExportParams params) {
    Timer timer = new SystemTimer().start();
    Map<Class<? extends IdentifiableObject>, List<? extends IdentifiableObject>> metadata = new HashMap<>();
    if (params.getUser() == null) {
        params.setUser(currentUserService.getCurrentUser());
    }
    if (params.getClasses().isEmpty()) {
        schemaService.getMetadataSchemas().stream().filter(Schema::isIdentifiableObject).forEach(schema -> params.getClasses().add((Class<? extends IdentifiableObject>) schema.getKlass()));
    }
    log.info("(" + params.getUsername() + ") Export:Start");
    for (Class<? extends IdentifiableObject> klass : params.getClasses()) {
        Query query;
        if (params.getQuery(klass) != null) {
            query = params.getQuery(klass);
        } else {
            OrderParams orderParams = new OrderParams(Sets.newHashSet(params.getDefaultOrder()));
            query = queryService.getQueryFromUrl(klass, params.getDefaultFilter(), orderParams.getOrders(schemaService.getDynamicSchema(klass)));
        }
        if (query.getUser() == null) {
            query.setUser(params.getUser());
        }
        query.setDefaultOrder();
        List<? extends IdentifiableObject> objects = queryService.query(query);
        if (!objects.isEmpty()) {
            log.info("(" + params.getUsername() + ") Exported " + objects.size() + " objects of type " + klass.getSimpleName());
            metadata.put(klass, objects);
        }
    }
    log.info("(" + params.getUsername() + ") Export:Done took " + timer.toString());
    return metadata;
}
Also used : SystemTimer(org.hisp.dhis.commons.timer.SystemTimer) Timer(org.hisp.dhis.commons.timer.Timer) Query(org.hisp.dhis.query.Query) HashMap(java.util.HashMap) List(java.util.List) ArrayList(java.util.ArrayList) OrderParams(org.hisp.dhis.dxf2.common.OrderParams) SystemTimer(org.hisp.dhis.commons.timer.SystemTimer) IdentifiableObject(org.hisp.dhis.common.IdentifiableObject)

Aggregations

ArrayList (java.util.ArrayList)2 HashMap (java.util.HashMap)2 List (java.util.List)2 IdentifiableObject (org.hisp.dhis.common.IdentifiableObject)2 OrderParams (org.hisp.dhis.dxf2.common.OrderParams)2 Query (org.hisp.dhis.query.Query)2 Map (java.util.Map)1 SetMap (org.hisp.dhis.common.SetMap)1 SystemTimer (org.hisp.dhis.commons.timer.SystemTimer)1 Timer (org.hisp.dhis.commons.timer.Timer)1 Schema (org.hisp.dhis.schema.Schema)1