Examples with NiFiAtlasHook org.apache.nifi.atlas.NiFiAtlasHook used on opensource projects

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);
}