io.trino.tests.product.launcher.docker.ContainerUtil Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of trino-product-tests-launcher Show documentation
Trino - Product tests launcher
There is a newer version: 448
Show newest version
/*
 * 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.
 */
package io.trino.tests.product.launcher.docker;

import com.github.dockerjava.api.DockerClient;
import com.github.dockerjava.api.command.ListContainersCmd;
import com.github.dockerjava.api.command.ListNetworksCmd;
import com.github.dockerjava.api.exception.ConflictException;
import com.github.dockerjava.api.exception.NotFoundException;
import com.github.dockerjava.api.model.Container;
import com.github.dockerjava.api.model.Network;
import io.trino.testing.containers.wait.strategy.SelectedPortWaitStrategy;
import org.testcontainers.DockerClientFactory;
import org.testcontainers.containers.wait.strategy.WaitStrategy;

import java.util.List;
import java.util.function.Function;

public final class ContainerUtil
{
    private ContainerUtil() {}

    public static void killContainers(DockerClient dockerClient, Function filter)
    {
        while (true) {
            ListContainersCmd listContainersCmd = filter.apply(dockerClient.listContainersCmd()
                    .withShowAll(true));

            List containers = listContainersCmd.exec();
            if (containers.isEmpty()) {
                break;
            }
            for (Container container : containers) {
                try {
                    dockerClient.removeContainerCmd(container.getId())
                            .withForce(true)
                            .exec();
                }
                catch (ConflictException | NotFoundException ignored) {
                }
            }
        }
    }

    public static void removeNetworks(DockerClient dockerClient, Function filter)
    {
        ListNetworksCmd listNetworksCmd = filter.apply(dockerClient.listNetworksCmd());
        List networks = listNetworksCmd.exec();
        for (Network network : networks) {
            try {
                dockerClient.removeNetworkCmd(network.getId())
                        .exec();
            }
            catch (NotFoundException ignored) {
                // Possible when previous tests invocation leaves a network behind and it is being garbage collected by Ryuk in the background.
            }
        }
    }

    public static void killContainersReaperContainer(DockerClient dockerClient)
    {
        dockerClient.removeContainerCmd("testcontainers-ryuk-" + DockerClientFactory.SESSION_ID)
                .withForce(true)
                .exec();
    }

    public static WaitStrategy forSelectedPorts(int... ports)
    {
        return new SelectedPortWaitStrategy(ports);
    }
}