73 lines
2.4 KiB
TypeScript
73 lines
2.4 KiB
TypeScript
|
/*
|
||
|
|
Copyright 2023 The Matrix.org Foundation C.I.C.
|
||
|
|
|
||
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
||
|
|
you may not use this file except in compliance with the License.
|
||
|
|
You may obtain a copy of the License at
|
||
|
|
|
||
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
||
|
|
|
||
|
|
Unless required by applicable law or agreed to in writing, software
|
||
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
||
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||
|
|
See the License for the specific language governing permissions and
|
||
|
|
limitations under the License.
|
||
|
|
*/
|
||
|
|
|
||
|
|
import { Docker } from "../docker";
|
||
|
|
|
||
|
|
export const PG_PASSWORD = "p4S5w0rD";
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Class to manage a postgres database in docker
|
||
|
|
*/
|
||
|
|
export class PostgresDocker extends Docker {
|
||
|
|
/**
|
||
|
|
* @param key an opaque string to use when naming the docker containers instantiated by this class
|
||
|
|
*/
|
||
|
|
public constructor(private key: string) {
|
||
|
|
super();
|
||
|
|
}
|
||
|
|
|
||
|
|
private async waitForPostgresReady(): Promise<void> {
|
||
|
|
const waitTimeMillis = 30000;
|
||
|
|
const startTime = new Date().getTime();
|
||
|
|
let lastErr: Error | null = null;
|
||
|
|
while (new Date().getTime() - startTime < waitTimeMillis) {
|
||
|
|
try {
|
||
|
|
await this.exec(["pg_isready", "-U", "postgres"], true);
|
||
|
|
lastErr = null;
|
||
|
|
break;
|
||
|
|
} catch (err) {
|
||
|
|
console.log("pg_isready: failed");
|
||
|
|
lastErr = err;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
if (lastErr) {
|
||
|
|
console.log("rethrowing");
|
||
|
|
throw lastErr;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
public async start(): Promise<{
|
||
|
|
ipAddress: string;
|
||
|
|
containerId: string;
|
||
|
|
}> {
|
||
|
|
console.log(new Date(), "starting postgres container");
|
||
|
|
const containerId = await this.run({
|
||
|
|
image: "postgres",
|
||
|
|
containerName: `react-sdk-playwright-postgres-${this.key}`,
|
||
|
|
params: ["--tmpfs=/pgtmpfs", "-e", "PGDATA=/pgtmpfs", "-e", `POSTGRES_PASSWORD=${PG_PASSWORD}`],
|
||
|
|
// Optimise for testing - https://www.postgresql.org/docs/current/non-durability.html
|
||
|
|
cmd: ["-c", `fsync=off`, "-c", `synchronous_commit=off`, "-c", `full_page_writes=off`],
|
||
|
|
});
|
||
|
|
|
||
|
|
const ipAddress = await this.getContainerIp();
|
||
|
|
console.log(new Date(), "postgres container up");
|
||
|
|
|
||
|
|
await this.waitForPostgresReady();
|
||
|
|
console.log(new Date(), "postgres container ready");
|
||
|
|
return { ipAddress, containerId };
|
||
|
|
}
|
||
|
|
}
|