Examples with CollectionRequest org.apache.atlas.catalog.CollectionRequest used on opensource projects

Example 1 with CollectionRequest

use of org.apache.atlas.catalog.CollectionRequest in project incubator-atlas by apache.

the class EntityService method getEntities.

@GET
@Produces(Servlets.JSON_MEDIA_TYPE)
public Response getEntities(@Context HttpHeaders headers, @Context UriInfo ui) throws CatalogException {
    AtlasPerfTracer perf = null;
    try {
        if (AtlasPerfTracer.isPerfTraceEnabled(PERF_LOG)) {
            perf = AtlasPerfTracer.getPerfTracer(PERF_LOG, "EntityService.getEntities()");
        }
        String queryString = decode(getQueryString(ui));
        BaseRequest request = new CollectionRequest(Collections.<String, Object>emptyMap(), queryString);
        Result result = getResources(entityResourceProvider, request);
        return Response.status(Response.Status.OK).entity(getSerializer().serialize(result, ui)).build();
    } finally {
        AtlasPerfTracer.log(perf);
    }
}

Example 2 with CollectionRequest

use of org.apache.atlas.catalog.CollectionRequest in project incubator-atlas by apache.

the class EntityService method tagEntities.

@POST
@Produces(Servlets.JSON_MEDIA_TYPE)
public Response tagEntities(String body, @Context HttpHeaders headers, @Context UriInfo ui) throws CatalogException {
    AtlasPerfTracer perf = null;
    try {
        if (AtlasPerfTracer.isPerfTraceEnabled(PERF_LOG)) {
            perf = AtlasPerfTracer.getPerfTracer(PERF_LOG, "EntityService.tagEntities()");
        }
        Map<String, Object> properties = parsePayload(body);
        if (properties.get("tags") == null || properties.size() != 1) {
            throw new CatalogException("Invalid Request, no 'tags' property specified. Creation of entity resource not supported.", 400);
        }
        String queryString = decode(getQueryString(ui));
        Collection<String> createResults = createResources(entityTagResourceProvider, new CollectionRequest(properties, queryString));
        Collection<Results> result = new ArrayList<>();
        for (String relativeUrl : createResults) {
            result.add(new Results(ui.getBaseUri().toString() + relativeUrl, 201));
        }
        return Response.status(Response.Status.CREATED).entity(new GenericEntity<Collection<Results>>(result) {
        }).build();
    } finally {
        AtlasPerfTracer.log(perf);
    }
}

Example 3 with CollectionRequest

use of org.apache.atlas.catalog.CollectionRequest in project incubator-atlas by apache.

the class QueryFactoryTest method testCollectionQuery_WildcardQuery.

@Test
public void testCollectionQuery_WildcardQuery() throws Exception {
    String queryString = "name:ta?onomy";
    Request request = new CollectionRequest(Collections.<String, Object>emptyMap(), queryString);
    QueryFactory factory = new QueryFactory();
    AtlasTaxonomyQuery query = (AtlasTaxonomyQuery) factory.createTaxonomyQuery(request);
    QueryExpression queryExpression = query.getQueryExpression();
    assertEquals(queryExpression.getClass(), WildcardQueryExpression.class);
    assertEquals(queryExpression.getField(), "name");
    assertEquals(queryExpression.getExpectedValue(), "ta?onomy");
    assertEquals(query.getRequest(), request);
    assertEquals(query.getResourceDefinition().getTypeName(), "Taxonomy");
}

Example 4 with CollectionRequest

use of org.apache.atlas.catalog.CollectionRequest in project incubator-atlas by apache.

the class TaxonomyResourceDefinitionTest method testFilterProperties_Collection.

@Test
public void testFilterProperties_Collection() {
    Map<String, Object> resourceProps = new HashMap<>();
    resourceProps.put("id", "111-222-333");
    resourceProps.put("name", "nameVal");
    resourceProps.put("type", "someType");
    resourceProps.put("foo", "fooVal");
    resourceProps.put("bar", "barVal");
    resourceProps.put("description", "desc");
    resourceProps.put("creation_time", "2016:10:10");
    Request request = new CollectionRequest(resourceProps, "someProperty:someValue");
    request.addAdditionalSelectProperties(Collections.singleton("foo"));
    ResourceDefinition taxonomyDefinition = new TaxonomyResourceDefinition();
    Map<String, Object> filteredProps = taxonomyDefinition.filterProperties(request, resourceProps);
    assertEquals(filteredProps.size(), 3);
    // registered collection props
    assertTrue(filteredProps.containsKey("name"));
    assertTrue(filteredProps.containsKey("description"));
    // added prop
    assertTrue(filteredProps.containsKey("foo"));
}

Example 5 with CollectionRequest

use of org.apache.atlas.catalog.CollectionRequest in project incubator-atlas by apache.

the class EntityResourceDefinitionTest method testFilterProperties_Collection.

@Test
public void testFilterProperties_Collection() {
    Map<String, Object> resourceProps = new HashMap<>();
    resourceProps.put("id", "111-222-333");
    resourceProps.put("name", "nameVal");
    resourceProps.put("type", "someType");
    resourceProps.put("foo", "fooVal");
    resourceProps.put("bar", "barVal");
    resourceProps.put("fooBar", "fooBarVal");
    resourceProps.put("other", "otherVal");
    Request request = new CollectionRequest(resourceProps, "someProperty:someValue");
    request.addAdditionalSelectProperties(Collections.singleton("foo"));
    ResourceDefinition entityDefinition = new EntityResourceDefinition();
    // no filtering should occur for entity instances
    Map<String, Object> filteredProps = entityDefinition.filterProperties(request, resourceProps);
    assertEquals(filteredProps.size(), 4);
    // registered collection props
    assertTrue(filteredProps.containsKey("name"));
    assertTrue(filteredProps.containsKey("id"));
    assertTrue(filteredProps.containsKey("type"));
    // added prop
    assertTrue(filteredProps.containsKey("foo"));
}