在H2数据库中适配MySQL的date_format函数用于Spring Boot测试

方法1：给H2自定义DATE_FORMAT函数 #

H2支持注册自定义函数模拟MySQL的行为，可在测试启动时配置：

• 实现函数逻辑的Java类：

public class H2DateFormatFunction {
    public static String dateFormat(Date date, String pattern) {
        DateTimeFormatter formatter = DateTimeFormatter.ofPattern(pattern);
        return formatter.format(date.toInstant().atZone(ZoneId.systemDefault()).toLocalDate());
    }
}

• 在H2初始化脚本（如schema-test.sql）中注册函数：

CREATE ALIAS DATE_FORMAT FOR "com.yourpackage.H2DateFormatFunction.dateFormat";

• 在测试配置里指定初始化脚本：

spring.datasource.schema=classpath:schema-test.sql

完成后H2就能识别DATE_FORMAT函数，与MySQL行为一致。

方法2：替换为跨数据库兼容的JPQL查询 #

放弃MySQL特定函数，改用JPA标准日期函数拼接，同时兼容两种数据库：

修改Repository查询：

@Query("SELECT e FROM EmployeeEntity e WHERE CONCAT(YEAR(e.joinedDate), LPAD(MONTH(e.joinedDate), 2, '0')) = :joinedMonth")
List<Employee> findByJoinedMonth(@Param("joinedMonth") String joinedMonth);

通过YEAR和MONTH获取年月，LPAD补零保证格式与%Y%m一致（比如2024年5月转为202405），该写法在MySQL和H2中均可正常执行。

方法3：使用环境特定的查询 #

若必须保留MySQL原生函数，可通过Spring Profiles区分生产与测试环境的查询：

• 生产环境Repository：

@Profile("prod")
public interface EmployeeRepositoryProd extends JpaRepository<EmployeeEntity, Long> {
    @Query("SELECT e FROM EmployeeEntity e WHERE DATE_FORMAT(e.joinedDate,'%Y%m') = :joinedMonth")
    List<Employee> findByJoinedMonth(@Param("joinedMonth") String joinedMonth);
}

• 测试环境Repository（H2兼容写法）：

@Profile("test")
public interface EmployeeRepositoryTest extends JpaRepository<EmployeeEntity, Long> {
    @Query("SELECT e FROM EmployeeEntity e WHERE CONCAT(YEAR(e.joinedDate), LPAD(MONTH(e.joinedDate), 2, '0')) = :joinedMonth")
    List<Employee> findByJoinedMonth(@Param("joinedMonth") String joinedMonth);
}

在测试类上激活test profile即可适配H2环境。

内容的提问来源于stack exchange，提问作者The Anh Nguyen