项目启动时获取所有指定注解的类

import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;
import org.springframework.stereotype.Component;

@Component
@Target({ElementType.TYPE})
@Retention(RetentionPolicy.RUNTIME)
public @interface JobConfig{
    String[] name();
}
import org.springframework.beans.BeansException;
import org.springframework.context.ApplicationContext;
import org.springframework.core.env.Environment;

import java.util.Iterator;
import java.util.Map;

@Component
@Configuration
public class ScanTest  implements ApplicationContextAware{

    private Environment environment;

	// 项目启动时会执行此方法 实现了ApplicationContextAware接口
    public void setApplicationContext(ApplicationContext ctx) throws BeansException {
        this.environment = ctx.getEnvironment();
        Map<String, Object> beanMap = ctx.getBeansWithAnnotation(JobConfig.class);
        Iterator iterator = beanMap.values().iterator();

        while (iterator.hasNext()) {
            Object confBean = iterator.next();
            Class<?> clz = confBean.getClass();
            JobConfig conf = this.findJobConfigAnnot(clz);
            String jobClass = clz.getName();
            if (jobClass.contains("$$EnhancerBySpringCGLIB")) {
                jobClass = jobClass.substring(0, jobClass.indexOf("$$EnhancerBySpringCGLIB"));
            }

            String[] jobNames = conf.name();
            if (jobNames == null) {
                throw new IllegalArgumentException("ElasticJobConf job name error. jobClass:" + jobClass);
            }
            System.out.println("扫描到:"+conf.name());
        }
    }

    public JobConfig findJobConfigAnnot(Class<?> clz) {
        JobConfig JobConfig = clz.getAnnotation(JobConfig.class);
        // 判断代理类
        if (JobConfig == null && clz.getSimpleName().contains("SpringCGLIB")) {
            Class tmpClass = clz.getSuperclass();
            for (int i = 0; i < 3; tmpClass = tmpClass.getSuperclass()) {
                JobConfig = (JobConfig) tmpClass.getAnnotation(JobConfig.class);
                if (JobConfig != null) {
                    return JobConfig;
                }
                ++i;
            }
        }

        return JobConfig;
    }
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值