org.nutz.integration.spring.SpringIocProvider Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of nutz-integration-spring Show documentation
The newest version!
package org.nutz.integration.spring;

import java.lang.annotation.Annotation;

import org.nutz.ioc.Ioc;
import org.nutz.ioc.IocException;
import org.nutz.ioc.ObjectLoadException;
import org.nutz.ioc.annotation.InjectName;
import org.nutz.lang.Lang;
import org.nutz.lang.Strings;
import org.nutz.mvc.IocProvider;
import org.nutz.mvc.NutConfig;
import org.springframework.beans.factory.config.SingletonBeanRegistry;
import org.springframework.context.ApplicationContext;
import org.springframework.context.event.ContextClosedEvent;
import org.springframework.context.event.ContextRefreshedEvent;
import org.springframework.context.support.ClassPathXmlApplicationContext;
import org.springframework.web.context.WebApplicationContext;

/**
 * 简单实现Nutz.IoC-Spring桥
 * 
 * Need Spring 2.0 or later
 * 
 * @author wendal([email protected])
 * 
 */
public class SpringIocProvider implements IocProvider, Ioc {

	protected ApplicationContext applicationContext;

	@Override
	public Ioc create(NutConfig config, String[] args) {
		if (config == null || Lang.eleSize(args) > 0)
			applicationContext = new ClassPathXmlApplicationContext(args);
		else
			applicationContext = (ApplicationContext) config.getAttribute(WebApplicationContext.ROOT_WEB_APPLICATION_CONTEXT_ATTRIBUTE);
		return this;
	}

	@Override
	public void depose() {
		if (applicationContext != null) {
			applicationContext.publishEvent(new ContextClosedEvent(applicationContext));
			applicationContext = null;
		}
	}

	@Override
	public  T get(Class type, String name) {
		return applicationContext.getBean(name, type);
	}

	@Override
	public String[] getNames() {
		return applicationContext.getBeanDefinitionNames();
	}

	@Override
	public boolean has(String name) {
		return applicationContext.containsBean(name);
	}

	@Override
	public void reset() {
		applicationContext.publishEvent(new ContextRefreshedEvent(applicationContext));
	}

	@Override
	public  T get(Class classZ) throws IocException {
		InjectName injectName = classZ.getAnnotation(InjectName.class);
		if (injectName != null && !Strings.isBlank(injectName.value()))
			return (T) applicationContext.getBean(injectName.value());
		return (T) applicationContext.getBean(applicationContext.getBeanNamesForType(classZ)[0]);
	}

	/*
	 * (non-Javadoc)
	 * 
	 * @see org.nutz.ioc.Ioc#getNamesByType(java.lang.Class)
	 */
	@Override
	public String[] getNamesByType(Class klass) {

		return applicationContext.getBeanDefinitionNames();
	}

	/*
	 * (non-Javadoc)
	 * 
	 * @see org.nutz.ioc.Ioc#getByType(java.lang.Class)
	 */
	@Override
	public  K getByType(Class klass) {
		return applicationContext.getBean(klass);
	}
	
	public String[] getNamesByAnnotation(Class klass) {
	    return applicationContext.getBeanNamesForAnnotation(klass);
	}

    public Ioc addBean(String name, Object obj) {
        if (this instanceof SingletonBeanRegistry) {
            ((SingletonBeanRegistry)this).registerSingleton(name, obj);
        }
        return this;
    }

    public Class getType(String name) throws ObjectLoadException {
        return applicationContext.getType(name);
    }
}