add BeanPostProcessor

Author: code4craft

src/main/java/us/codecraft/tinyioc/beans/BeanPostProcessor.java

@@ -0,0 +1,9 @@
+package us.codecraft.tinyioc.beans;
+
+public interface BeanPostProcessor {
+
+	Object postProcessBeforeInitialization(Object bean, String beanName) throws Exception;
+
+	Object postProcessAfterInitialization(Object bean, String beanName) throws Exception;
+
+}

src/main/java/us/codecraft/tinyioc/beans/factory/AbstractBeanFactory.java

@@ -1,6 +1,7 @@
 package us.codecraft.tinyioc.beans.factory;
 
 import us.codecraft.tinyioc.beans.BeanDefinition;
+import us.codecraft.tinyioc.beans.BeanPostProcessor;
 
 import java.util.ArrayList;
 import java.util.Iterator;
@@ -17,6 +18,8 @@ public abstract class AbstractBeanFactory implements BeanFactory {
 
 	private final List<String> beanDefinitionNames = new ArrayList<String>();
 
+	private List<BeanPostProcessor> beanPostProcessors = new ArrayList<BeanPostProcessor>();
+
 	@Override
 	public Object getBean(String name) throws Exception {
 		BeanDefinition beanDefinition = beanDefinitionMap.get(name);
@@ -26,10 +29,26 @@ public Object getBean(String name) throws Exception {
 		Object bean = beanDefinition.getBean();
 		if (bean == null) {
 			bean = doCreateBean(beanDefinition);
+			initializeBean(bean, name);
 		}
 		return bean;
 	}
 
+	protected void initializeBean(Object bean, String name) throws Exception {
+		for (BeanPostProcessor beanPostProcessor : beanPostProcessors) {
+			bean = beanPostProcessor.postProcessBeforeInitialization(bean, name);
+		}
+
+		// TODO:call initialize method
+		for (BeanPostProcessor beanPostProcessor : beanPostProcessors) {
+            bean = beanPostProcessor.postProcessAfterInitialization(bean, name);
+		}
+	}
+
+	protected Object createBeanInstance(BeanDefinition beanDefinition) throws Exception {
+		return beanDefinition.getBeanClass().newInstance();
+	}
+
 	public void registerBeanDefinition(String name, BeanDefinition beanDefinition) throws Exception {
 		beanDefinitionMap.put(name, beanDefinition);
 		beanDefinitionNames.add(name);
@@ -42,12 +61,29 @@ public void preInstantiateSingletons() throws Exception {
 		}
 	}
 
-	/**
-	 * 初始化bean
-	 * 
-	 * @param beanDefinition
-	 * @return
-	 */
-	protected abstract Object doCreateBean(BeanDefinition beanDefinition) throws Exception;
+	protected Object doCreateBean(BeanDefinition beanDefinition) throws Exception {
+		Object bean = createBeanInstance(beanDefinition);
+		beanDefinition.setBean(bean);
+		applyPropertyValues(bean, beanDefinition);
+		return bean;
+	}
+
+	protected void applyPropertyValues(Object bean, BeanDefinition beanDefinition) throws Exception {
+
+	}
+
+	public void addBeanPostProcessor(BeanPostProcessor beanPostProcessor) throws Exception {
+		this.beanPostProcessors.add(beanPostProcessor);
+	}
+
+	public List getBeansForType(Class type) throws Exception {
+		List beans = new ArrayList<Object>();
+		for (String beanDefinitionName : beanDefinitionNames) {
+			if (type.isAssignableFrom(beanDefinitionMap.get(beanDefinitionName).getBeanClass())) {
+				beans.add(getBean(beanDefinitionName));
+			}
+		}
+		return beans;
+	}
 
 }

src/main/java/us/codecraft/tinyioc/beans/factory/AutowireCapableBeanFactory.java

@@ -1,29 +1,18 @@
 package us.codecraft.tinyioc.beans.factory;
 
-import java.lang.reflect.Field;
 import us.codecraft.tinyioc.BeanReference;
 import us.codecraft.tinyioc.beans.BeanDefinition;
 import us.codecraft.tinyioc.beans.PropertyValue;
 
+import java.lang.reflect.Field;
+
 /**
  * 可自动装配内容的BeanFactory
  * 
  * @author [email protected]
  */
 public class AutowireCapableBeanFactory extends AbstractBeanFactory {
 
-	@Override
-	protected Object doCreateBean(BeanDefinition beanDefinition) throws Exception {
-		Object bean = createBeanInstance(beanDefinition);
-        beanDefinition.setBean(bean);
-		applyPropertyValues(bean, beanDefinition);
-		return bean;
-	}
-
-	protected Object createBeanInstance(BeanDefinition beanDefinition) throws Exception {
-		return beanDefinition.getBeanClass().newInstance();
-	}
-
 	protected void applyPropertyValues(Object bean, BeanDefinition mbd) throws Exception {
 		for (PropertyValue propertyValue : mbd.getPropertyValues().getPropertyValues()) {
 			Field declaredField = bean.getClass().getDeclaredField(propertyValue.getName());

src/main/java/us/codecraft/tinyioc/context/AbstractApplicationContext.java

@@ -1,22 +1,41 @@
 package us.codecraft.tinyioc.context;
 
+import us.codecraft.tinyioc.beans.BeanPostProcessor;
 import us.codecraft.tinyioc.beans.factory.AbstractBeanFactory;
 
+import java.util.List;
+
 /**
  * @author [email protected]
  */
 public abstract class AbstractApplicationContext implements ApplicationContext {
-    protected AbstractBeanFactory beanFactory;
+	protected AbstractBeanFactory beanFactory;
 
-    public AbstractApplicationContext(AbstractBeanFactory beanFactory) {
-        this.beanFactory = beanFactory;
-    }
+	public AbstractApplicationContext(AbstractBeanFactory beanFactory) {
+		this.beanFactory = beanFactory;
+	}
 
-    public void refresh() throws Exception{
-    }
+	public void refresh() throws Exception {
+		loadBeanDefinitions(beanFactory);
+		registerBeanPostProcessors(beanFactory);
+		onRefresh();
+	}
 
-    @Override
-    public Object getBean(String name) throws Exception {
-        return beanFactory.getBean(name);
+	protected abstract void loadBeanDefinitions(AbstractBeanFactory beanFactory) throws Exception;
+
+	protected void registerBeanPostProcessors(AbstractBeanFactory beanFactory) throws Exception {
+		List beanPostProcessors = beanFactory.getBeansForType(BeanPostProcessor.class);
+		for (Object beanPostProcessor : beanPostProcessors) {
+			beanFactory.addBeanPostProcessor((BeanPostProcessor) beanPostProcessor);
+		}
+	}
+
+	protected void onRefresh() throws Exception{
+        beanFactory.preInstantiateSingletons();
     }
+
+	@Override
+	public Object getBean(String name) throws Exception {
+		return beanFactory.getBean(name);
+	}
 }

src/main/java/us/codecraft/tinyioc/context/ClassPathXmlApplicationContext.java

@@ -26,7 +26,7 @@ public ClassPathXmlApplicationContext(String configLocation, AbstractBeanFactory
 	}
 
 	@Override
-	public void refresh() throws Exception {
+	protected void loadBeanDefinitions(AbstractBeanFactory beanFactory) throws Exception {
 		XmlBeanDefinitionReader xmlBeanDefinitionReader = new XmlBeanDefinitionReader(new ResourceLoader());
 		xmlBeanDefinitionReader.loadBeanDefinitions(configLocation);
 		for (Map.Entry<String, BeanDefinition> beanDefinitionEntry : xmlBeanDefinitionReader.getRegistry().entrySet()) {

src/test/java/us/codecraft/tinyioc/BeanInitializeLogger.java

@@ -0,0 +1,20 @@
+package us.codecraft.tinyioc;
+
+import us.codecraft.tinyioc.beans.BeanPostProcessor;
+
+/**
+ * @author [email protected]
+ */
+public class BeanInitializeLogger implements BeanPostProcessor {
+	@Override
+	public Object postProcessBeforeInitialization(Object bean, String beanName) throws Exception {
+		System.out.println("Initialize bean " + beanName + " start!");
+		return bean;
+	}
+
+	@Override
+	public Object postProcessAfterInitialization(Object bean, String beanName) throws Exception {
+		System.out.println("Initialize bean " + beanName + " end!");
+		return bean;
+	}
+}

src/test/resources/tinyioc.xml

@@ -17,4 +17,7 @@
         <property name="outputService" ref="outputService"></property>
     </bean>
 
+    <bean name="beanInitializeLogger" class="us.codecraft.tinyioc.BeanInitializeLogger">
+    </bean>
+
 </beans>