org.cloudml.connectors.DockerConnector.java Source code

Introduction

Here is the source code for org.cloudml.connectors.DockerConnector.java

Source

/**
 * This file is part of CloudML [ http://cloudml.org ]
 *
 * Copyright (C) 2012 - SINTEF ICT
 * Contact: Franck Chauvel <franck.chauvel@sintef.no>
 *
 * Module: root
 *
 * CloudML is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as
 * published by the Free Software Foundation, either version 3 of
 * the License, or (at your option) any later version.
 *
 * CloudML is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General
 * Public License along with CloudML. If not, see
 * <http://www.gnu.org/licenses/>.
 */
package org.cloudml.connectors;

import com.github.dockerjava.api.DockerClient;
import com.github.dockerjava.api.command.CreateContainerResponse;
import com.github.dockerjava.api.model.ExposedPort;
import com.github.dockerjava.core.DockerClientBuilder;
import com.github.dockerjava.core.DockerClientConfig;
import org.apache.commons.io.IOUtils;
import org.apache.commons.io.LineIterator;

import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.io.StringWriter;
import java.util.logging.Level;
import java.util.logging.Logger;

/**
 * Created by nicolasf on 05.12.14.
 */
public class DockerConnector {

    private static final Logger journal = Logger.getLogger(DockerConnector.class.getName());
    private DockerClient dockerClient;

    public DockerConnector(String endPoint, String version) {
        DockerClientConfig config = DockerClientConfig.createDefaultConfigBuilder().withVersion(version)
                .withUri(endPoint).build();
        dockerClient = DockerClientBuilder.getInstance(config).build();
    }

    public void startDockerDaemon(String keyPath, String user, String host, String dockerPort) {
        SSHConnector connector = new SSHConnector(keyPath, user, host);
        connector.execCommandSsh("sudo docker -H 0.0.0.0 " + dockerPort + " -d");
    }

    public void commit(String msg) {
        dockerClient.commitCmd(msg);
    }

    public void pullImage(String image) {
        dockerClient.pullImageCmd(image).exec();
    }

    public String createContainerWithPorts(String image, String command, int[] ports) {
        ExposedPort[] exposedPorts = new ExposedPort[ports.length];
        for (int i = 0; i < ports.length; i++) {
            exposedPorts[i] = ExposedPort.tcp(ports[i]);
        }

        CreateContainerResponse container = dockerClient.createContainerCmd(image).withCmd(command)
                .withExposedPorts(exposedPorts).exec();
        return container.getId();
    }

    public void BuildImageFromDockerFile(String path) {
        File baseDir = new File(path);

        InputStream response = dockerClient.buildImageCmd(baseDir).exec();

        StringWriter logwriter = new StringWriter();

        try {
            LineIterator itr = IOUtils.lineIterator(response, "UTF-8");
            while (itr.hasNext()) {
                String line = itr.next();
                logwriter.write(line);
                journal.log(Level.INFO, line);
            }
        } catch (IOException e) {
            e.printStackTrace();
        } finally {
            IOUtils.closeQuietly(response);
        }
    }

    public String createContainer(String image, String command) {
        CreateContainerResponse container = dockerClient.createContainerCmd(image).withCmd(command).exec();
        return container.getId();
    }

    public void startContainer(String id) {
        dockerClient.startContainerCmd(id);
    }

    public void stopContainer(String id) {
        dockerClient.stopContainerCmd(id);
    }

}