Examples with AtlasEntityWithExtInfo org.apache.atlas.model.instance.AtlasEntity.AtlasEntityWithExtInfo used on opensource projects

Example 11 with AtlasEntityWithExtInfo

use of org.apache.atlas.model.instance.AtlasEntity.AtlasEntityWithExtInfo in project incubator-atlas by apache.

the class AtlasEntityStoreV1 method getById.

@Override
@GraphTransaction
public AtlasEntityWithExtInfo getById(String guid) throws AtlasBaseException {
    if (LOG.isDebugEnabled()) {
        LOG.debug("==> getById({})", guid);
    }
    EntityGraphRetriever entityRetriever = new EntityGraphRetriever(typeRegistry);
    AtlasEntityWithExtInfo ret = entityRetriever.toAtlasEntityWithExtInfo(guid);
    if (ret == null) {
        throw new AtlasBaseException(AtlasErrorCode.INSTANCE_GUID_NOT_FOUND, guid);
    }
    if (LOG.isDebugEnabled()) {
        LOG.debug("<== getById({}): {}", guid, ret);
    }
    return ret;
}

Example 12 with AtlasEntityWithExtInfo

use of org.apache.atlas.model.instance.AtlasEntity.AtlasEntityWithExtInfo in project incubator-atlas by apache.

the class ImportTransformsTest method transformEntityWithExtInfo.

@Test
public void transformEntityWithExtInfo() throws AtlasBaseException {
    addColumnTransform(transform);
    AtlasEntityWithExtInfo entityWithExtInfo = getAtlasEntityWithExtInfo();
    AtlasEntity entity = entityWithExtInfo.getEntity();
    String attrValue = (String) entity.getAttribute(qualifiedName);
    String[] expectedValues = getExtEntityExpectedValues(entityWithExtInfo);
    transform.apply(entityWithExtInfo);
    assertEquals(entityWithExtInfo.getEntity().getAttribute(qualifiedName), applyDefaultTransform(attrValue));
    for (int i = 0; i < expectedValues.length; i++) {
        assertEquals(entityWithExtInfo.getReferredEntities().get(Integer.toString(i)).getAttribute(qualifiedName), expectedValues[i]);
    }
}

Example 13 with AtlasEntityWithExtInfo

use of org.apache.atlas.model.instance.AtlasEntity.AtlasEntityWithExtInfo in project incubator-atlas by apache.

the class ImportTransformsTest method transformEntityWithExtInfoNullCheck.

@Test
public void transformEntityWithExtInfoNullCheck() throws AtlasBaseException {
    addColumnTransform(transform);
    AtlasEntityWithExtInfo entityWithExtInfo = getAtlasEntityWithExtInfo();
    entityWithExtInfo.setReferredEntities(null);
    AtlasEntityWithExtInfo transformedEntityWithExtInfo = transform.apply(entityWithExtInfo);
    assertNotNull(transformedEntityWithExtInfo);
    assertEquals(entityWithExtInfo.getEntity().getGuid(), transformedEntityWithExtInfo.getEntity().getGuid());
}

Example 14 with AtlasEntityWithExtInfo

use of org.apache.atlas.model.instance.AtlasEntity.AtlasEntityWithExtInfo in project incubator-atlas by apache.

the class TestEntityREST method testGetEntityWithAssociations.

@Test(dependsOnMethods = "testAddAndGetClassification")
public void testGetEntityWithAssociations() throws Exception {
    AtlasEntityWithExtInfo entity = entityREST.getById(dbEntity.getGuid());
    final List<AtlasClassification> retrievedClassifications = entity.getEntity().getClassifications();
    Assert.assertNotNull(retrievedClassifications);
    Assert.assertEquals(new ArrayList<AtlasClassification>() {

        {
            add(testClassification);
        }
    }, retrievedClassifications);
}

Example 15 with AtlasEntityWithExtInfo

use of org.apache.atlas.model.instance.AtlasEntity.AtlasEntityWithExtInfo in project incubator-atlas by apache.

the class EntityGraphRetriever method toAtlasEntityWithExtInfo.

public AtlasEntityWithExtInfo toAtlasEntityWithExtInfo(AtlasVertex entityVertex) throws AtlasBaseException {
    AtlasEntityExtInfo entityExtInfo = new AtlasEntityExtInfo();
    AtlasEntity entity = mapVertexToAtlasEntity(entityVertex, entityExtInfo);
    AtlasEntityWithExtInfo ret = new AtlasEntityWithExtInfo(entity, entityExtInfo);
    ret.compact();
    return ret;
}