private SdkComponents(RunnerApi.Components components) { if (components == null) { return; } reservedIds.addAll(components.getTransformsMap().keySet()); reservedIds.addAll(components.getPcollectionsMap().keySet()); reservedIds.addAll(components.getWindowingStrategiesMap().keySet()); reservedIds.addAll(components.getCodersMap().keySet()); reservedIds.addAll(components.getEnvironmentsMap().keySet()); componentsBuilder.mergeFrom(components); }
@Test public void toAndFromProto() throws Exception { SdkComponents sdkComponents = SdkComponents.create(); sdkComponents.registerEnvironment(Environments.createDockerEnvironment("java")); RunnerApi.Coder coderProto = CoderTranslation.toProto(coder, sdkComponents); Components encodedComponents = sdkComponents.toComponents(); Coder<?> decodedCoder = CoderTranslation.fromProto( coderProto, RehydratedComponents.forComponents(encodedComponents)); assertThat(decodedCoder, equalTo(coder)); if (KNOWN_CODERS.contains(coder)) { for (RunnerApi.Coder encodedCoder : encodedComponents.getCodersMap().values()) { assertThat( encodedCoder.getSpec().getSpec().getUrn(), not(equalTo(CoderTranslation.JAVA_SERIALIZED_CODER_URN))); } } }
for (String coderId : components.getCodersMap().keySet()) { for (String componentCoderId : components.getCodersOrThrow(coderId).getComponentCoderIdsList()) {
fusedPipelineProto.getComponents().getCodersMap(), equalTo(protoPipeline.getComponents().getCodersMap())); assertThat( fusedPipelineProto.getComponents().getWindowingStrategiesMap(),