当前位置: 代码迷 >> java >> 如何在spring boot test的@sql注释中打印脚本文件的完整路径
  详细解决方案

如何在spring boot test的@sql注释中打印脚本文件的完整路径

热度:97   发布时间:2023-07-17 20:10:57.0

在多模块项目中,我想确保 Spring 的 @sql 注释使用正确的资源。 有没有办法以某种方式将这些文件的完整路径记录到控制台? Spring 在执行之前会记录脚本文件名,但在不同模块的测试中,这些文件名有时是相同的。

SqlScriptsTestExecutionListener -负责处理@Sql ,在第一步中,您可以更改添加属性,以调试相关的日志logging.level.org.springframework.test.context.jdbc=debug ,但调试信息不完全,并且如果还不够,您应该创建自己的TestExecutionListener并在测试类@TestExecutionListeners(listeners = SqlScriptsCustomTestExecutionListener.class) ,例如:

public class SqlScriptsCustomTestExecutionListener extends AbstractTestExecutionListener {

    @Override
    public void beforeTestMethod(TestContext testContext) {
        List<Resource> scriptResources = new ArrayList<>();
        Set<Sql> sqlAnnotations = AnnotatedElementUtils.getMergedRepeatableAnnotations(testContext.getTestMethod(), Sql.class);
        for (Sql sqlAnnotation : sqlAnnotations) {
            String[] scripts = sqlAnnotation.scripts();
            scripts = TestContextResourceUtils.convertToClasspathResourcePaths(testContext.getTestClass(), scripts);
            scriptResources.addAll(TestContextResourceUtils.convertToResourceList(testContext.getApplicationContext(), scripts));
        }
        if (!scriptResources.isEmpty()) {

            String debugString = scriptResources.stream().map(r -> {
                try {
                    return r.getFile().getAbsolutePath();
                } catch (IOException e) {
                    System.out.println("Unable to found file resource");
                }
                return null;
            }).collect(Collectors.joining(","));

            System.out.println(String.format("Execute sql script :[%s]", debugString));
        }
    }

这只是一个简单的例子,它的工作原理。 我从SqlScriptsTestExecutionListener复制的大部分源代码只是为了解释。 它只是在方法级别的@Sql注释的情况下实现,不包括类级别。 我希望它会帮助你。

  相关解决方案