ru.yandex.qatools.embed.service.PostgresEmbeddedService Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of embedded-services Show documentation

There is a newer version: 1.21

package ru.yandex.qatools.embed.service;

import ru.yandex.qatools.embed.postgresql.PostgresExecutable;
import ru.yandex.qatools.embed.postgresql.PostgresProcess;
import ru.yandex.qatools.embed.postgresql.PostgresStarter;
import ru.yandex.qatools.embed.postgresql.config.PostgresConfig;

import java.io.IOException;

import static ru.yandex.qatools.embed.postgresql.config.AbstractPostgresConfig.*;
import static ru.yandex.qatools.embed.postgresql.distribution.Version.Main.PRODUCTION;

/**
 * Embedded Postgres server
 */
public class PostgresEmbeddedService extends AbstractEmbeddedService {

    protected final String host;
    protected final int port;
    protected final String dbName;
    protected final String username;
    protected final String password;
    PostgresProcess process;

    public PostgresEmbeddedService(String host, int port, String username, String password, String dbName) throws IOException {
        this(host, port, username, password, dbName, null, true, 10000);
    }

    public PostgresEmbeddedService(String host, int port,
                                   String username, String password, String dbName,
                                   String dataDirectory, boolean enabled, int initTimeout) throws IOException {
        super(dataDirectory, enabled, initTimeout);
        this.username = username;
        this.password = password;
        this.host = host;
        this.port = port;
        this.dbName = dbName;
    }

    @Override
    public void doStart() throws Exception {
        PostgresStarter runtime = PostgresStarter.getDefaultInstance();
        final PostgresConfig configDb;
        configDb = new PostgresConfig(
                PRODUCTION, new Net(host, port), new Storage(dbName, dataDirectory),
                new Timeout(initTimeout), new Credentials(username, password));
        PostgresExecutable exec = runtime.prepare(configDb);
        process = exec.start();
    }

    @Override
    public void doStop() throws Exception {
        if (process != null) {
            process.stop();
            process = null;
        }
    }
}