Examples with Referenceable org.apache.atlas.v1.model.instance.Referenceable used on opensource projects

Example 1 with Referenceable

use of org.apache.atlas.v1.model.instance.Referenceable in project atlas by apache.

the class AtlasEntityChangeNotifier method onClassificationDeletedFromEntity.

public void onClassificationDeletedFromEntity(AtlasEntity entity, List<String> deletedClassificationNames) throws AtlasBaseException {
    if (isV2EntityNotificationEnabled()) {
        doFullTextMapping(entity.getGuid());
        for (EntityChangeListenerV2 listener : entityChangeListenersV2) {
            listener.onClassificationsDeleted(entity, deletedClassificationNames);
        }
    } else {
        doFullTextMapping(entity.getGuid());
        Referenceable entityRef = toReferenceable(entity.getGuid());
        if (entityRef == null || CollectionUtils.isEmpty(deletedClassificationNames)) {
            return;
        }
        for (EntityChangeListener listener : entityChangeListeners) {
            try {
                listener.onTraitsDeleted(entityRef, deletedClassificationNames);
            } catch (AtlasException e) {
                throw new AtlasBaseException(AtlasErrorCode.NOTIFICATION_FAILED, e, getListenerName(listener), "TraitDelete");
            }
        }
    }
}

Example 2 with Referenceable

use of org.apache.atlas.v1.model.instance.Referenceable in project atlas by apache.

the class AtlasEntityChangeNotifier method onClassificationAddedToEntity.

public void onClassificationAddedToEntity(AtlasEntity entity, List<AtlasClassification> addedClassifications) throws AtlasBaseException {
    if (isV2EntityNotificationEnabled()) {
        doFullTextMapping(entity.getGuid());
        for (EntityChangeListenerV2 listener : entityChangeListenersV2) {
            listener.onClassificationsAdded(entity, addedClassifications);
        }
    } else {
        updateFullTextMapping(entity.getGuid(), addedClassifications);
        Referenceable entityRef = toReferenceable(entity.getGuid());
        List<Struct> traits = toStruct(addedClassifications);
        if (entity == null || CollectionUtils.isEmpty(traits)) {
            return;
        }
        for (EntityChangeListener listener : entityChangeListeners) {
            try {
                listener.onTraitsAdded(entityRef, traits);
            } catch (AtlasException e) {
                throw new AtlasBaseException(AtlasErrorCode.NOTIFICATION_FAILED, e, getListenerName(listener), "TraitAdd");
            }
        }
    }
}

Example 3 with Referenceable

use of org.apache.atlas.v1.model.instance.Referenceable in project atlas by apache.

the class AtlasEntityChangeNotifier method onClassificationUpdatedToEntity.

public void onClassificationUpdatedToEntity(AtlasEntity entity, List<AtlasClassification> updatedClassifications) throws AtlasBaseException {
    if (isV2EntityNotificationEnabled()) {
        doFullTextMapping(entity.getGuid());
        for (EntityChangeListenerV2 listener : entityChangeListenersV2) {
            listener.onClassificationsUpdated(entity, updatedClassifications);
        }
    } else {
        doFullTextMapping(entity.getGuid());
        Referenceable entityRef = toReferenceable(entity.getGuid());
        List<Struct> traits = toStruct(updatedClassifications);
        if (entityRef == null || CollectionUtils.isEmpty(traits)) {
            return;
        }
        for (EntityChangeListener listener : entityChangeListeners) {
            try {
                listener.onTraitsUpdated(entityRef, traits);
            } catch (AtlasException e) {
                throw new AtlasBaseException(AtlasErrorCode.NOTIFICATION_FAILED, e, getListenerName(listener), "TraitUpdate");
            }
        }
    }
}

Example 4 with Referenceable

use of org.apache.atlas.v1.model.instance.Referenceable in project atlas by apache.

the class AuditRepositoryTestBase method testAddEvents.

@Test
public void testAddEvents() throws Exception {
    EntityAuditEvent event = new EntityAuditEvent(rand(), System.currentTimeMillis(), "u1", EntityAuditEvent.EntityAuditAction.ENTITY_CREATE, "d1", new Referenceable(rand()));
    eventRepository.putEventsV1(event);
    List<EntityAuditEvent> events = eventRepository.listEventsV1(event.getEntityId(), null, (short) 10);
    assertEquals(events.size(), 1);
    assertEventEquals(events.get(0), event);
}

Example 5 with Referenceable

use of org.apache.atlas.v1.model.instance.Referenceable in project atlas by apache.

the class AuditRepositoryTestBase method testListPagination.

@Test
public void testListPagination() throws Exception {
    String id1 = "id1" + rand();
    String id2 = "id2" + rand();
    String id3 = "id3" + rand();
    long ts = System.currentTimeMillis();
    Referenceable entity = new Referenceable(rand());
    List<EntityAuditEvent> expectedEvents = new ArrayList<>(3);
    for (int i = 0; i < 3; i++) {
        // Add events for both ids
        EntityAuditEvent event = new EntityAuditEvent(id2, ts - i, "user" + i, EntityAuditEvent.EntityAuditAction.ENTITY_UPDATE, "details" + i, entity);
        eventRepository.putEventsV1(event);
        expectedEvents.add(event);
        eventRepository.putEventsV1(new EntityAuditEvent(id1, ts - i, "user" + i, EntityAuditEvent.EntityAuditAction.TAG_DELETE, "details" + i, entity));
        eventRepository.putEventsV1(new EntityAuditEvent(id3, ts - i, "user" + i, EntityAuditEvent.EntityAuditAction.TAG_ADD, "details" + i, entity));
    }
    // Use ts for which there is no event - ts + 2
    List<EntityAuditEvent> events = eventRepository.listEventsV1(id2, null, (short) 3);
    assertEquals(events.size(), 3);
    assertEventEquals(events.get(0), expectedEvents.get(0));
    assertEventEquals(events.get(1), expectedEvents.get(1));
    assertEventEquals(events.get(2), expectedEvents.get(2));
    // Use last event's timestamp for next list(). Should give only 1 event and shouldn't include events from other id
    events = eventRepository.listEventsV1(id2, events.get(2).getEventKey(), (short) 3);
    assertEquals(events.size(), 1);
    assertEventEquals(events.get(0), expectedEvents.get(2));
}