Examples with Id org.apache.atlas.typesystem.persistence.Id used on opensource projects

Example 6 with Id

use of org.apache.atlas.typesystem.persistence.Id in project incubator-atlas by apache.

the class TypedInstanceToGraphMapper method getClassVertex.

private <V, E> AtlasVertex<V, E> getClassVertex(ITypedReferenceableInstance typedReference) throws EntityNotFoundException {
    AtlasVertex<V, E> referenceVertex = null;
    Id id = null;
    if (typedReference != null) {
        id = getExistingId(typedReference);
        referenceVertex = idToVertexMap.get(id);
        if (referenceVertex == null && id.isAssigned()) {
            referenceVertex = graphHelper.getVertexForGUID(id.id);
        }
    }
    return referenceVertex;
}

Example 7 with Id

use of org.apache.atlas.typesystem.persistence.Id in project incubator-atlas by apache.

the class SoftDeleteHandlerV1Test method assertMaxForTestDisconnectBidirectionalReferences.

@Override
protected void assertMaxForTestDisconnectBidirectionalReferences(final Map<String, String> nameGuidMap) throws Exception {
    // Verify that the Department.employees reference to the deleted employee
    // was disconnected.
    ITypedReferenceableInstance hrDept = metadataService.getEntityDefinition(nameGuidMap.get("hr"));
    List<ITypedReferenceableInstance> employees = (List<ITypedReferenceableInstance>) hrDept.get("employees");
    Assert.assertEquals(employees.size(), 4);
    String maxGuid = nameGuidMap.get("Max");
    for (ITypedReferenceableInstance employee : employees) {
        if (employee.getId()._getId().equals(maxGuid)) {
            assertEquals(employee.getId().getState(), Id.EntityState.DELETED);
        }
    }
    // Verify that the Manager.subordinates still references deleted employee
    ITypedReferenceableInstance jane = metadataService.getEntityDefinition(nameGuidMap.get("Jane"));
    List<ITypedReferenceableInstance> subordinates = (List<ITypedReferenceableInstance>) jane.get("subordinates");
    assertEquals(subordinates.size(), 2);
    for (ITypedReferenceableInstance subordinate : subordinates) {
        if (subordinate.getId()._getId().equals(maxGuid)) {
            assertEquals(subordinate.getId().getState(), Id.EntityState.DELETED);
        }
    }
    // Verify that max's Person.mentor unidirectional reference to john was disconnected.
    ITypedReferenceableInstance john = metadataService.getEntityDefinition(nameGuidMap.get("John"));
    Id mentor = (Id) john.get("mentor");
    assertEquals(mentor._getId(), maxGuid);
    assertEquals(mentor.getState(), Id.EntityState.DELETED);
}

Example 8 with Id

use of org.apache.atlas.typesystem.persistence.Id in project incubator-atlas by apache.

the class QuickStartIT method testViewIsAdded.

@Test
public void testViewIsAdded() throws AtlasServiceException, JSONException {
    Referenceable view = atlasClientV1.getEntity(QuickStart.VIEW_TYPE, AtlasClient.REFERENCEABLE_ATTRIBUTE_NAME, QuickStart.PRODUCT_DIM_VIEW);
    assertEquals(QuickStart.PRODUCT_DIM_VIEW, view.get(AtlasClient.NAME));
    Id productDimId = getTable(QuickStart.PRODUCT_DIM_TABLE).getId();
    Id inputTableId = ((List<Id>) view.get(QuickStart.INPUT_TABLES_ATTRIBUTE)).get(0);
    assertEquals(productDimId, inputTableId);
}

Example 9 with Id

use of org.apache.atlas.typesystem.persistence.Id in project incubator-atlas by apache.

the class RequestContext method recordEntityDelete.

public void recordEntityDelete(String entityId, String typeName) throws AtlasException {
    ClassType type = typeSystem.getDataType(ClassType.class, typeName);
    ITypedReferenceableInstance entity = type.createInstance(new Id(entityId, 0, typeName));
    if (deletedEntityIds.add(entityId)) {
        deletedEntities.add(entity);
    }
}

Example 10 with Id

use of org.apache.atlas.typesystem.persistence.Id in project incubator-atlas by apache.

the class DefaultMetadataServiceTest method testCreateEntityWithUniqueAttributeWithReference.

@Test
public void testCreateEntityWithUniqueAttributeWithReference() throws Exception {
    Referenceable db = createDBEntity();
    String dbId = TestUtils.createInstance(metadataService, db);
    //Assert that there is just 1 audit events and thats for entity create
    assertAuditEvents(dbId, 1);
    assertAuditEvents(dbId, EntityAuditEvent.EntityAuditAction.ENTITY_CREATE);
    Referenceable table = new Referenceable(TestUtils.TABLE_TYPE);
    table.set(NAME, TestUtils.randomString());
    table.set("description", "random table");
    table.set("type", "type");
    table.set("tableType", "MANAGED");
    table.set("database", new Id(dbId, 0, TestUtils.DATABASE_TYPE));
    table.set("databaseComposite", db);
    TestUtils.createInstance(metadataService, table);
    //table create should re-use the db instance created earlier
    String tableDefinitionJson = metadataService.getEntityDefinition(TestUtils.TABLE_TYPE, NAME, (String) table.get(NAME));
    Referenceable tableDefinition = InstanceSerialization.fromJsonReferenceable(tableDefinitionJson, true);
    Referenceable actualDb = (Referenceable) tableDefinition.get("databaseComposite");
    assertEquals(actualDb.getId().id, dbId);
    //Assert that as part table create, db is not created and audit event is not added to db
    assertAuditEvents(dbId, 1);
}