Spring Boot测试中如何排除特定实体及仓库类？

首先，咱们先理清你遇到的第一个核心问题：为什么用@ComponentScan排除MyEntity和MyEntityRepository完全无效？

原因很简单：@ComponentScan的过滤规则只对被@Component、@Service、@Repository、@Controller这类Spring组件注解标记的类生效。而@Entity注解的类是由Spring Data JPA的@EntityScan负责扫描和管理的，所以你用@ComponentScan去排除实体类，根本不会被Spring识别到，自然起不到作用。

接下来给你几个可行的解决方案，从临时修复到最佳实践都有：

方案一：正确配置扫描排除（针对主应用或测试类） #

要同时排除实体类和仓库类，需要分别使用@EntityScan和@ComponentScan的过滤规则：

@SpringBootApplication
// 排除实体类：用@EntityScan的excludeFilters
@EntityScan(basePackages = "com.your.base.package", excludeFilters = {
    @ComponentScan.Filter(type = FilterType.ASSIGNABLE_TYPE, classes = MyEntity.class)
})
// 排除仓库类：@Repository属于Spring组件，所以用@ComponentScan的过滤
@ComponentScan(excludeFilters = {
    @ComponentScan.Filter(type = FilterType.ASSIGNABLE_TYPE, classes = MyEntityRepository.class)
})
public class ApiApplication {
    public static void main(String[] args) {
        SpringApplication.run(ApiApplication.class, args);
    }
}

如果是在测试类中，你可以单独在测试类上配置这些扫描规则，避免影响主应用的配置。

方案二：让H2支持自定义向量类型（更优，无需排除实体） #

其实你完全不用排除实体类，只要让H2方言支持你的自定义UserType映射即可，这样测试环境和生产环境的代码一致性更好：

1. 自定义H2方言，注册ARRAY类型映射 #

创建一个继承自H2Dialect的自定义方言，添加对JDBC类型2003（也就是Types.ARRAY）的映射：

public class CustomH2Dialect extends H2Dialect {
    public CustomH2Dialect() {
        super();
        // 把JDBC的ARRAY类型映射到Hibernate的"array"类型
        registerHibernateType(Types.ARRAY, "array");
        // 如果你的向量是Double数组，可以指定更具体的H2类型，比如"double array"
        // registerHibernateType(Types.ARRAY, "double array");
    }
}

2. 配置测试环境使用自定义方言 #

在application-test.properties中添加：

spring.jpa.properties.hibernate.dialect=com.your.package.CustomH2Dialect

3. 完善你的MyVectorType实现 #

你之前的示例代码只实现了sqlTypes()方法，完整的UserType需要实现所有必要方法，比如nullSafeGet、nullSafeSet等，这样Hibernate才能正确在Java对象和数据库类型之间转换：

public class MyVectorType implements UserType {
    @Override
    public int[] sqlTypes() {
        return new int[]{Types.ARRAY};
    }

    @Override
    public Class<?> returnedClass() {
        return MyVector.class; // 替换成你实际的向量Java类
    }

    @Override
    public boolean equals(Object x, Object y) throws HibernateException {
        return Objects.equals(x, y);
    }

    @Override
    public int hashCode(Object x) throws HibernateException {
        return Objects.hashCode(x);
    }

    @Override
    public Object nullSafeGet(ResultSet rs, String[] names, SharedSessionContractImplementor session, Object owner) throws HibernateException, SQLException {
        Array array = rs.getArray(names[0]);
        if (rs.wasNull()) {
            return null;
        }
        // 将数据库的Array转换为你的MyVector对象
        Double[] vectorData = (Double[]) array.getArray();
        return new MyVector(vectorData);
    }

    @Override
    public void nullSafeSet(PreparedStatement st, Object value, int index, SharedSessionContractImplementor session) throws HibernateException, SQLException {
        if (value == null) {
            st.setNull(index, Types.ARRAY);
            return;
        }
        MyVector vector = (MyVector) value;
        // 将MyVector转换为数据库的Array类型
        Array dbArray = session.connection().createArrayOf("DOUBLE", vector.getData());
        st.setArray(index, dbArray);
    }

    // 实现其他必要方法
    @Override
    public Object deepCopy(Object value) throws HibernateException {
        if (value == null) return null;
        MyVector original = (MyVector) value;
        return new MyVector(original.getData().clone());
    }

    @Override
    public boolean isMutable() {
        return true;
    }

    @Override
    public Serializable disassemble(Object value) throws HibernateException {
        return (Serializable) deepCopy(value);
    }

    @Override
    public Object assemble(Serializable cached, Object owner) throws HibernateException {
        return deepCopy(cached);
    }

    @Override
    public Object replace(Object original, Object target, Object owner) throws HibernateException {
        return deepCopy(original);
    }
}

方案三：用Profile实现环境隔离（最佳实践之一） #

如果只是想在测试环境中排除这些类，最干净的方式是使用Spring的@Profile注解：

1. 给实体类和仓库类添加@Profile("!test")，表示只有在非test环境下才加载：

@Entity
@Table("MY_ENTITY")
@Profile("!test")
public class MyEntity {
    @Column
    @Type(type = "com.your.package.MyVectorType")
    private MyVector myVector;
    // 其他字段和方法
}

@Repository
@Profile("!test")
public interface MyEntityRepository extends JpaRepository<MyEntity, Long> {
}

2. 在测试类中激活test环境：

@SpringBootTest
@ActiveProfiles("test")
public class YourTestClass {
    // 测试逻辑
}

这种方式不需要修改主应用的扫描配置，通过环境标签实现隔离，非常灵活。

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