SpringApplication启动时如何获取带有指定注解的实体？

嘿，这个需求我之前在项目里也遇到过，尤其是要批量处理JPA实体（比如动态生成SQL、初始化基础数据）的时候，下面给你两种靠谱的实现方式，都是Spring生态里常用的套路：

方法一：手动类路径扫描（通用型） #

这种方法不依赖JPA的特定机制，只要是带有指定注解的类，不管有没有被Spring容器管理，都能扫到。核心是用Spring提供的ClassPathScanningCandidateComponentProvider工具类来做类路径扫描。

实现代码 #

import org.springframework.boot.CommandLineRunner;
import org.springframework.context.annotation.Configuration;
import org.springframework.core.type.filter.AnnotationTypeFilter;
import org.springframework.context.annotation.ClassPathScanningCandidateComponentProvider;
import javax.persistence.Table;
import java.util.Set;

@Configuration
public class EntityScanner implements CommandLineRunner {

    @Override
    public void run(String... args) throws Exception {
        // 创建扫描器，关闭默认过滤器（默认只扫@Component系列注解的类）
        ClassPathScanningCandidateComponentProvider scanner = 
            new ClassPathScanningCandidateComponentProvider(false);
        
        // 添加要扫描的注解过滤器：这里指定@Table
        scanner.addIncludeFilter(new AnnotationTypeFilter(Table.class));
        
        // 替换成你的实体类所在的基础包路径
        Set<org.springframework.beans.factory.config.BeanDefinition> candidates = 
            scanner.findCandidateComponents("com.yourproject.entity");
        
        // 遍历扫描结果，转换成Class对象并处理
        for (org.springframework.beans.factory.config.BeanDefinition beanDef : candidates) {
            Class<?> entityClass = Class.forName(beanDef.getBeanClassName());
            System.out.println("找到带@Table注解的实体类：" + entityClass.getName());
            // 这里可以添加你的业务逻辑，比如记录实体信息、做统一初始化等
        }
    }
}

关键说明 #

• 关闭默认过滤器是因为ClassPathScanningCandidateComponentProvider默认只扫描带有@Component、@Service等Spring组件注解的类，而JPA实体通常只加@Entity和@Table，所以要关掉默认规则，自定义扫描条件。
• 扫描包路径一定要准确，避免漏扫实体或者扫到无关类。

方法二：借助JPA EntityManager（适用于Spring Data JPA项目） #

如果你的项目已经用了Spring Data JPA，那么可以直接从JPA的元模型里获取所有注册的实体类，再筛选带有@Table注解的。这种方法不需要手动指定包路径，更贴合JPA生态。

实现代码 #

import org.springframework.boot.CommandLineRunner;
import org.springframework.context.annotation.Configuration;
import javax.persistence.EntityManager;
import javax.persistence.PersistenceContext;
import javax.persistence.metamodel.EntityType;
import java.util.Set;

@Configuration
public class JpaEntityScanner implements CommandLineRunner {

    @PersistenceContext
    private EntityManager entityManager;

    @Override
    public void run(String... args) throws Exception {
        // 从JPA元模型中获取所有已注册的实体类型
        Set<EntityType<?>> entityTypes = entityManager.getMetamodel().getEntities();
        
        // 遍历筛选带@Table注解的实体
        for (EntityType<?> entityType : entityTypes) {
            Class<?> entityClass = entityType.getJavaType();
            if (entityClass.isAnnotationPresent(Table.class)) {
                System.out.println("找到带@Table注解的实体类：" + entityClass.getName());
            }
        }
    }
}

关键说明 #

• 这种方法的前提是你的实体类已经被JPA正确识别：也就是加了@Entity注解，并且在Spring的扫描范围内（比如配置了@EntityScan或者默认扫主包下的类）。
• 有些实体可能只加了@Entity没加@Table（JPA允许这种情况，会用类名作为默认表名），如果你的需求是必须带有@Table，一定要加上isAnnotationPresent的判断。

额外注意事项 #

• 如果你想在Spring启动更早的阶段执行扫描逻辑，可以把CommandLineRunner换成ApplicationListener<ApplicationStartedEvent>或者ApplicationListener<ContextStartedEvent>，不过通常用CommandLineRunner或者ApplicationListener<ContextRefreshedEvent>就足够了，这时候所有Bean都已经初始化完成。
• 扫描到实体类后，你可以根据需求做任何处理：比如记录实体的表名、字段信息，或者执行批量的数据初始化操作。

内容的提问来源于stack exchange，提问作者Ragnar