Examples with ServerCodeRestriction net.geoprism.registry.query.ServerCodeRestriction used on opensource projects

Example 1 with ServerCodeRestriction

use of net.geoprism.registry.query.ServerCodeRestriction in project geoprism-registry by terraframe.

the class BusinessObjectImporter method getGeoObject.

/**
 * Returns the entity as defined by the 'parent' and 'parentType' attributes
 * of the given feature. If an entity is not found then Earth is returned by
 * default. The 'parent' value of the feature must define an entity name or a
 * geo oid. The 'parentType' value of the feature must define the localized
 * display label of the universal.
 *
 * This algorithm resolves parent contexts starting at the top of the
 * hierarchy and traversing downward, resolving hierarchy inheritance as
 * needed. If at any point a location cannot be found, a SmartException will
 * be thrown which varies depending on the ParentMatchStrategy.
 *
 * @param feature
 *          Shapefile feature used to determine the parent
 * @return Parent entity
 */
private ServerGeoObjectIF getGeoObject(FeatureRow feature) {
    List<Location> locations = this.configuration.getLocations();
    ServerGeoObjectIF parent = null;
    JSONArray context = new JSONArray();
    ArrayList<String> parentKeyBuilder = new ArrayList<String>();
    for (Location location : locations) {
        Object label = getParentCode(feature, location);
        if (label != null) {
            String key = parent != null ? parent.getCode() + "-" + label : label.toString();
            parentKeyBuilder.add(label.toString());
            if (this.configuration.isExclusion(GeoObjectImportConfiguration.PARENT_EXCLUSION, key)) {
                throw new IgnoreRowException();
            }
            // Check the parent cache
            String parentChainKey = StringUtils.join(parentKeyBuilder, parentConcatToken);
            if (this.cache.containsKey(parentChainKey)) {
                parent = this.cache.get(parentChainKey);
                JSONObject element = new JSONObject();
                element.put("label", label.toString());
                element.put("type", location.getType().getLabel().getValue());
                context.put(element);
                continue;
            }
            final ParentMatchStrategy ms = location.getMatchStrategy();
            // Search
            ServerGeoObjectQuery query = new ServerGeoObjectService().createQuery(location.getType(), this.configuration.getDate());
            query.setLimit(20);
            if (ms.equals(ParentMatchStrategy.CODE)) {
                query.setRestriction(new ServerCodeRestriction(label.toString()));
            } else if (ms.equals(ParentMatchStrategy.EXTERNAL)) {
                query.setRestriction(new ServerExternalIdRestriction(this.getConfiguration().getExternalSystem(), label.toString()));
            } else if (ms.equals(ParentMatchStrategy.DHIS2_PATH)) {
                String path = label.toString();
                String dhis2Parent;
                try {
                    if (path.startsWith("/")) {
                        path = path.substring(1);
                    }
                    String[] pathArr = path.split("/");
                    dhis2Parent = pathArr[pathArr.length - 2];
                } catch (Throwable t) {
                    InvalidDhis2PathException ex = new InvalidDhis2PathException(t);
                    ex.setDhis2Path(path);
                    throw ex;
                }
                query.setRestriction(new ServerExternalIdRestriction(this.getConfiguration().getExternalSystem(), dhis2Parent));
            } else {
                query.setRestriction(new ServerSynonymRestriction(label.toString(), this.configuration.getDate(), parent, location.getHierarchy()));
            }
            List<ServerGeoObjectIF> results = query.getResults();
            if (results != null && results.size() > 0) {
                ServerGeoObjectIF result = null;
                // into the query SQL.
                for (ServerGeoObjectIF loop : results) {
                    if (result != null && !result.getCode().equals(loop.getCode())) {
                        AmbiguousParentException ex = new AmbiguousParentException();
                        ex.setParentLabel(label.toString());
                        ex.setContext(context.toString());
                        throw ex;
                    }
                    result = loop;
                }
                parent = result;
                JSONObject element = new JSONObject();
                element.put("label", label.toString());
                element.put("type", location.getType().getLabel().getValue());
                context.put(element);
                this.cache.put(parentChainKey, parent);
            } else {
                if (ms.equals(ParentMatchStrategy.CODE)) {
                    final ParentCodeException ex = new ParentCodeException();
                    ex.setParentCode(label.toString());
                    ex.setParentType(location.getType().getLabel().getValue());
                    ex.setContext(context.toString());
                    throw ex;
                } else if (ms.equals(ParentMatchStrategy.EXTERNAL)) {
                    final ExternalParentReferenceException ex = new ExternalParentReferenceException();
                    ex.setExternalId(label.toString());
                    ex.setParentType(location.getType().getLabel().getValue());
                    ex.setContext(context.toString());
                    throw ex;
                } else {
                    String parentCode = (parent == null) ? null : parent.getCode();
                    ParentReferenceProblem prp = new ParentReferenceProblem(location.getType().getCode(), label.toString(), parentCode, context.toString());
                    prp.addAffectedRowNumber(this.progressListener.getWorkProgress() + 1);
                    prp.setHistoryId(this.configuration.historyId);
                    this.progressListener.addReferenceProblem(prp);
                }
                return null;
            }
        }
    }
    return parent;
}

