一起写框架-Ioc内核容器的实现-对象的调用-@Bean注解注入容器的对象(十二)

实现功能

现实需求中，有一些类似无法加入扫描组件注解的。如jdk里面的类。

那么框架必须要支持将没有组件注解标识的类也可以有一个方式注入到容器里面。

那么，我们通过类似Spring的@Bean的方案，来实现这个需求。

通过在配置类的方法的上面，使用@Bean注解，将返回的对象加到容器中。

 

实现思路

获得有@Configuration注解标识的类。检索它的方法，如果有@Bean，执行这个方法并将返回的对象放在容器中。

实现步骤

1.定义一个Bean注解

package ioc.core.annotation;

import java.lang.annotation.Documented;
import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;

//表示用于运行时的注解
@Retention(RetentionPolicy.RUNTIME)
//表示只能在类或者接口的上面使用
@Target(value=ElementType.METHOD)
@Documented
public @interface Bean {
    
    /**
     * 用于设置对象名
     * @return
     */
    String name() default "";

}

2.修改AbstractApplicationContext代码

（1）创建一个bean的方法处理从bean注解的方法中获得对象

（2）将从组件注解的类扫描的方式创建对象放在scan方法内

 根据这个修改方法，首先将AbstractApplicationContext类构造方法原来使用@ComponentScan创建对象的代码移到scan()方法中，然后创建一个bean()方法编写通过@Bean获得对象

private void bean(Class<?> classType,Context context) throws InstantiationException, IllegalAccessException, IllegalArgumentException, InvocationTargetException{
        Map<String, Object> objects = context.getObjects();
        //1.获得配置类的对象
        Object configuration = classType.newInstance();
        //2.获得配置类的所有方法
        Method[] methods = classType.getMethods();
        for(Method method:methods){
            Bean bean = method.getDeclaredAnnotation(Bean.class);
            if(bean!=null){
                Object object = method.invoke(configuration);
                if(bean.name()!=null&&!"".equals(bean.name())){
                    objects.put(bean.name(), object);
                }else{
                    objects.put(NamingUtils.firstCharToLower(object.getClass().getSimpleName()), object);
                }
            }
        }
        
    }
    
    private void scan(Class<?> classType,Context context) throws ClassNotFoundException, InstantiationException, IllegalAccessException{

            // 获得组件扫描注解
            ComponentScan componentScan = classType.getDeclaredAnnotation(ComponentScan.class);
            // 获得包名
            this.basePackage = componentScan.basePackages();
            // 根据包名获得类全限制名
            // Set<String> classNames =
            // PackageUtils.getClassName(this.basePackage[0], true);
            // 将扫描一个包，修改为多个包
            Set<String> classNames = PackageUtils.getClassNames(this.basePackage, true);
            // 通过类名创建对象
            Iterator<String> iteratorClassName = classNames.iterator();
            while (iteratorClassName.hasNext()) {

                String className = iteratorClassName.next();
                // System.out.println(className);

                // 通过类全名创建对象
                Class<?> objectClassType = Class.forName(className);
                /*
                 * 判断如果类权限名对应的不是接口，并且包含有@Component|@Controller|@Service|
                 * 
                 * @Repository 才可以创建对象
                 */
                if (this.isComponent(objectClassType)) {
                    Object instance = objectClassType.newInstance();
                    // 修改为,默认对象支持首字符小写
                    String objectName = null;
                    // 获得组件注解的name属性值
                    String componentName = this.getComponentOfName(objectClassType);

                    if (componentName == null) {
                        // 如果组件注解的name属性没有值，使用默认命名对象
                        objectName = NamingUtils.firstCharToLower(instance.getClass().getSimpleName());
                    } else {
                        // 如果组件注解的name属性有值，使用自定义命名对象
                        objectName = componentName;
                    }
                    System.out.println("1."+instance);
                    //将对象加入到容器
                    context.addObject(objectName, instance);
                }

            }
        }
    

 2.修改AbstractApplicationContext的构造方法的代码。如下：

    /**
     * 将容器操作加载创建对象的代码写抽象类里面，这样可以方便以后扩展多种实现。
     * 
     * @param classType
     */
    public AbstractApplicationContext(Class<?> classType) {
        try {
            // 判断配置类是否有Configuration注解
            Configuration annotation = classType.getDeclaredAnnotation(Configuration.class);
            if(annotation!=null){
                bean(classType, this.getContext());
                scan(classType, this.getContext());
    
            // 注入对象到有@Autowired注解属性和方法中。
            autowired();
        } catch (InstantiationException e) {
            e.printStackTrace();
        } catch (IllegalAccessException e) {
            e.printStackTrace();
        } catch (ClassNotFoundException e) {
            e.printStackTrace();
        } catch (IllegalArgumentException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        } catch (InvocationTargetException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        } catch (SecurityException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }

    }

 测试代码

1.创建一个没有注解的普通Java类

package ioc.core.test;

public class TestUtils {
    
    public void test(){
        System.out.println("--测试没有组件注解的类-");
    }

}

2.在配置类使用@Bean注入这个类

package ioc.core.test.config;

import ioc.core.annotation.Bean;
import ioc.core.annotation.ComponentScan;
import ioc.core.annotation.Configuration;
import ioc.core.test.TestUtils;

//使用定义@Configuration定义该类是一个配置类
@Configuration
//使用ComponentScan设置扫描包的路径
@ComponentScan(basePackages={"ioc.core"})
public class Config {
    
    @Bean
    public Object getTestUtils(){
        TestUtils tu=new TestUtils();
        return tu;
    }

}

3.测试输出TestUtils的test() 方法

package ioc.core.test;

import org.junit.Test;

import ioc.core.impl.AnntationApplicationContext;
import ioc.core.test.config.Config;

public class ConfigruationTest {
    
    @Test
    public void bean(){
        try {
            AnntationApplicationContext context=new AnntationApplicationContext(Config.class);
             TestUtils testUtils = context.getBean("testUtils", TestUtils.class);
             System.out.println(context.getContext().getObjects().keySet()+"-----");
             testUtils.test();
        } catch (Exception e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }
    }

}

4.测试结果，输出了这个方法的内容，成功！

总结

至此，一个满足基本开发需要的Ioc的框架就写完了。后续会根据需求，再不断更新众多功能。