stagePayload = RunnerApi.ExecutableStagePayload.parseFrom(transform.getSpec().getPayload()); } catch (IOException e) { throw new RuntimeException(e);
stagePayload = RunnerApi.ExecutableStagePayload.parseFrom(transform.getSpec().getPayload()); } catch (IOException e) { throw new RuntimeException(e);
private RemoteStageEvaluator(PTransformNode transform) throws Exception { this.transform = transform; ExecutableStage stage = ExecutableStage.fromPayload( ExecutableStagePayload.parseFrom(transform.getTransform().getSpec().getPayload())); this.outputs = new ArrayList<>(); StageBundleFactory stageFactory = jobFactory.forStage(stage); this.bundle = stageFactory.getBundle( BundleFactoryOutputReceiverFactory.create( bundleFactory, stage.getComponents(), outputs::add), StateRequestHandler.unsupported(), BundleProgressHandler.ignored()); // TODO(BEAM-4680): Add support for timers as inputs to the ULR this.mainInput = Iterables.getOnlyElement(bundle.getInputReceivers().values()); }
try { stagePayload = RunnerApi.ExecutableStagePayload.parseFrom( transform.getTransform().getSpec().getPayload()); } catch (IOException e) {
try { stagePayload = RunnerApi.ExecutableStagePayload.parseFrom( transform.getTransform().getSpec().getPayload()); } catch (IOException e) {
ExecutableStagePayload.parseFrom(stagePTransform.getSpec().getPayload()); assertThat(payload.getTransformsList(), contains("pt")); assertThat(ExecutableStage.fromPayload(payload), equalTo(stage));
ExecutableStagePayload.parseFrom(ptransform.getSpec().getPayload()); assertThat(payload.getTransformsList(), contains("parDo", "window")); ExecutableStage desered = ExecutableStage.fromPayload(payload);
ExecutableStagePayload.parseFrom(stagePTransform.getSpec().getPayload()); assertThat(payload.getTransformsList(), contains("pt")); assertThat(ExecutableStage.fromPayload(payload), equalTo(stage));
private static void validateExecutableStage( String id, PTransform transform, Components outerComponents) throws Exception { ExecutableStagePayload payload = ExecutableStagePayload.parseFrom(transform.getSpec().getPayload());
this.stage = ExecutableStage.fromPayload( ExecutableStagePayload.parseFrom(transform.getTransform().getSpec().getPayload())); this.outputs = new ArrayList<>();
/** Extracts an {@link ExecutableStagePayload} from the given transform. */ public static ExecutableStagePayload getExecutableStagePayload( AppliedPTransform<?, ?, ?> appliedTransform) throws IOException { RunnerApi.PTransform transform = PTransformTranslation.toProto( appliedTransform, SdkComponents.create(appliedTransform.getPipeline().getOptions())); checkArgument(ExecutableStage.URN.equals(transform.getSpec().getUrn())); return ExecutableStagePayload.parseFrom(transform.getSpec().getPayload()); } }