项目启动后自动启动某个包下带自定义标签的方法

import lombok.extern.slf4j.Slf4j;
import org.springframework.boot.CommandLineRunner;
import org.springframework.boot.autoconfigure.condition.ConditionalOnExpression;
import org.springframework.stereotype.Component;

/**
 * 集成测试总入口
 * 项目启动后会自动执行所有方法
 * com.test.integrate包下类中加上@RunMethod的方法,
 */
@Component
@ConditionalOnExpression("${unitcase.biz.enable:false}")
@Slf4j
public class IntegrationTest implements CommandLineRunner {

    @Override
    public void run(String... args){
        RunClassHolder.run();
    }
}
import com.test.annotation.RunMethod;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.BeansException;
import org.springframework.context.ApplicationContext;
import org.springframework.context.ApplicationContextAware;
import org.springframework.stereotype.Component;

import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.util.List;
import java.util.concurrent.ExecutorService;

/**
 * 根据class获取bean
 */
@Component
@Slf4j
public class RunClassHolder implements ApplicationContextAware {
    private static ExecutorService analysisExecutor = ThreadUtil.newFixedThreadPool("integration-test", 32);

    public static ApplicationContext ctx;

    public static void run(){
        List<Class> classList = ClazzUtils.getClazz();

        for(Class clazz : classList){
            analysisExecutor.submit(() ->  {
                Method[] methods = clazz.getDeclaredMethods();
                for (Method method : methods) {
                    if (method.getAnnotation(RunMethod.class) == null){
                        continue;
                    }
                    try {
                        method.invoke(RunClassHolder.getBean(clazz));
                    } catch (IllegalAccessException e) {
                        log.error("执行集成测试报错",e);
                    } catch (InvocationTargetException e) {
                        log.error("执行集成测试报错",e);
                    }
                }
            });
        }
    }

    @Override
    public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
        ctx = applicationContext;
    }

    /**
     * 根据类型获得bean
     */
    public static <T> T getBean(Class<T> clazz) {
        return ctx.getBean(clazz);
    }

    /**
     * 根据名称名称获得bean
     */
    public static Object getBean(String name) {
        return ctx.getBean(name);
    }


}


import lombok.extern.slf4j.Slf4j;
import org.springframework.core.io.Resource;
import org.springframework.core.io.support.PathMatchingResourcePatternResolver;
import org.springframework.core.io.support.ResourcePatternResolver;
import org.springframework.core.type.classreading.CachingMetadataReaderFactory;
import org.springframework.core.type.classreading.MetadataReader;
import org.springframework.core.type.classreading.MetadataReaderFactory;
import org.springframework.util.ClassUtils;

import java.io.IOException;
import java.util.ArrayList;
import java.util.List;


/**
 * 获取某个包下所有class
 */
@Slf4j
public class ClazzUtils {
    private static final String BASE_PACKAGE = "com.test.integrate";
    private static final String RESOURCE_PATTERN = "/**/*.class";

    public static List<Class> getClazz() {
        List<Class> classList = new ArrayList<>();

        ResourcePatternResolver resourcePatternResolver = new PathMatchingResourcePatternResolver();
        try {
            String pattern = ResourcePatternResolver.CLASSPATH_ALL_URL_PREFIX +
                    ClassUtils.convertClassNameToResourcePath(BASE_PACKAGE) + RESOURCE_PATTERN;
            Resource[] resources = resourcePatternResolver.getResources(pattern);

            MetadataReaderFactory readerfactory = new CachingMetadataReaderFactory(resourcePatternResolver);
            for (Resource resource : resources) {
                MetadataReader reader = readerfactory.getMetadataReader(resource);

                String classname = reader.getClassMetadata().getClassName();
                Class<?> clazz = Class.forName(classname);
                classList.add(clazz);
            }
        } catch (IOException | ClassNotFoundException e) {
            log.error("获取集成测试类失败",e);
        }

        return classList;
    }

}

package com.test.integrate;


import org.springframework.stereotype.Service;

import javax.annotation.Resource;

/**
 * 测试类
 */
@Service
public class Test {
    @Resource
    private TestService testService;

    @RunMethod
    public void smsTemplateTest(){
      //todo 这里组装逻辑
    }

}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值