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

Example 66 with Referenceable

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

the class BaseResourceIT method createHiveDBInstanceV1.

protected Referenceable createHiveDBInstanceV1(String dbName) {
    Referenceable databaseInstance = new Referenceable(DATABASE_TYPE);
    databaseInstance.set(NAME, dbName);
    databaseInstance.set(DESCRIPTION, "foo database");
    databaseInstance.set(CLUSTER_NAME, "fooCluster");
    return databaseInstance;
}

Example 67 with Referenceable

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

the class DataSetLineageJerseyResourceIT method database.

Id database(String name, String description, String owner, String locationUri, String... traitNames) throws Exception {
    Referenceable referenceable = new Referenceable(DATABASE_TYPE, traitNames);
    referenceable.set(NAME, name);
    referenceable.set(QUALIFIED_NAME, name);
    referenceable.set(CLUSTER_NAME, locationUri + name);
    referenceable.set("description", description);
    referenceable.set("owner", owner);
    referenceable.set("locationUri", locationUri);
    referenceable.set("createTime", System.currentTimeMillis());
    return createInstance(referenceable);
}

Example 68 with Referenceable

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

the class KafkaConsumerTest method testReceive.

@Test
public void testReceive() throws Exception {
    Referenceable entity = getEntity(TRAIT_NAME);
    EntityUpdateRequest message = new EntityUpdateRequest("user1", entity);
    String json = AtlasType.toV1Json(new AtlasNotificationMessage<>(new MessageVersion("1.0.0"), message));
    TopicPartition tp = new TopicPartition("ATLAS_HOOK", 0);
    List<ConsumerRecord<String, String>> klist = Collections.singletonList(new ConsumerRecord<>("ATLAS_HOOK", 0, 0L, "mykey", json));
    Map mp = Collections.singletonMap(tp, klist);
    ConsumerRecords records = new ConsumerRecords(mp);
    when(kafkaConsumer.poll(100)).thenReturn(records);
    kafkaConsumer.assign(Collections.singletonList(tp));
    AtlasKafkaConsumer consumer = new AtlasKafkaConsumer(NotificationType.HOOK, kafkaConsumer, false, 100L);
    List<AtlasKafkaMessage<HookNotification>> messageList = consumer.receive();
    assertTrue(messageList.size() > 0);
    HookNotification consumedMessage = messageList.get(0).getMessage();
    assertMessagesEqual(message, consumedMessage, entity);
}

Example 69 with Referenceable

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

the class KafkaConsumerTest method assertMessagesEqual.

private void assertMessagesEqual(EntityUpdateRequest message, HookNotification consumedMessage, Referenceable entity) {
    assertEquals(consumedMessage.getType(), message.getType());
    assertEquals(consumedMessage.getUser(), message.getUser());
    assertTrue(consumedMessage instanceof EntityUpdateRequest);
    EntityUpdateRequest deserializedEntityUpdateRequest = (EntityUpdateRequest) consumedMessage;
    Referenceable deserializedEntity = deserializedEntityUpdateRequest.getEntities().get(0);
    assertEquals(deserializedEntity.getId(), entity.getId());
    assertEquals(deserializedEntity.getTypeName(), entity.getTypeName());
    assertEquals(deserializedEntity.getTraits(), entity.getTraits());
    assertEquals(deserializedEntity.getTrait(TRAIT_NAME), entity.getTrait(TRAIT_NAME));
}

Example 70 with Referenceable

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

the class KafkaNotificationTest method testReceiveKafkaMessages.

@Test
public void testReceiveKafkaMessages() throws Exception {
    kafkaNotification.send(NotificationInterface.NotificationType.HOOK, new EntityCreateRequest("u1", new Referenceable("type")));
    kafkaNotification.send(NotificationInterface.NotificationType.HOOK, new EntityCreateRequest("u2", new Referenceable("type")));
    kafkaNotification.send(NotificationInterface.NotificationType.HOOK, new EntityCreateRequest("u3", new Referenceable("type")));
    kafkaNotification.send(NotificationInterface.NotificationType.HOOK, new EntityCreateRequest("u4", new Referenceable("type")));
    NotificationConsumer<Object> consumer = kafkaNotification.createConsumers(NotificationInterface.NotificationType.HOOK, 1).get(0);
    List<AtlasKafkaMessage<Object>> messages = null;
    // fetch starting time
    long startTime = System.currentTimeMillis();
    while ((System.currentTimeMillis() - startTime) < 10000) {
        messages = consumer.receive();
        if (messages.size() > 0) {
            break;
        }
    }
    int i = 1;
    for (AtlasKafkaMessage<Object> msg : messages) {
        HookNotification message = (HookNotification) msg.getMessage();
        assertEquals(message.getUser(), "u" + i++);
    }
    consumer.close();
}