Example 2 with ServerCodeRestriction

use of net.geoprism.registry.query.ServerCodeRestriction in project geoprism-registry by terraframe.

the class GeoObjectImporter method getParent.

/**
 * Returns the entity as defined by the 'parent' and 'parentType' attributes
 * of the given feature. If an entity is not found then Earth is returned by
 * default. The 'parent' value of the feature must define an entity name or a
 * geo oid. The 'parentType' value of the feature must define the localized
 * display label of the universal.
 *
 * This algorithm resolves parent contexts starting at the top of the
 * hierarchy and traversing downward, resolving hierarchy inheritance as
 * needed. If at any point a location cannot be found, a SmartException will
 * be thrown which varies depending on the ParentMatchStrategy.
 *
 * @param feature
 *          Shapefile feature used to determine the parent
 * @return Parent entity
 */
private ServerGeoObjectIF getParent(FeatureRow feature) {
    List<Location> locations = this.configuration.getLocations();
    ServerGeoObjectIF parent = null;
    JSONArray context = new JSONArray();
    ArrayList<String> parentKeyBuilder = new ArrayList<String>();
    for (Location location : locations) {
        Object label = getParentCode(feature, location);
        if (label != null) {
            String key = parent != null ? parent.getCode() + "-" + label : label.toString();
            parentKeyBuilder.add(label.toString());
            if (this.configuration.isExclusion(GeoObjectImportConfiguration.PARENT_EXCLUSION, key)) {
                throw new IgnoreRowException();
            }
            // Check the parent cache
            String parentChainKey = StringUtils.join(parentKeyBuilder, parentConcatToken);
            if (this.parentCache.containsKey(parentChainKey)) {
                parent = this.parentCache.get(parentChainKey);
                JSONObject element = new JSONObject();
                element.put("label", label.toString());
                element.put("type", location.getType().getLabel().getValue());
                context.put(element);
                continue;
            }
            final ParentMatchStrategy ms = location.getMatchStrategy();
            // Search
            ServerGeoObjectQuery query = this.service.createQuery(location.getType(), this.configuration.getStartDate());
            query.setLimit(20);
            if (ms.equals(ParentMatchStrategy.CODE)) {
                query.setRestriction(new ServerCodeRestriction(label.toString()));
            } else if (ms.equals(ParentMatchStrategy.EXTERNAL)) {
                query.setRestriction(new ServerExternalIdRestriction(this.getConfiguration().getExternalSystem(), label.toString()));
            } else if (ms.equals(ParentMatchStrategy.DHIS2_PATH)) {
                String path = label.toString();
                String dhis2Parent;
                try {
                    if (path.startsWith("/")) {
                        path = path.substring(1);
                    }
                    String[] pathArr = path.split("/");
                    dhis2Parent = pathArr[pathArr.length - 2];
                } catch (Throwable t) {
                    InvalidDhis2PathException ex = new InvalidDhis2PathException(t);
                    ex.setDhis2Path(path);
                    throw ex;
                }
                query.setRestriction(new ServerExternalIdRestriction(this.getConfiguration().getExternalSystem(), dhis2Parent));
            } else {
                query.setRestriction(new ServerSynonymRestriction(label.toString(), this.configuration.getStartDate(), parent, location.getHierarchy()));
            }
            List<ServerGeoObjectIF> results = query.getResults();
            if (results != null && results.size() > 0) {
                ServerGeoObjectIF result = null;
                // into the query SQL.
                for (ServerGeoObjectIF loop : results) {
                    if (result != null && !result.getCode().equals(loop.getCode())) {
                        AmbiguousParentException ex = new AmbiguousParentException();
                        ex.setParentLabel(label.toString());
                        ex.setContext(context.toString());
                        throw ex;
                    }
                    result = loop;
                }
                parent = result;
                JSONObject element = new JSONObject();
                element.put("label", label.toString());
                element.put("type", location.getType().getLabel().getValue());
                context.put(element);
                this.parentCache.put(parentChainKey, parent);
            } else {
                if (context.length() == 0) {
                    GeoObject root = this.configuration.getRoot();
                    if (root != null) {
                        JSONObject element = new JSONObject();
                        element.put("label", root.getLocalizedDisplayLabel());
                        element.put("type", root.getType().getLabel().getValue());
                        context.put(element);
                    }
                }
                if (ms.equals(ParentMatchStrategy.CODE)) {
                    final ParentCodeException ex = new ParentCodeException();
                    ex.setParentCode(label.toString());
                    ex.setParentType(location.getType().getLabel().getValue());
                    ex.setContext(context.toString());
                    throw ex;
                } else if (ms.equals(ParentMatchStrategy.EXTERNAL)) {
                    final ExternalParentReferenceException ex = new ExternalParentReferenceException();
                    ex.setExternalId(label.toString());
                    ex.setParentType(location.getType().getLabel().getValue());
                    ex.setContext(context.toString());
                    throw ex;
                } else {
                    String parentCode = (parent == null) ? null : parent.getCode();
                    ParentReferenceProblem prp = new ParentReferenceProblem(location.getType().getCode(), label.toString(), parentCode, context.toString());
                    prp.addAffectedRowNumber(this.progressListener.getWorkProgress() + 1);
                    prp.setHistoryId(this.configuration.historyId);
                    this.progressListener.addReferenceProblem(prp);
                }
                return null;
            }
        }
    }
    return parent;
}

