testcontainers - 在 init 脚本之前在 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';

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

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);
    }
}