Search in sources :

Example 1 with LifecycleTrackableMessageSourceManagement

use of org.springframework.integration.support.management.LifecycleTrackableMessageSourceManagement in project spring-integration by spring-projects.

the class IntegrationMBeanExporter method enhanceSourceMonitor.

private MessageSourceMetrics enhanceSourceMonitor(MessageSourceMetrics monitor) {
    MessageSourceMetrics result = monitor;
    if (monitor.getManagedName() != null) {
        return monitor;
    }
    // Assignment algorithm and bean id, with bean id pulled reflectively out of enclosing endpoint if possible
    String[] names = this.applicationContext.getBeanNamesForType(AbstractEndpoint.class);
    String name = null;
    String endpointName = null;
    String source = "endpoint";
    Object endpoint = null;
    for (String beanName : names) {
        endpoint = this.applicationContext.getBean(beanName);
        Object field = null;
        if (monitor instanceof MessagingGatewaySupport && endpoint == monitor) {
            field = monitor;
        } else {
            try {
                field = extractTarget(getField(endpoint, "source"));
            } catch (Exception e) {
                logger.trace("Could not get source from bean = " + beanName);
            }
        }
        if (field == monitor) {
            name = beanName;
            endpointName = beanName;
            break;
        }
    }
    if (endpointName == null) {
        endpoint = null;
    }
    if (name != null && endpoint != null && name.startsWith("_org.springframework.integration")) {
        name = getInternalComponentName(name);
        source = "internal";
    }
    if (name != null && endpoint != null && name.startsWith("org.springframework.integration")) {
        Object target = endpoint;
        if (endpoint instanceof Advised) {
            TargetSource targetSource = ((Advised) endpoint).getTargetSource();
            if (targetSource != null) {
                try {
                    target = targetSource.getTarget();
                } catch (Exception e) {
                    logger.error("Could not get handler from bean = " + name);
                }
            }
        }
        Object outputChannel = null;
        if (target instanceof MessagingGatewaySupport) {
            outputChannel = ((MessagingGatewaySupport) target).getRequestChannel();
        } else {
            outputChannel = getField(target, "outputChannel");
        }
        if (outputChannel != null) {
            if (!this.anonymousSourceCounters.containsKey(outputChannel)) {
                this.anonymousSourceCounters.put(outputChannel, new AtomicLong());
            }
            AtomicLong count = this.anonymousSourceCounters.get(outputChannel);
            long total = count.incrementAndGet();
            String suffix = "";
            /*
				 * Short hack to makes sure object names are unique if more than one endpoint has the same input channel
				 */
            if (total > 1) {
                suffix = "#" + total;
            }
            name = outputChannel + suffix;
            source = "anonymous";
        }
    }
    if (endpoint instanceof Lifecycle) {
        // Wrap the monitor in a lifecycle so it exposes the start/stop operations
        if (endpoint instanceof TrackableComponent) {
            if (monitor instanceof MessageSourceManagement) {
                result = new LifecycleTrackableMessageSourceManagement((Lifecycle) endpoint, (MessageSourceManagement) monitor);
            } else {
                result = new LifecycleTrackableMessageSourceMetrics((Lifecycle) endpoint, monitor);
            }
        } else {
            if (monitor instanceof MessageSourceManagement) {
                result = new LifecycleMessageSourceManagement((Lifecycle) endpoint, (MessageSourceManagement) monitor);
            } else {
                result = new LifecycleMessageSourceMetrics((Lifecycle) endpoint, monitor);
            }
        }
    }
    if (name == null) {
        name = monitor.toString();
        source = "source";
    }
    if (endpointName != null) {
        this.beansByEndpointName.put(name, endpointName);
    }
    monitor.setManagedType(source);
    monitor.setManagedName(name);
    return result;
}
Also used : TargetSource(org.springframework.aop.TargetSource) MessageSourceMetrics(org.springframework.integration.support.management.MessageSourceMetrics) LifecycleTrackableMessageSourceMetrics(org.springframework.integration.support.management.LifecycleTrackableMessageSourceMetrics) LifecycleMessageSourceMetrics(org.springframework.integration.support.management.LifecycleMessageSourceMetrics) Lifecycle(org.springframework.context.Lifecycle) MessagingGatewaySupport(org.springframework.integration.gateway.MessagingGatewaySupport) TrackableComponent(org.springframework.integration.support.management.TrackableComponent) LifecycleMessageSourceMetrics(org.springframework.integration.support.management.LifecycleMessageSourceMetrics) UnableToRegisterMBeanException(org.springframework.jmx.export.UnableToRegisterMBeanException) JMException(javax.management.JMException) BeansException(org.springframework.beans.BeansException) AtomicLong(java.util.concurrent.atomic.AtomicLong) MessageSourceManagement(org.springframework.integration.support.management.MessageSourceManagement) LifecycleMessageSourceManagement(org.springframework.integration.support.management.LifecycleMessageSourceManagement) LifecycleTrackableMessageSourceManagement(org.springframework.integration.support.management.LifecycleTrackableMessageSourceManagement) LifecycleTrackableMessageSourceManagement(org.springframework.integration.support.management.LifecycleTrackableMessageSourceManagement) Advised(org.springframework.aop.framework.Advised) LifecycleTrackableMessageSourceMetrics(org.springframework.integration.support.management.LifecycleTrackableMessageSourceMetrics) LifecycleMessageSourceManagement(org.springframework.integration.support.management.LifecycleMessageSourceManagement)

Aggregations

AtomicLong (java.util.concurrent.atomic.AtomicLong)1 JMException (javax.management.JMException)1 TargetSource (org.springframework.aop.TargetSource)1 Advised (org.springframework.aop.framework.Advised)1 BeansException (org.springframework.beans.BeansException)1 Lifecycle (org.springframework.context.Lifecycle)1 MessagingGatewaySupport (org.springframework.integration.gateway.MessagingGatewaySupport)1 LifecycleMessageSourceManagement (org.springframework.integration.support.management.LifecycleMessageSourceManagement)1 LifecycleMessageSourceMetrics (org.springframework.integration.support.management.LifecycleMessageSourceMetrics)1 LifecycleTrackableMessageSourceManagement (org.springframework.integration.support.management.LifecycleTrackableMessageSourceManagement)1 LifecycleTrackableMessageSourceMetrics (org.springframework.integration.support.management.LifecycleTrackableMessageSourceMetrics)1 MessageSourceManagement (org.springframework.integration.support.management.MessageSourceManagement)1 MessageSourceMetrics (org.springframework.integration.support.management.MessageSourceMetrics)1 TrackableComponent (org.springframework.integration.support.management.TrackableComponent)1 UnableToRegisterMBeanException (org.springframework.jmx.export.UnableToRegisterMBeanException)1