Environments.createOrGetDefaultEnvironment(Environments.ENVIRONMENT_DOCKER, "java"), is( Environment.newBuilder() .setUrl("java") .setUrn(BeamUrns.getUrn(StandardEnvironments.Environments.DOCKER)) "{\"os\": \"linux\", \"arch\": \"amd64\", \"command\": \"run.sh\", \"env\":{\"k1\": \"v1\", \"k2\": \"v2\"} }"), is( Environment.newBuilder() .setUrn(BeamUrns.getUrn(StandardEnvironments.Environments.PROCESS)) .setPayload( Environments.ENVIRONMENT_PROCESS, "{\"command\": \"run.sh\"}"), is( Environment.newBuilder() .setUrn(BeamUrns.getUrn(StandardEnvironments.Environments.PROCESS)) .setPayload(ProcessPayload.newBuilder().setCommand("run.sh").build().toByteString())
public static Environment createProcessEnvironment( String os, String arch, String command, Map<String, String> env) { ProcessPayload.Builder builder = ProcessPayload.newBuilder(); if (!Strings.isNullOrEmpty(os)) { builder.setOs(os); } if (!Strings.isNullOrEmpty(arch)) { builder.setArch(arch); } if (!Strings.isNullOrEmpty(command)) { builder.setCommand(command); } if (env != null) { builder.putAllEnv(env); } return Environment.newBuilder() .setUrn(BeamUrns.getUrn(StandardEnvironments.Environments.PROCESS)) .setPayload(builder.build().toByteString()) .build(); }
private static Environment createEmbeddedEnvironment(String config) { return Environment.newBuilder() .setUrn(ENVIRONMENT_EMBEDDED) .setPayload(ByteString.copyFromUtf8(MoreObjects.firstNonNull(config, ""))) .build(); }
public static Environment createDockerEnvironment(String dockerImageUrl) { return Environment.newBuilder() .setUrl(dockerImageUrl) .setUrn(BeamUrns.getUrn(StandardEnvironments.Environments.DOCKER)) .setPayload( DockerPayload.newBuilder().setContainerImage(dockerImageUrl).build().toByteString()) .build(); }