com.blade.IocApplication Maven / Gradle / Ivy

Go to download
/**
 * Copyright (c) 2015, biezhi 王爵 ([email protected])
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * 	http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.blade;

import java.util.ArrayList;
import java.util.List;
import java.util.Set;

import blade.kit.log.Logger;
import blade.kit.resource.ClassPathClassReader;
import blade.kit.resource.ClassReader;

import com.blade.ioc.Container;
import com.blade.ioc.Scope;
import com.blade.plugin.Plugin;

/**
 * IOC container, used to initialize the IOC object
 *
 * @author	biezhi
 * @since	1.0
 */
public class IocApplication {

	private static final Logger LOGGER = Logger.getLogger(IocApplication.class);
	
	/**
	 * Ioc Container
	 */
	private Container container = null;
	
	/**
	 * Class to read object, load class
	 */
	private ClassReader classReader = null;
	
	/**
	 * Plugin List
	 */
	private List plugins = null;
	
	public IocApplication(Container container) {
		this.classReader = new ClassPathClassReader();
		this.plugins = new ArrayList();
		this.container = container;
	}
	
	/**
	 * IOC initialize
	 * @param iocs		ioc packages
	 * @param bootstrap	bootstrap object
	 */
	public void init(String[] iocs, Bootstrap bootstrap){
		
		// Initialize the global configuration class
		if(null == container.getBean(Bootstrap.class, Scope.SINGLE)){
			container.registerBean(bootstrap);
		}
		
		// The object to initialize the IOC container loads the IOC package to configure the class that conforms to the IOC
		if(null != iocs && iocs.length > 0){
			for(String packageName : iocs){
				registerBean(packageName);
			}
		}
		
		// Initialization injection
		container.initWired();
		
		Set names = container.getBeanNames();
		for(String name : names){
			LOGGER.info("Add Object：" + name + "=" + container.getBean(name, null));
		}
		
	}
	
	@SuppressWarnings("unchecked")
	public  T registerPlugin(Class plugin){
		Object object = container.registerBean(Aop.create(plugin));
		T t = (T) object;
		plugins.add(t);
		return t;
	}

	public  T getPlugin(Class plugin){
		if(null != plugin && null != container){
			return container.getBean(plugin, null);
		}
		return null;
	}
	
	/**
	 * Register all objects in a package
	 * 
	 * @param packageName package name
	 */
	private void registerBean(String packageName) {
		
		// Recursive scan
		boolean recursive = false; 
		if (packageName.endsWith(".*")) {
			packageName = packageName.substring(0, packageName.length() - 2);
			recursive = true;
		}
		
		// Scan package all class
		Set> classes = classReader.getClass(packageName, recursive);
		for (Class clazz : classes) {
			// 注册带有Component和Service注解的类
			if (container.isRegister(clazz.getAnnotations())) {
				container.registerBean(Aop.create(clazz));
			}
		}
	}
	
	public List getPlugins() {
		return plugins;
	}

	/**
	 * destroy
	 */
	public void destroy() {
		// clean ioc container
		container.removeAll();
		for(Plugin plugin : plugins){
			plugin.destroy();
		}
	}
}