Examples with MultipleLevelOneOrgUnitException net.geoprism.registry.etl.export.dhis2.MultipleLevelOneOrgUnitException used on opensource projects

Example 1 with MultipleLevelOneOrgUnitException

use of net.geoprism.registry.etl.export.dhis2.MultipleLevelOneOrgUnitException in project geoprism-registry by terraframe.

the class DHIS2SynchronizationManager method exportGeoObject.

/**
 * It's important that we try our best to maintain an accurate state between
 * our database and the DHIS2 database. The DHIS2Serailizer will create new
 * ids and save them as externalIds for GeoObjects that do not have
 * externalIds. If our push to DHIS2 for this new GeoObject fails for whatever
 * reason, then we want to rollback our database so that we do not store the
 * id which does not exist in the DHIS2 database.
 *
 * TODO : In the future perhaps we should directly ask DHIS2 if an object
 * exists and then we'll know
 *
 * @param level
 * @param go
 * @return
 * @throws ExportError
 */
@Transaction
private void exportGeoObject(DHIS2SyncConfig dhis2Config, DHIS2SyncLevel level, SortedSet<DHIS2SyncLevel> levels, Long rowIndex, VertexServerGeoObject serverGo, Boolean includeTranslations) throws DHIS2SyncError {
    DHIS2ImportResponse resp = null;
    JsonObject orgUnitJsonTree = null;
    String orgUnitJson = null;
    String externalId = null;
    try {
        externalId = serverGo.getExternalId(dhis2Config.getSystem());
        boolean isNew = (externalId == null);
        if (isNew && level.getSyncType() != DHIS2SyncLevel.Type.ALL) {
            NewGeoObjectInvalidSyncTypeError err = new NewGeoObjectInvalidSyncTypeError();
            err.setGeoObject(serverGo.getDisplayLabel().getValue());
            throw err;
        }
        if (isNew && level.getLevel() == 0 && this.ouLevel1.size() > 0) {
            throw new MultipleLevelOneOrgUnitException();
        }
        GsonBuilder builder = new GsonBuilder();
        builder.registerTypeAdapter(VertexServerGeoObject.class, new DHIS2GeoObjectJsonAdapters.DHIS2Serializer(dhis2, dhis2Config, level, this.date));
        orgUnitJsonTree = builder.create().toJsonTree(serverGo, serverGo.getClass()).getAsJsonObject();
        orgUnitJson = orgUnitJsonTree.toString();
        externalId = serverGo.getExternalId(dhis2Config.getSystem());
        List<NameValuePair> params = new ArrayList<NameValuePair>();
        params.add(new BasicNameValuePair("mergeMode", "MERGE"));
        try {
            JsonObject metadataPayload = new JsonObject();
            JsonArray orgUnits = new JsonArray();
            metadataPayload.add("organisationUnits", orgUnits);
            if (level.getSyncType() == DHIS2SyncLevel.Type.ALL) {
                orgUnits.add(orgUnitJsonTree);
            } else if (level.getSyncType() == DHIS2SyncLevel.Type.RELATIONSHIPS) {
                if (!orgUnitJsonTree.has("parent")) {
                    // Root level types do not have parents.
                    return;
                }
                JsonObject orgUnitRelationships = new JsonObject();
                // These attributes must be included at the requirement of DHIS2 API
                orgUnitRelationships.addProperty("id", orgUnitJsonTree.get("id").getAsString());
                orgUnitRelationships.addProperty("name", orgUnitJsonTree.get("name").getAsString());
                orgUnitRelationships.addProperty("shortName", orgUnitJsonTree.get("shortName").getAsString());
                orgUnitRelationships.addProperty("openingDate", orgUnitJsonTree.get("openingDate").getAsString());
                // We don't want to actually update any of these attributes, so if we can fetch the object from DHIS2
                // then use the values they have in their server instead.
                DHIS2Response orgUnitGetResp = dhis2.entityIdGet(DHIS2Objects.ORGANISATION_UNITS, externalId, params);
                if (orgUnitGetResp.isSuccess()) {
                    JsonObject jo = JsonParser.parseString(orgUnitGetResp.getResponse()).getAsJsonObject();
                    orgUnitRelationships.addProperty("name", jo.get("name").getAsString());
                    orgUnitRelationships.addProperty("shortName", jo.get("shortName").getAsString());
                    orgUnitRelationships.addProperty("openingDate", jo.get("openingDate").getAsString());
                }
                // These attributes are the ones we need to include to change the relationship
                orgUnitRelationships.add("parent", orgUnitJsonTree.get("parent").getAsJsonObject());
                orgUnitRelationships.addProperty("path", orgUnitJsonTree.get("path").getAsString());
                orgUnitRelationships.addProperty("level", orgUnitJsonTree.get("level").getAsInt());
                orgUnits.add(orgUnitRelationships);
            } else if (level.getSyncType() == DHIS2SyncLevel.Type.ORG_UNITS) {
                JsonObject orgUnitAttributes = orgUnitJsonTree.deepCopy();
                // Drop all attributes related to the parent / hierarchy
                orgUnitAttributes.remove("parent");
                orgUnitAttributes.remove("path");
                orgUnitAttributes.remove("level");
                orgUnits.add(orgUnitAttributes);
            }
            ImportStrategy strategy = isNew ? ImportStrategy.CREATE : ImportStrategy.UPDATE;
            params.add(new BasicNameValuePair("importStrategy", strategy.name()));
            resp = dhis2.metadataPost(params, new StringEntity(metadataPayload.toString(), Charset.forName("UTF-8")));
        } catch (InvalidLoginException e) {
            LoginException cgrlogin = new LoginException(e);
            throw cgrlogin;
        } catch (HTTPException e) {
            HttpError cgrhttp = new HttpError(e);
            throw cgrhttp;
        }
        this.service.validateDhis2Response(resp);
        if (isNew && level.getLevel() == 1 && this.ouLevel1.size() > 0) {
            this.ouLevel1.add(new OrganisationUnit());
        }
    } catch (Throwable t) {
        DHIS2SyncError er = new DHIS2SyncError(rowIndex, resp, orgUnitJson, t, serverGo.getCode());
        throw er;
    }
}