Search in sources :

Example 1 with NiFiAtlasHook

use of org.apache.nifi.atlas.NiFiAtlasHook in project nifi by apache.

the class ReportLineageToAtlas method initLineageStrategy.

private void initLineageStrategy(ConfigurationContext context) throws IOException {
    nifiAtlasHook = new NiFiAtlasHook();
    final String strategy = context.getProperty(NIFI_LINEAGE_STRATEGY).getValue();
    if (LINEAGE_STRATEGY_SIMPLE_PATH.equals(strategy)) {
        lineageStrategy = new SimpleFlowPathLineage();
    } else if (LINEAGE_STRATEGY_COMPLETE_PATH.equals(strategy)) {
        lineageStrategy = new CompleteFlowPathLineage();
    }
    lineageStrategy.setLineageContext(nifiAtlasHook);
    initProvenanceConsumer(context);
}
Also used : CompleteFlowPathLineage(org.apache.nifi.atlas.provenance.lineage.CompleteFlowPathLineage) NiFiAtlasHook(org.apache.nifi.atlas.NiFiAtlasHook) SimpleFlowPathLineage(org.apache.nifi.atlas.provenance.lineage.SimpleFlowPathLineage)

Aggregations

NiFiAtlasHook (org.apache.nifi.atlas.NiFiAtlasHook)1 CompleteFlowPathLineage (org.apache.nifi.atlas.provenance.lineage.CompleteFlowPathLineage)1 SimpleFlowPathLineage (org.apache.nifi.atlas.provenance.lineage.SimpleFlowPathLineage)1