Examples with OnEnabled org.apache.nifi.annotation.lifecycle.OnEnabled used on opensource projects

Example 21 with OnEnabled

use of org.apache.nifi.annotation.lifecycle.OnEnabled in project nifi by apache.

the class LivySessionController method onConfigured.

@OnEnabled
public void onConfigured(final ConfigurationContext context) {
    ComponentLog log = getLogger();
    final String livyHost = context.getProperty(LIVY_HOST).evaluateAttributeExpressions().getValue();
    final String livyPort = context.getProperty(LIVY_PORT).evaluateAttributeExpressions().getValue();
    final String sessionPoolSize = context.getProperty(SESSION_POOL_SIZE).evaluateAttributeExpressions().getValue();
    final String sessionKind = context.getProperty(SESSION_TYPE).getValue();
    final long sessionManagerStatusInterval = context.getProperty(SESSION_MGR_STATUS_INTERVAL).asTimePeriod(TimeUnit.MILLISECONDS);
    final String jars = context.getProperty(JARS).evaluateAttributeExpressions().getValue();
    final String files = context.getProperty(FILES).evaluateAttributeExpressions().getValue();
    sslContextService = context.getProperty(SSL_CONTEXT_SERVICE).asControllerService(SSLContextService.class);
    sslContext = sslContextService == null ? null : sslContextService.createSSLContext(SSLContextService.ClientAuth.NONE);
    connectTimeout = Math.toIntExact(context.getProperty(CONNECT_TIMEOUT).asTimePeriod(TimeUnit.MILLISECONDS));
    this.livyUrl = "http" + (sslContextService != null ? "s" : "") + "://" + livyHost + ":" + livyPort;
    this.controllerKind = sessionKind;
    this.jars = jars;
    this.files = files;
    this.sessionPoolSize = Integer.valueOf(sessionPoolSize);
    this.enabled = true;
    livySessionManagerThread = new Thread(() -> {
        while (enabled) {
            try {
                manageSessions();
                Thread.sleep(sessionManagerStatusInterval);
            } catch (InterruptedException e) {
                Thread.currentThread().interrupt();
                enabled = false;
            } catch (IOException ioe) {
                throw new ProcessException(ioe);
            }
        }
    });
    livySessionManagerThread.setName("Livy-Session-Manager-" + controllerKind);
    livySessionManagerThread.start();
}

Example 22 with OnEnabled

use of org.apache.nifi.annotation.lifecycle.OnEnabled in project nifi by apache.

the class ScriptedLookupService method onEnabled.

@Override
@OnEnabled
public void onEnabled(final ConfigurationContext context) {
    synchronized (scriptingComponentHelper.isInitialized) {
        if (!scriptingComponentHelper.isInitialized.get()) {
            scriptingComponentHelper.createResources();
        }
    }
    super.onEnabled(context);
    // Call an non-interface method onEnabled(context), to allow a scripted LookupService the chance to set up as necessary
    final Invocable invocable = (Invocable) scriptEngine;
    if (configurationContext != null) {
        try {
            // Get the actual object from the script engine, versus the proxy stored in lookupService. The object may have additional methods,
            // where lookupService is a proxied interface
            final Object obj = scriptEngine.get("lookupService");
            if (obj != null) {
                try {
                    invocable.invokeMethod(obj, "onEnabled", context);
                } catch (final NoSuchMethodException nsme) {
                    if (getLogger().isDebugEnabled()) {
                        getLogger().debug("Configured script LookupService does not contain an onEnabled() method.");
                    }
                }
            } else {
                throw new ScriptException("No LookupService was defined by the script.");
            }
        } catch (ScriptException se) {
            throw new ProcessException("Error executing onEnabled(context) method", se);
        }
    }
}

Example 23 with OnEnabled

use of org.apache.nifi.annotation.lifecycle.OnEnabled in project nifi by apache.

the class GrokReader method preCompile.

@OnEnabled
public void preCompile(final ConfigurationContext context) throws GrokException, IOException {
    grok = new Grok();
    try (final InputStream in = getClass().getResourceAsStream(DEFAULT_PATTERN_NAME);
        final Reader reader = new InputStreamReader(in)) {
        grok.addPatternFromReader(reader);
    }
    if (context.getProperty(PATTERN_FILE).isSet()) {
        grok.addPatternFromFile(context.getProperty(PATTERN_FILE).evaluateAttributeExpressions().getValue());
    }
    grok.compile(context.getProperty(GROK_EXPRESSION).getValue());
    appendUnmatchedLine = context.getProperty(NO_MATCH_BEHAVIOR).getValue().equalsIgnoreCase(APPEND_TO_PREVIOUS_MESSAGE.getValue());
    this.recordSchemaFromGrok = createRecordSchema(grok);
    final String schemaAccess = context.getProperty(getSchemaAcessStrategyDescriptor()).getValue();
    if (STRING_FIELDS_FROM_GROK_EXPRESSION.getValue().equals(schemaAccess)) {
        this.recordSchema = recordSchemaFromGrok;
    } else {
        this.recordSchema = null;
    }
}

Example 24 with OnEnabled

use of org.apache.nifi.annotation.lifecycle.OnEnabled in project nifi by apache.

the class JsonPathReader method compileJsonPaths.

@OnEnabled
public void compileJsonPaths(final ConfigurationContext context) {
    this.dateFormat = context.getProperty(DateTimeUtils.DATE_FORMAT).getValue();
    this.timeFormat = context.getProperty(DateTimeUtils.TIME_FORMAT).getValue();
    this.timestampFormat = context.getProperty(DateTimeUtils.TIMESTAMP_FORMAT).getValue();
    final LinkedHashMap<String, JsonPath> compiled = new LinkedHashMap<>();
    for (final PropertyDescriptor descriptor : context.getProperties().keySet()) {
        if (!descriptor.isDynamic()) {
            continue;
        }
        final String fieldName = descriptor.getName();
        final String expression = context.getProperty(descriptor).getValue();
        final JsonPath jsonPath = JsonPath.compile(expression);
        compiled.put(fieldName, jsonPath);
    }
    jsonPaths = compiled;
}