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

Example 36 with Referenceable

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

the class QuickStartIT method testProcessIsAdded.

@Test
public void testProcessIsAdded() throws AtlasServiceException {
    Referenceable loadProcess = atlasClientV1.getEntity(QuickStart.LOAD_PROCESS_TYPE, AtlasClient.REFERENCEABLE_ATTRIBUTE_NAME, QuickStart.LOAD_SALES_DAILY_PROCESS);
    assertEquals(QuickStart.LOAD_SALES_DAILY_PROCESS, loadProcess.get(AtlasClient.NAME));
    assertEquals(QuickStart.LOAD_SALES_DAILY_PROCESS_DESCRIPTION, loadProcess.get("description"));
    List<Id> inputs = (List<Id>) loadProcess.get(QuickStart.INPUTS_ATTRIBUTE);
    List<Id> outputs = (List<Id>) loadProcess.get(QuickStart.OUTPUTS_ATTRIBUTE);
    assertEquals(2, inputs.size());
    String salesFactTableId = getTableId(QuickStart.SALES_FACT_TABLE);
    String timeDimTableId = getTableId(QuickStart.TIME_DIM_TABLE);
    String salesFactDailyMVId = getTableId(QuickStart.SALES_FACT_DAILY_MV_TABLE);
    assertEquals(salesFactTableId, inputs.get(0)._getId());
    assertEquals(timeDimTableId, inputs.get(1)._getId());
    assertEquals(salesFactDailyMVId, outputs.get(0)._getId());
}

Example 37 with Referenceable

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

the class QuickStart method database.

Id database(String name, String description, String owner, String locationUri, String... traitNames) throws AtlasBaseException {
    try {
        Referenceable referenceable = new Referenceable(DATABASE_TYPE, traitNames);
        referenceable.set("name", name);
        referenceable.set("description", description);
        referenceable.set("owner", owner);
        referenceable.set("locationUri", locationUri);
        referenceable.set("createTime", System.currentTimeMillis());
        return createInstance(referenceable);
    } catch (Exception e) {
        throw new AtlasBaseException(AtlasErrorCode.QUICK_START, e, String.format("%s database entity creation failed", name));
    }
}

Example 38 with Referenceable

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

the class QuickStart method rawStorageDescriptor.

Referenceable rawStorageDescriptor(String location, String inputFormat, String outputFormat, boolean compressed) {
    Referenceable referenceable = new Referenceable(STORAGE_DESC_TYPE);
    referenceable.set("location", location);
    referenceable.set("inputFormat", inputFormat);
    referenceable.set("outputFormat", outputFormat);
    referenceable.set("compressed", compressed);
    return referenceable;
}

Example 39 with Referenceable

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

the class ColumnLineageUtils method populateColumnReferenceableMap.

static void populateColumnReferenceableMap(Map<String, Referenceable> m, Referenceable r) {
    if (r.getTypeName().equals(HiveDataTypes.HIVE_TABLE.getName())) {
        String qName = (String) r.get(ATTRIBUTE_QUALIFIED_NAME);
        String[] qNameComps = extractComponents(qName);
        for (Referenceable col : (List<Referenceable>) r.get(ATTRIBUTE_COLUMNS)) {
            String cName = (String) col.get(ATTRIBUTE_QUALIFIED_NAME);
            String[] colQNameComps = extractComponents(cName);
            String colQName = colQNameComps[0] + "." + colQNameComps[1] + "." + colQNameComps[2];
            m.put(colQName, col);
        }
        String tableQName = qNameComps[0] + "." + qNameComps[1];
        m.put(tableQName, r);
    }
}

Example 40 with Referenceable

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

the class SqoopHookIT method assertEntityIsRegistered.

protected String assertEntityIsRegistered(final String typeName, final String property, final String value, final AssertPredicate assertPredicate) throws Exception {
    waitFor(80000, new Predicate() {

        @Override
        public void evaluate() throws Exception {
            Referenceable entity = atlasClient.getEntity(typeName, property, value);
            assertNotNull(entity);
            if (assertPredicate != null) {
                assertPredicate.assertOnEntity(entity);
            }
        }
    });
    Referenceable entity = atlasClient.getEntity(typeName, property, value);
    return entity.getId()._getId();
}