Java Code Examples for com.spotify.docker.client.messages.HostConfig#Builder

private static Builder keycloakConfigBuilder(final String image,
                                             final String userName,
                                             final String password,
                                             final List<String> portMappingPorts,
                                             final DockerCommandLineConfig config)
{
  HostConfig.Builder hostConfig = HostConfig.builder()
      .portBindings(portMappingPorts.stream()
          .collect(toMap(o -> o, b -> singletonList(randomPort(PORT_MAPPING_IP)))));

  return DockerContainerConfig.builder()
      .image(image)
      .env("KEYCLOAK_USER=" + userName, "KEYCLOAK_PASSWORD=" + password)
      .withHostConfigBuilder(hostConfig.appendBinds(config.getPathBinds()))
      .withDockerClientBuilder(defaultDockerClientBuilder().readTimeoutMillis(SECONDS.toMillis(5000)));
}
 

@Override
public void decorateHostConfig(Job job, Optional<String> dockerVersion,
                               HostConfig.Builder hostConfigBuilder) {
  final HostConfig hostConfig = hostConfigBuilder.build();
  if (useSyslogRedirector(dockerVersion)) {
    final List<String> binds = Lists.newArrayList();
    if (hostConfig.binds() != null) {
      binds.addAll(hostConfig.binds());
    }
    binds.add("/usr/lib/helios:/helios:ro");
    hostConfigBuilder.binds(binds);
  } else {
    final ImmutableMap.Builder<String, String> logOpts = ImmutableMap.builder();

    logOpts.put("syslog-address", "udp://" + syslogHostPort);
    logOpts.put("syslog-facility", "local0"); // match the behavior of syslog-redirector

    logOpts.put("tag", job.getId().toString());

    hostConfigBuilder.logConfig(LogConfig.create("syslog", logOpts.build()));
  }
}
 

@Test
public void testWithDockerVersionPre1_9() {
  final Optional<String> dockerVersion = Optional.of("1.6.0");

  final SyslogRedirectingContainerDecorator decorator =
      new SyslogRedirectingContainerDecorator(SYSLOG_HOST_PORT);

  final HostConfig.Builder hostBuilder = HostConfig.builder();
  decorator.decorateHostConfig(JOB, dockerVersion, hostBuilder);

  final ContainerConfig.Builder containerBuilder = ContainerConfig.builder();
  decorator.decorateContainerConfig(JOB, imageInfo, dockerVersion, containerBuilder);

  final ContainerConfig containerConfig = containerBuilder.build();
  assertThat(containerConfig.entrypoint(), hasItem("/helios/syslog-redirector"));

  final HostConfig hostConfig = hostBuilder.build();
  assertNull(hostConfig.logConfig());
  assertFalse(hostConfig.binds().isEmpty());
}
 

@Test
public void testWithDockerVersionPost1_9() {
  final Optional<String> dockerVersion = Optional.of("1.12.1");

  final SyslogRedirectingContainerDecorator decorator =
      new SyslogRedirectingContainerDecorator(SYSLOG_HOST_PORT);

  final HostConfig.Builder hostBuilder = HostConfig.builder();
  decorator.decorateHostConfig(JOB, dockerVersion, hostBuilder);

  final ContainerConfig.Builder containerBuilder = ContainerConfig.builder();
  decorator.decorateContainerConfig(JOB, imageInfo, dockerVersion, containerBuilder);

  final ContainerConfig containerConfig = containerBuilder.build();
  assertThat(containerConfig.entrypoint(), not(hasItem("/helios/syslog-redirector")));

  final HostConfig hostConfig = hostBuilder.build();
  final LogConfig logConfig = hostConfig.logConfig();
  assertEquals("syslog", logConfig.logType());
  assertEquals(JOB.getId().toString(), logConfig.logOptions().get("tag"));
  assertEquals("udp://" + SYSLOG_HOST_PORT, logConfig.logOptions().get("syslog-address"));
}
 

private static DockerContainerConfig.Builder nginxConfigBuilder(final String image,
                                                                final List<String> portMappingPorts,
                                                                final DockerCommandLineConfig config)
{
  HostConfig.Builder hostConfig = HostConfig.builder()
      .portBindings(portMappingPorts.stream()
          .collect(toMap(o -> o, b -> singletonList(randomPort(PORT_MAPPING_IP)))));

  return DockerContainerConfig.builder()
      .image(image)
      .withHostConfigBuilder(hostConfig.appendBinds(config.getPathBinds()))
      .withDockerClientBuilder(defaultDockerClientBuilder().readTimeoutMillis(SECONDS.toMillis(5000)));
}
 

/**
 * Create a container host configuration for the job.
 *
 * @return The host configuration.
 */
public HostConfig hostConfig(final Optional<String> dockerVersion) {
  final List<String> securityOpt = job.getSecurityOpt();
  final HostConfig.Builder builder = HostConfig.builder()
      .binds(binds())
      .portBindings(portBindings())
      .dns(dns)
      .securityOpt(securityOpt.toArray(new String[securityOpt.size()]))
      .runtime(job.getRuntime())
      .networkMode(job.getNetworkMode());

  if (!job.getRamdisks().isEmpty()) {
    builder.tmpfs(job.getRamdisks());
  }

  final Resources resources = job.getResources();
  if (resources != null) {
    builder.memory(resources.getMemory());
    builder.memorySwap(resources.getMemorySwap());
    builder.cpusetCpus(resources.getCpuset());
    builder.cpuShares(resources.getCpuShares());
  }

  builder.capAdd(ImmutableList.copyOf(job.getAddCapabilities()));
  builder.capDrop(ImmutableList.copyOf(job.getDropCapabilities()));

  for (final ContainerDecorator decorator : containerDecorators) {
    decorator.decorateHostConfig(job, dockerVersion, builder);
  }

  return builder.build();
}
 