Example 3 with ServerCodeRestriction

use of net.geoprism.registry.query.ServerCodeRestriction in project geoprism-registry by terraframe.

the class GeoObjectImporter method parsePostalCode.

private ServerGeoObjectIF parsePostalCode(FeatureRow feature) {
    LocationBuilder builder = PostalCodeFactory.get(this.configuration.getType());
    Location location = builder.build(this.configuration.getFunction(GeoObject.CODE));
    ShapefileFunction function = location.getFunction();
    String code = (String) function.getValue(feature);
    if (code != null) {
        // Search
        ServerGeoObjectQuery query = new ServerGeoObjectService().createQuery(location.getType(), this.configuration.getStartDate());
        query.setRestriction(new ServerCodeRestriction(code));
        // Assert.assertNull(query.getSingleResult());
        ServerGeoObjectIF result = query.getSingleResult();
        if (result != null) {
            return result;
        } else {
            PostalCodeLocationException e = new PostalCodeLocationException();
            e.setCode(code);
            e.setTypeLabel(location.getType().getLabel().getValue());
            throw e;
        }
    }
    return null;
}

Example 4 with ServerCodeRestriction

use of net.geoprism.registry.query.ServerCodeRestriction in project geoprism-registry by terraframe.

the class GeoObjectQueryTest method testLeafCodeRestriction.

