• Home
• com.github.mike10004
• containment-core
• 0.4
• source code
• DjContainerCopier.java

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

io.github.mike10004.containment.dockerjava.DjContainerCopier Maven / Gradle / Ivy

package io.github.mike10004.containment.dockerjava;

import com.github.dockerjava.api.DockerClient;
import com.github.dockerjava.api.exception.DockerException;
import com.google.common.io.ByteStreams;
import io.github.mike10004.containment.ContainerCopier;
import io.github.mike10004.containment.ContainerInfo;
import io.github.mike10004.containment.ContainmentException;
import org.apache.commons.compress.archivers.tar.TarArchiveEntry;
import org.apache.commons.compress.archivers.tar.TarArchiveInputStream;
import org.apache.commons.io.FilenameUtils;

import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.util.Set;

import static java.util.Objects.requireNonNull;

class DjContainerCopier implements ContainerCopier {

    private final DockerClient client;
    private final ContainerInfo containerInfo;

    public DjContainerCopier(DockerClient client, ContainerInfo containerInfo) {
        this.client = requireNonNull(client);
        this.containerInfo = requireNonNull(containerInfo);
    }

    @Override
    public void copyToContainer(File srcFile, String destinationPathname) throws IOException, ContainmentException {
        try {
            client.copyArchiveToContainerCmd(containerInfo.id())
                    .withHostResource(srcFile.getAbsolutePath())
                    .withRemotePath(destinationPathname)
                    .withNoOverwriteDirNonDir(true)
                    .exec();
        } catch (DockerException e) {
            throw new ContainmentException(e);
        }
    }

    @Override
    public void copyFromContainer(String path, File destinationFile, Set options) throws IOException, ContainmentException {
        if (!options.isEmpty()) {
            throw new ContainmentException("this copier does not support any options, but these were specified: " + options);
        }
        try (InputStream in = client.copyArchiveFromContainerCmd(containerInfo.id(), path).exec();
             TarArchiveInputStream tarIn = new TarArchiveInputStream(in)
        ) {
            TarArchiveEntry entry;
            while ((entry = tarIn.getNextTarEntry()) != null) {
                if (entry.getName().equals(FilenameUtils.getName(path))) {
                    try (OutputStream out = new FileOutputStream(destinationFile)) {
                        ByteStreams.copy(tarIn, out);
                    }
                }
            }
        }
    }

    @Override
    public void unpackTarArchiveToContainer(TarSource source, String destination) throws IOException, ContainmentException {
        try (InputStream tarInput = source.open()) {
            client.copyArchiveToContainerCmd(containerInfo.id())
                    .withRemotePath(destination)
                    .withTarInputStream(tarInput)
                    .exec();
        } catch (DockerException e) {
            throw new ContainmentException(e);
        }
    }
}