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

最新推荐文章于 2024-04-04 08:09:34 发布

原创最新推荐文章于 2024-04-04 08:09:34 发布 · 1.3k 阅读

3 ·

CC 4.0 BY-SA版权

文章标签：

#spring cloud #java

springcloud部署专栏收录该内容

4 篇文章

订阅专栏

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;
    }
}