com.agileapes.motorex.cli.target.impl.SpringExecutionTargetContext.java Source code

Introduction

Here is the source code for com.agileapes.motorex.cli.target.impl.SpringExecutionTargetContext.java

Source

/*
 * Copyright (c) 2012. AgileApes (http://www.agileapes.scom/), and
 * associated organization.
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy of this
 * software and associated documentation files (the "Software"), to deal in the Software
 * without restriction, including without limitation the rights to use, copy, modify,
 * merge, publish, distribute, sublicense, and/or sell copies of the Software, and to
 * permit persons to whom the Software is furnished to do so, subject to the following
 * conditions:
 *
 * The above copyright notice and this permission notice shall be included in all copies
 * or substantial portions of the Software.
 */

package com.agileapes.motorex.cli.target.impl;

import com.agileapes.motorex.cli.target.ExecutionTarget;
import com.agileapes.motorex.cli.target.ExecutionTargetContextAware;
import org.springframework.beans.BeansException;
import org.springframework.beans.factory.BeanCreationException;
import org.springframework.beans.factory.config.BeanDefinition;
import org.springframework.beans.factory.config.BeanFactoryPostProcessor;
import org.springframework.beans.factory.config.BeanPostProcessor;
import org.springframework.beans.factory.config.ConfigurableListableBeanFactory;
import org.springframework.util.ClassUtils;

/**
 * This implementation of the context and registry interfaces is an extension to
 * {@link DefaultExecutionTargetContext} and adds Spring support and automatic target
 * bean discovery to the framework.
 *
 * @author Mohammad Milad Naseri (m.m.naseri@gmail.com)
 * @since 1.0 (2012/12/8, 17:02)
 */
public class SpringExecutionTargetContext extends DefaultExecutionTargetContext
        implements BeanFactoryPostProcessor, BeanPostProcessor {

    @Override
    public void postProcessBeanFactory(ConfigurableListableBeanFactory beanFactory) throws BeansException {
        final String[] names = beanFactory.getBeanDefinitionNames();
        for (String name : names) {
            final BeanDefinition definition = beanFactory.getBeanDefinition(name);
            if (definition.isSingleton()) {
                final String className = definition.getBeanClassName();
                try {
                    final Class<?> beanClass = ClassUtils.forName(className, beanFactory.getBeanClassLoader());
                    if (ExecutionTarget.class.isAssignableFrom(beanClass)) {
                        final ExecutionTarget target = beanFactory.getBean(name, ExecutionTarget.class);
                        if (target instanceof AbstractIdentifiedExecutionTarget) {
                            AbstractIdentifiedExecutionTarget executionTarget = (AbstractIdentifiedExecutionTarget) target;
                            executionTarget.setIdentifier(name);
                        }
                        register(target);
                    }
                } catch (ClassNotFoundException e) {
                    throw new BeanCreationException(name, "Could not access bean class");
                }
            }
        }
    }

    @Override
    public Object postProcessBeforeInitialization(Object bean, String beanName) throws BeansException {
        return bean;
    }

    @Override
    public Object postProcessAfterInitialization(Object bean, String beanName) throws BeansException {
        if (bean instanceof ExecutionTargetContextAware) {
            ((ExecutionTargetContextAware) bean).setExecutionTargetContext(this);
        }
        return bean;
    }
}