在初始化脚本之前在 postgres testcontainer 中创建一个文件夹

Question

我有如下一段代码：

public static PostgreSQLContainer<?> postgreDBContainer = new PostgreSQLContainer<>("postgres:12")
        .withInitScript("init-database-test.sql")
        .withUsername("dba")
        .withPassword("dba");

在初始化脚本中，我正在创建一些表空间并关联文件夹：

CREATE TABLESPACE tsd01 OWNER dba LOCATION '/tsd01';
CREATE TABLESPACE tsi01 OWNER dba LOCATION '/tsi01';
CREATE TABLESPACE tsisecurity01 OWNER dba LOCATION '/tsisecurity01';

表空间的这些文件夹应该在初始化脚本运行之前创建。我怎样才能做到这一点？

Answer 1

我能够通过扩展默认 PostgreSQLContainer 并更改 containerIsStarted 方法来解决此问题：

public class CustomPostgreSQLContainer<SELF extends CustomPostgreSQLContainer<SELF>> extends PostgreSQLContainer<SELF> {

    private static final Logger log = LoggerFactory.getLogger(CustomPostgreSQLContainer.class);

    public CustomPostgreSQLContainer() {
        super("postgres:12");
    }

    @Override
    protected void containerIsStarted(InspectContainerResponse containerInfo) {
        try {
            log.debug("M=containerIsStarted, creating database namespace folders and setting permissions");
            execInContainer("mkdir", "/tsd01");
            execInContainer("chown", "-R", "postgres.postgres", "/tsd01/");
            execInContainer("mkdir", "/tsi01");
            execInContainer("chown", "-R", "postgres.postgres", "/tsi01/");
            execInContainer("mkdir", "/tsisecurity01");
            execInContainer("chown", "-R", "postgres.postgres", "/tsisecurity01/");
        } catch (IOException | InterruptedException e) {
            e.printStackTrace();
        }
        super.containerIsStarted(containerInfo);
    }
}

在初始化脚本之前在 postgres testcontainer 中创建一个文件夹

Create a folder inside a postgres testcontainer before init script

testcontainers

testcontainers-junit5