@Test
@Request
public void testLeafCodeRestriction() {
    ServerGeoObjectType type = USATestData.DISTRICT.getServerObject();
    VertexGeoObjectQuery query = new VertexGeoObjectQuery(type, null);
    query.setRestriction(new ServerCodeRestriction(USATestData.CO_D_ONE.getCode()));
    ServerGeoObjectIF result = query.getSingleResult();
    Assert.assertEquals(USATestData.CO_D_ONE.getCode(), result.getCode());
    Assert.assertNotNull(result.getGeometry());
    Assert.assertEquals(USATestData.CO_D_ONE.getDisplayLabel(), result.getDisplayLabel().getValue());
    Assert.assertEquals(true, result.getExists());
}

Example 5 with ServerCodeRestriction

use of net.geoprism.registry.query.ServerCodeRestriction in project geoprism-registry by terraframe.

the class ShapefileServiceTest method testImportShapefileWithBadParentCode.

@Test
@Request
public void testImportShapefileWithBadParentCode() throws Throwable {
    GeoObject geoObj = ServiceFactory.getRegistryService().newGeoObjectInstance(testData.clientRequest.getSessionId(), USATestData.COUNTRY.getCode());
    geoObj.setCode("00");
    geoObj.setDisplayLabel(LocalizedValue.DEFAULT_LOCALE, "Test Label");
    geoObj.setUid(ServiceFactory.getIdService().getUids(1)[0]);
    ServerGeoObjectIF serverGO = new ServerGeoObjectService(new AllowAllGeoObjectPermissionService()).apply(geoObj, TestDataSet.DEFAULT_OVER_TIME_DATE, TestDataSet.DEFAULT_END_TIME_DATE, true, false);
    geoObj = RegistryService.getInstance().getGeoObjectByCode(Session.getCurrentSession().getOid(), serverGO.getCode(), serverGO.getType().getCode(), TestDataSet.DEFAULT_OVER_TIME_DATE);
    InputStream istream = this.getClass().getResourceAsStream("/cb_2017_us_state_500k.zip.test");
    Assert.assertNotNull(istream);
    ShapefileService service = new ShapefileService();
    GeoObjectImportConfiguration config = this.getTestConfiguration(istream, service, null, ImportStrategy.NEW_AND_UPDATE);
    ServerHierarchyType hierarchyType = ServerHierarchyType.get(USATestData.HIER_ADMIN.getCode());
    config.setHierarchy(hierarchyType);
    config.addParent(new Location(USATestData.COUNTRY.getServerObject(), hierarchyType, new BasicColumnFunction("GEOID"), ParentMatchStrategy.CODE));
    ImportHistory hist = mockImport(config);
    Assert.assertTrue(hist.getStatus().get(0).equals(AllJobStatus.FEEDBACK));
    hist = ImportHistory.get(hist.getOid());
    Assert.assertEquals(new Long(56), hist.getWorkTotal());
    Assert.assertEquals(new Long(56), hist.getWorkProgress());
    Assert.assertEquals(new Long(0), hist.getImportedRecords());
    Assert.assertEquals(ImportStage.VALIDATION_RESOLVE, hist.getStage().get(0));
    final GeoObjectImportConfiguration test = new GeoObjectImportConfiguration();
    test.fromJSON(hist.getConfigJson(), false);
    // TODO
    // Assert.assertEquals(config.getParentLookupType(),
    // test.getParentLookupType());
    // JSONArray errors = new JSONArray(hist.getErrorJson());
    // 
    // Assert.assertEquals(0, errors.length());
    // Ensure the geo objects were not created
    ServerGeoObjectQuery query = new ServerGeoObjectService().createQuery(USATestData.STATE.getServerObject(), config.getStartDate());
    query.setRestriction(new ServerCodeRestriction("01"));
    Assert.assertNull(query.getSingleResult());
}