org.testcontainers.containers.PgVectorContainerProvider Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of postgresql Show documentation

Isolated container management for Java code testing

There is a newer version: 1.8.0

package org.testcontainers.containers;

import org.testcontainers.jdbc.ConnectionUrl;
import org.testcontainers.utility.DockerImageName;

/**
 * Factory for PgVector containers.
 *
 * @see https://github.com/pgvector/pgvector
 */
public class PgVectorContainerProvider extends JdbcDatabaseContainerProvider {

    private static final String NAME = "pgvector";

    private static final String DEFAULT_TAG = "pg16";

    private static final DockerImageName DEFAULT_IMAGE = DockerImageName.parse("pgvector/pgvector");

    public static final String USER_PARAM = "user";

    public static final String PASSWORD_PARAM = "password";

    @Override
    public boolean supports(String databaseType) {
        return databaseType.equals(NAME);
    }

    @Override
    public JdbcDatabaseContainer newInstance() {
        return newInstance(DEFAULT_TAG);
    }

    @Override
    public JdbcDatabaseContainer newInstance(String tag) {
        return new PostgreSQLContainer(DEFAULT_IMAGE.withTag(tag));
    }

    @Override
    public JdbcDatabaseContainer newInstance(ConnectionUrl connectionUrl) {
        return newInstanceFromConnectionUrl(connectionUrl, USER_PARAM, PASSWORD_PARAM);
    }
}