private void mapClassifications(AtlasVertex entityVertex, AtlasEntity entity) throws AtlasBaseException { List<AtlasEdge> edges = getAllClassificationEdges(entityVertex); if (CollectionUtils.isNotEmpty(edges)) { List<AtlasClassification> allClassifications = new ArrayList<>(); for (AtlasEdge edge : edges) { AtlasVertex classificationVertex = edge.getInVertex(); allClassifications.add(toAtlasClassification(classificationVertex)); } entity.setClassifications(allClassifications); } }
private void mapClassifications(AtlasVertex entityVertex, AtlasEntity entity) throws AtlasBaseException { List<AtlasEdge> edges = getAllClassificationEdges(entityVertex); if (CollectionUtils.isNotEmpty(edges)) { List<AtlasClassification> allClassifications = new ArrayList<>(); for (AtlasEdge edge : edges) { AtlasVertex classificationVertex = edge.getInVertex(); allClassifications.add(toAtlasClassification(classificationVertex)); } entity.setClassifications(allClassifications); } }
@Override public void apply(AtlasTransformableEntity transformableEntity) { AtlasEntity entity = transformableEntity.getEntity(); if (entity.getClassifications() == null) { entity.setClassifications(new ArrayList<>()); } boolean hasClassification = false; for (AtlasClassification c : entity.getClassifications()) { hasClassification = c.getTypeName().equals(classificationName); if (hasClassification) { break; } } if (!hasClassification) { entity.getClassifications().add(new AtlasClassification(classificationName)); } }
protected AtlasEntity column(String name, String dataType, String comment, String... traitNames) { AtlasEntity column = new AtlasEntity(COLUMN_TYPE); column.setAttribute("name", name); column.setAttribute(AtlasClient.REFERENCEABLE_ATTRIBUTE_NAME, "qualified:" + name); column.setAttribute("type", dataType); column.setAttribute("comment", comment); column.setClassifications(Stream.of(traitNames).map(AtlasClassification::new).collect(Collectors.toList())); return column; }
AtlasEntity createDatabase(String name, String description, String owner, String locationUri, String... traitNames) throws Exception { AtlasEntity entity = new AtlasEntity(DATABASE_TYPE); entity.setClassifications(toAtlasClassifications(traitNames)); entity.setAttribute("name", name); entity.setAttribute("description", description); entity.setAttribute("owner", owner); entity.setAttribute("locationuri", locationUri); entity.setAttribute("createTime", System.currentTimeMillis()); return createInstance(entity, traitNames); }
protected AtlasEntity column(String name, String dataType, String comment, String... traitNames) { AtlasEntity column = new AtlasEntity(COLUMN_TYPE); column.setAttribute("name", name); column.setAttribute(AtlasClient.REFERENCEABLE_ATTRIBUTE_NAME, "qualified:" + name); column.setAttribute("type", dataType); column.setAttribute("comment", comment); column.setClassifications(Stream.of(traitNames).map(AtlasClassification::new).collect(Collectors.toList())); return column; }
AtlasEntity createColumn(String name, String dataType, String comment, String... traitNames) throws Exception { AtlasEntity entity = new AtlasEntity(COLUMN_TYPE); entity.setClassifications(toAtlasClassifications(traitNames)); entity.setAttribute("name", name); entity.setAttribute("dataType", dataType); entity.setAttribute("comment", comment); return createInstance(entity, traitNames); }
AtlasEntity database(String name, String description, String owner, String locationUri, String... traitNames) { AtlasEntity database = new AtlasEntity(DATABASE_TYPE); database.setAttribute("name", name); database.setAttribute(AtlasClient.REFERENCEABLE_ATTRIBUTE_NAME, "qualified:" + name); database.setAttribute("description", description); database.setAttribute("owner", owner); database.setAttribute("locationUri", locationUri); database.setAttribute("createTime", System.currentTimeMillis()); database.setAttribute("clusterName", "cl1"); database.setClassifications(Stream.of(traitNames).map(AtlasClassification::new).collect(Collectors.toList())); return database; }
AtlasEntity database(String name, String description, String owner, String locationUri, String... traitNames) { AtlasEntity database = new AtlasEntity(DATABASE_TYPE); database.setAttribute("name", name); database.setAttribute(AtlasClient.REFERENCEABLE_ATTRIBUTE_NAME, "qualified:" + name); database.setAttribute("description", description); database.setAttribute("owner", owner); database.setAttribute("locationUri", locationUri); database.setAttribute("createTime", System.currentTimeMillis()); database.setAttribute("clusterName", "cl1"); database.setClassifications(Stream.of(traitNames).map(AtlasClassification::new).collect(Collectors.toList())); return database; }
private void init() { setGuid(nextInternalId()); setStatus(null); setCreatedBy(null); setUpdatedBy(null); setCreateTime(null); setUpdateTime(null); setClassifications(null); }
AtlasEntity createView(String name, AtlasEntity db, List<AtlasEntity> inputTables, String... traitNames) throws Exception { AtlasEntity entity = new AtlasEntity(VIEW_TYPE); entity.setClassifications(toAtlasClassifications(traitNames)); entity.setAttribute("name", name); entity.setAttribute(AtlasClient.REFERENCEABLE_ATTRIBUTE_NAME, name); entity.setAttribute("db", db); entity.setAttribute("inputTables", inputTables); return createInstance(entity, traitNames); }
AtlasEntity createTable(String name, String description, AtlasEntity db, AtlasEntity sd, String owner, String tableType, List<AtlasEntity> columns, String... traitNames) throws Exception { AtlasEntity entity = new AtlasEntity(TABLE_TYPE); entity.setClassifications(toAtlasClassifications(traitNames)); entity.setAttribute("name", name); entity.setAttribute(AtlasClient.REFERENCEABLE_ATTRIBUTE_NAME, name); entity.setAttribute("description", description); entity.setAttribute("owner", owner); entity.setAttribute("tableType", tableType); entity.setAttribute("createTime", System.currentTimeMillis()); entity.setAttribute("lastAccessTime", System.currentTimeMillis()); entity.setAttribute("retention", System.currentTimeMillis()); entity.setAttribute("db", AtlasTypeUtil.getAtlasObjectId(db)); entity.setAttribute("sd", AtlasTypeUtil.getAtlasObjectId(sd)); entity.setAttribute("columns", AtlasTypeUtil.toObjectIds(columns)); return createInstance(entity, traitNames); }
AtlasEntity createProcess(String name, String description, String user, List<AtlasEntity> inputs, List<AtlasEntity> outputs, String queryText, String queryPlan, String queryId, String queryGraph, String... traitNames) throws Exception { AtlasEntity entity = new AtlasEntity(LOAD_PROCESS_TYPE); entity.setClassifications(toAtlasClassifications(traitNames)); entity.setAttribute(AtlasClient.NAME, name); entity.setAttribute(AtlasClient.REFERENCEABLE_ATTRIBUTE_NAME, name); entity.setAttribute("description", description); entity.setAttribute("inputs", inputs); entity.setAttribute("outputs", outputs); entity.setAttribute("user", user); entity.setAttribute("startTime", System.currentTimeMillis()); entity.setAttribute("endTime", System.currentTimeMillis() + 10000); entity.setAttribute("queryText", queryText); entity.setAttribute("queryPlan", queryPlan); entity.setAttribute("queryId", queryId); entity.setAttribute("queryGraph", queryGraph); return createInstance(entity, traitNames); }
private void init() { setGuid(nextInternalId()); setHomeId(null); setIsProxy(Boolean.FALSE); setStatus(null); setCreatedBy(null); setUpdatedBy(null); setCreateTime(null); setUpdateTime(null); setClassifications(null); setMeanings(null); }
AtlasEntity view(String name, AtlasEntity dbId, List<AtlasEntity> inputTables, String... traitNames) { AtlasEntity view = new AtlasEntity(VIEW_TYPE); view.setAttribute("name", name); view.setAttribute(AtlasClient.REFERENCEABLE_ATTRIBUTE_NAME, name); view.setAttribute("userName", "testUser"); view.setAttribute("startTime", System.currentTimeMillis()); view.setAttribute("endTime", System.currentTimeMillis() + 10000); view.setAttribute("operationType", "view"); view.setAttribute("query", "create table as select"); view.setAttribute("queryText", "create table as select"); view.setAttribute("queryPlan", "viewPlan"); view.setAttribute("queryId", "view1"); view.setAttribute("db", getAtlasObjectId(dbId)); view.setAttribute("inputs", getAtlasObjectIds(inputTables)); view.setClassifications(Stream.of(traitNames).map(AtlasClassification::new).collect(Collectors.toList())); return view; }
AtlasEntity view(String name, AtlasEntity dbId, List<AtlasEntity> inputTables, String... traitNames) { AtlasEntity view = new AtlasEntity(VIEW_TYPE); view.setAttribute("name", name); view.setAttribute(AtlasClient.REFERENCEABLE_ATTRIBUTE_NAME, name); view.setAttribute("userName", "testUser"); view.setAttribute("startTime", System.currentTimeMillis()); view.setAttribute("endTime", System.currentTimeMillis() + 10000); view.setAttribute("operationType", "view"); view.setAttribute("query", "create table as select"); view.setAttribute("queryText", "create table as select"); view.setAttribute("queryPlan", "viewPlan"); view.setAttribute("queryId", "view1"); view.setAttribute("db", getAtlasObjectId(dbId)); view.setAttribute("inputs", getAtlasObjectIds(inputTables)); view.setClassifications(Stream.of(traitNames).map(AtlasClassification::new).collect(Collectors.toList())); return view; }
protected AtlasEntity loadProcess(String name, String description, String user, List<AtlasEntity> inputTables, List<AtlasEntity> outputTables, String queryText, String queryPlan, String queryId, String queryGraph, String... traitNames) { AtlasEntity process = new AtlasEntity(HIVE_PROCESS_TYPE); process.setAttribute("name", name); process.setAttribute(AtlasClient.REFERENCEABLE_ATTRIBUTE_NAME, name); process.setAttribute("description", description); process.setAttribute("userName", user); process.setAttribute("startTime", System.currentTimeMillis()); process.setAttribute("endTime", System.currentTimeMillis() + 10000); process.setAttribute("operationType", "load"); process.setAttribute("inputs", getAtlasObjectIds(inputTables)); process.setAttribute("outputs", getAtlasObjectIds(outputTables)); process.setAttribute("queryText", queryText); process.setAttribute("queryPlan", queryPlan); process.setAttribute("queryId", queryId); process.setAttribute("queryGraph", queryGraph); process.setClassifications(Stream.of(traitNames).map(AtlasClassification::new).collect(Collectors.toList())); return process; }
protected AtlasEntity loadProcess(String name, String description, String user, List<AtlasEntity> inputTables, List<AtlasEntity> outputTables, String queryText, String queryPlan, String queryId, String queryGraph, String... traitNames) { AtlasEntity process = new AtlasEntity(HIVE_PROCESS_TYPE); process.setAttribute("name", name); process.setAttribute(AtlasClient.REFERENCEABLE_ATTRIBUTE_NAME, name); process.setAttribute("description", description); process.setAttribute("userName", user); process.setAttribute("startTime", System.currentTimeMillis()); process.setAttribute("endTime", System.currentTimeMillis() + 10000); process.setAttribute("operationType", "load"); process.setAttribute("inputs", getAtlasObjectIds(inputTables)); process.setAttribute("outputs", getAtlasObjectIds(outputTables)); process.setAttribute("queryText", queryText); process.setAttribute("queryPlan", queryPlan); process.setAttribute("queryId", queryId); process.setAttribute("queryGraph", queryGraph); process.setClassifications(Stream.of(traitNames).map(AtlasClassification::new).collect(Collectors.toList())); return process; }
public AtlasEntity(AtlasEntity other) { super(other); if (other != null) { setGuid(other.getGuid()); setStatus(other.getStatus()); setCreatedBy(other.getCreatedBy()); setUpdatedBy(other.getUpdatedBy()); setCreateTime(other.getCreateTime()); setUpdateTime(other.getUpdateTime()); setVersion(other.getVersion()); setClassifications(other.getClassifications()); } }
public AtlasEntity(AtlasEntity other) { super(other); if (other != null) { setGuid(other.getGuid()); setHomeId(other.getHomeId()); setIsProxy(other.isProxy()); setStatus(other.getStatus()); setCreatedBy(other.getCreatedBy()); setUpdatedBy(other.getUpdatedBy()); setCreateTime(other.getCreateTime()); setUpdateTime(other.getUpdateTime()); setVersion(other.getVersion()); setClassifications(other.getClassifications()); } }