Examples with ITypedReferenceableInstance org.apache.atlas.typesystem.ITypedReferenceableInstance used on opensource projects

Example 66 with ITypedReferenceableInstance

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

the class GraphBackedMetadataRepositoryDeleteTestBase method testDeleteMixOfNullAndNonNullGuids.

@Test
public void testDeleteMixOfNullAndNonNullGuids() throws Exception {
    ITypedReferenceableInstance entity1 = compositeMapValueType.createInstance();
    ITypedReferenceableInstance entity2 = compositeMapValueType.createInstance();
    List<String> createEntitiesResult = repositoryService.createEntities(entity1, entity2).getCreatedEntities();
    Assert.assertEquals(createEntitiesResult.size(), 2);
    List<String> guids = Arrays.asList(createEntitiesResult.get(0), null, null, createEntitiesResult.get(1));
    EntityResult deleteEntitiesResult = repositoryService.deleteEntities(guids);
    Assert.assertEquals(deleteEntitiesResult.getDeletedEntities().size(), 2);
    Assert.assertTrue(deleteEntitiesResult.getDeletedEntities().containsAll(createEntitiesResult));
}

Example 67 with ITypedReferenceableInstance

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

the class DefaultGraphPersistenceStrategy method constructClassInstanceId.

@Override
public ITypedReferenceableInstance constructClassInstanceId(ClassType classType, Object value) {
    try {
        AtlasVertex classVertex = (AtlasVertex) value;
        ITypedReferenceableInstance classInstance = classType.createInstance(GraphHelper.getIdFromVertex(classVertex), new String[0]);
        return classType.convert(classInstance, Multiplicity.OPTIONAL);
    } catch (AtlasException e) {
        LOG.error("error while constructing an instance", e);
    }
    return null;
}

Example 68 with ITypedReferenceableInstance

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

the class DefaultMetadataService method updateEntities.

/**
     * Updates an entity, instance of the type based on the guid set.
     *
     * @param entityInstanceDefinition json array of entity definitions
     * @return guids - json array of guids
     */
@Override
public CreateUpdateEntitiesResult updateEntities(String entityInstanceDefinition) throws AtlasException {
    entityInstanceDefinition = ParamChecker.notEmpty(entityInstanceDefinition, "Entity instance definition");
    ITypedReferenceableInstance[] typedInstances = deserializeClassInstances(entityInstanceDefinition);
    CreateUpdateEntitiesResult result = repository.updateEntities(typedInstances);
    onEntitiesAddedUpdated(result.getEntityResult());
    return result;
}

Example 69 with ITypedReferenceableInstance

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

the class DefaultMetadataService method updateEntityByUniqueAttribute.

@Override
public CreateUpdateEntitiesResult updateEntityByUniqueAttribute(String typeName, String uniqueAttributeName, String attrValue, Referenceable updatedEntity) throws AtlasException {
    typeName = ParamChecker.notEmpty(typeName, "typeName");
    uniqueAttributeName = ParamChecker.notEmpty(uniqueAttributeName, "uniqueAttributeName");
    attrValue = ParamChecker.notNull(attrValue, "unique attribute value");
    updatedEntity = ParamChecker.notNull(updatedEntity, "updatedEntity");
    ITypedReferenceableInstance oldInstance = getEntityDefinitionReference(typeName, uniqueAttributeName, attrValue);
    final ITypedReferenceableInstance newInstance = validateAndConvertToTypedInstance(updatedEntity, typeName);
    ((ReferenceableInstance) newInstance).replaceWithNewId(oldInstance.getId());
    CreateUpdateEntitiesResult result = repository.updatePartial(newInstance);
    onEntitiesAddedUpdated(result.getEntityResult());
    return result;
}

Example 70 with ITypedReferenceableInstance

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

the class DefaultMetadataService method getTraitDefinition.

@Override
public IStruct getTraitDefinition(String guid, final String traitName) throws AtlasException {
    guid = ParamChecker.notEmpty(guid, "entity id");
    final ITypedReferenceableInstance instance = repository.getEntityDefinition(guid);
    return instance.getTrait(traitName);
}