@Override
public void decorateHostConfig(Job job, Optional<String> dockerVersion,
                               HostConfig.Builder hostConfigBuilder) {
  final List<String> b = Lists.newArrayList();

  final HostConfig hostConfig = hostConfigBuilder.build();

  if (hostConfig.binds() != null) {
    b.addAll(hostConfig.binds());
  }

  b.addAll(this.binds);

  hostConfigBuilder.binds(b);
}
 

@Test
public void simpleTest() {
  final List<String> hosts = ImmutableList.of("one:one", "two:two");
  final AddExtraHostContainerDecorator decorator = new AddExtraHostContainerDecorator(hosts);

  final HostConfig.Builder hostBuilder = HostConfig.builder();
  decorator.decorateHostConfig(null, Optional.absent(), hostBuilder);

  final HostConfig config = hostBuilder.build();
  assertThat(config.extraHosts(), equalTo(hosts));
}
 

public static DockerContainer create(CreateAgentRequest request, PluginSettings settings, DockerClient docker,
                                     ConsoleLogAppender consoleLogAppender) throws InterruptedException, DockerException {
    String containerName = UUID.randomUUID().toString();

    HashMap<String, String> labels = labelsFrom(request);
    String imageName = image(request.properties());
    List<String> env = environmentFrom(request, settings, containerName);

    try {
        docker.inspectImage(imageName);
        if (settings.pullOnContainerCreate()) {
            consoleLogAppender.accept("Pulling a fresh version of " + imageName + ".");
            LOG.info("Pulling a fresh version of " + imageName + ".");
            docker.pull(imageName);
        }
    } catch (ImageNotFoundException ex) {
        consoleLogAppender.accept("Image " + imageName + " not found, attempting to download.");
        LOG.info("Image " + imageName + " not found, attempting to download.");
        docker.pull(imageName);
    }

    ContainerConfig.Builder containerConfigBuilder = ContainerConfig.builder();
    if (StringUtils.isNotBlank(request.properties().get("Command"))) {
        containerConfigBuilder.cmd(splitIntoLinesAndTrimSpaces(request.properties().get("Command")).toArray(new String[]{}));
    }

    final String hostConfig = request.properties().get("Hosts");
    final String reservedMemory = request.properties().get("ReservedMemory");
    final String maxMemory = request.properties().get("MaxMemory");
    final String cpus = request.properties().get("Cpus");
    final String volumeMounts = request.properties().get("Mounts");

    HostConfig.Builder hostBuilder = HostConfig.builder()
            .privileged(privileged(request.properties()))
            .extraHosts(new Hosts(hostConfig))
            .memoryReservation(new MemorySpecification(reservedMemory).getMemory())
            .memory(new MemorySpecification(maxMemory).getMemory());

    CpusSpecification cpusValue = new CpusSpecification(cpus);
    if (cpusValue.getCpus() != null) {
        hostBuilder
                .cpuPeriod(cpusValue.getCpuPeriod())
                .cpuQuota(cpusValue.getCpuQuota());
    }
    if (volumeMounts != null) {
        hostBuilder.appendBinds(Util.splitIntoLinesAndTrimSpaces(volumeMounts));
    }

    ContainerConfig containerConfig = containerConfigBuilder
            .image(imageName)
            .labels(labels)
            .env(env)
            .hostConfig(hostBuilder.build())
            .build();

    consoleLogAppender.accept(String.format("Creating container: %s", containerName));
    ContainerCreation container = docker.createContainer(containerConfig, containerName);
    String id = container.id();

    ContainerInfo containerInfo = docker.inspectContainer(id);

    LOG.debug("Created container " + containerName);
    consoleLogAppender.accept(String.format("Starting container: %s", containerName));
    docker.startContainer(containerName);
    consoleLogAppender.accept(String.format("Started container: %s", containerName));
    LOG.debug("container " + containerName + " started");
    return new DockerContainer(id, containerName, request.jobIdentifier(), containerInfo.created(), request.properties(), request.environment());
}
 

public Builder withHostConfigBuilder(HostConfig.Builder hostConfigBuilder) {
  this.hostConfigBuilder = hostConfigBuilder;
  return this;
}
 

public static HostConfig.Builder defaultHostConfigBuilder() {
  return HostConfig.builder()
      .portBindings(stream(PORT_MAPPING_PORTS)
          .collect(toMap(o -> o, b -> singletonList(randomPort(PORT_MAPPING_IP)))));
}
 

public HostConfig.Builder getHostConfigBuilder() {
  return hostConfigBuilder;
}
 

@Override
public void decorateHostConfig(final Job job, final Optional<String> dockerVersion,
                               final HostConfig.Builder hostConfig) {
  hostConfig.extraHosts(this.extraHosts);
}
 

@Override
public void decorateHostConfig(Job job, Optional<String> dockerVersion,
                               HostConfig.Builder hostConfig) {
  //noop
}
 

void decorateHostConfig(Job job, Optional<String> dockerVersion, HostConfig.Builder hostConfig);