一起写框架-Ioc内核容器的实现-基础功能-组件注解支持自定义的对象名(九)

实现功能

如果扫描组件注解（@Controller，@Service，@Repository，@Component）默认对象名，已经实现了默认使用首字母小写类名的作为对象名。

但，现实需求中。我们有时候希望可以自己定义对象的名。

实现思路

1.获得扫描组件注解的name属性的值。

2.将这个值作为对象名

实现步骤

1.在AbstractApplicationContext增加一个方法getComponentOfName，用于判断组件注解是否设置了name属性。如果设置了就获得该值

 

/**
     * 获得组件的对象名
     * 
     * @param classType
     * @return
     */
    private String getComponentOfName(Class<?> classType) {
        //获得四个组件注解的对象
        Component component = classType.getDeclaredAnnotation(Component.class);
        Service service = classType.getDeclaredAnnotation(Service.class);
        Controller controller = classType.getDeclaredAnnotation(Controller.class);
        Repository repository = classType.getDeclaredAnnotation(Repository.class);
        //判断注解对象是否为空，注解对象的的name属性是否有值
        if (component != null) {
            if (!"".equals(component.name()) && component.name() != null) {
                return component.name();
            }
        } 
        if (service != null) {
            if (!"".equals(service.name()) && service.name() != null) {
                return service.name();
            }
        } 
        if (controller != null) {
            if (!"".equals(controller.name()) && controller.name() != null) {
                return controller.name();
            }
        } 
        if (repository != null) {
            if (!"".equals(repository.name()) && repository.name() != null) {
                return repository.name();
            }
        }
        return null;

    }

 

2.修改AbstractApplicationContext类的构造方法。标红处

 

/**
     * 将容器操作加载创建对象的代码写抽象类里面，这样可以方便以后扩展多种实现。
     * 
     * @param classType
     */
    public AbstractApplicationContext(Class<?> classType) {
        // 判断配置类是否有Configuration注解
        Configuration annotation = classType.getDeclaredAnnotation(Configuration.class);
        if (annotation != null) {
            // 获得组件扫描注解
            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);
                try {
                    // 通过类全名创建对象
                    Class<?> objectClassType = Class.forName(className);
                    /*
                     * 判断如果类权限名对应的不是接口，并且包含有@Component|@Controller|@Service|
                     * @Repository 才可以创建对象
                     */
                    if (this.isComponent(objectClassType)) {
                        Object instance = objectClassType.newInstance();
                        // 修改为,默认对象支持首字符小写
35                         String objectName=null;
36                         //获得组件注解的name属性值
37                         String componentName = this.getComponentOfName(objectClassType);
38                     
39                         if(componentName==null){
40                             //如果组件注解的name属性没有值，使用默认命名对象
41                             objectName= NamingUtils.firstCharToLower(instance.getClass().getSimpleName());
42                         }else{
43                             //如果组件注解的name属性有值，使用自定义命名对象
44                             objectName=componentName;
45                         }
46                         this.getContext().addObject(objectName, instance);
                    }
                } catch (InstantiationException e) {
                    e.printStackTrace();
                } catch (IllegalAccessException e) {
                    e.printStackTrace();
                } catch (ClassNotFoundException e) {
                    e.printStackTrace();
                }
            }
        }
    }

 

测试代码

1.修改UserService类的组件注解

 

package ioc.core.test.service;

import ioc.core.annotation.stereotype.Service;

/**
 * 一个普通的类，用于测试是否可以创建对象
 * @author ranger
 *
 */
@Service(name="uService")
public class UserService {
    
    public void login(){
        System.out.println("-登录Service-");
    }

}

 

2.测试类

package ioc.core.test;

import org.junit.Test;

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

public class AnntationApplicationContextTest {
    
    @Test
    public void login(){
        try {
            AnntationApplicationContext context=new AnntationApplicationContext(Config.class);
            UserService userService = context.getBean("uService", UserService.class);
            userService.login();
            System.out.println(context.getContext().getObjects());
        
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

}

3.测试结果