How to use

fromProto

method

in

org.apache.beam.runners.core.construction.WindowingStrategyTranslation

 @Override
 public WindowingStrategy<?, ?> load(String id) throws Exception {
  @Nullable
  RunnerApi.WindowingStrategy windowingStrategyProto =
    components.getWindowingStrategiesOrDefault(id, null);
  checkState(
    windowingStrategyProto != null,
    "No WindowingStrategy with id '%s' in serialized components",
    id);
  return WindowingStrategyTranslation.fromProto(
    windowingStrategyProto, RehydratedComponents.this);
 }
});

/**
 * Converts from a {@link RunnerApi.WindowingStrategy} accompanied by {@link Components} to the
 * SDK's {@link WindowingStrategy}.
 */
public static WindowingStrategy<?, ?> fromProto(RunnerApi.MessageWithComponents proto)
  throws InvalidProtocolBufferException {
 switch (proto.getRootCase()) {
  case WINDOWING_STRATEGY:
   return fromProto(
     proto.getWindowingStrategy(),
     RehydratedComponents.forComponents(proto.getComponents()));
  default:
   throw new IllegalArgumentException(
     String.format(
       "Expected a %s with components but received %s",
       RunnerApi.WindowingStrategy.class.getCanonicalName(), proto));
 }
}

 public static WindowingStrategy getWindowingStrategy(
   String pCollectionId, RunnerApi.Components components) {
  RunnerApi.WindowingStrategy windowingStrategyProto =
    components.getWindowingStrategiesOrThrow(
      components.getPcollectionsOrThrow(pCollectionId).getWindowingStrategyId());

  final WindowingStrategy<?, ?> windowingStrategy;
  try {
   return WindowingStrategyTranslation.fromProto(
     windowingStrategyProto, RehydratedComponents.forComponents(components));
  } catch (InvalidProtocolBufferException e) {
   throw new IllegalStateException(
     String.format(
       "Unable to hydrate windowing strategy %s for %s.",
       windowingStrategyProto, pCollectionId),
     e);
  }
 }
}

 public static WindowingStrategy getWindowingStrategy(
   String pCollectionId, RunnerApi.Components components) {
  RunnerApi.WindowingStrategy windowingStrategyProto =
    components.getWindowingStrategiesOrThrow(
      components.getPcollectionsOrThrow(pCollectionId).getWindowingStrategyId());

  final WindowingStrategy<?, ?> windowingStrategy;
  try {
   return WindowingStrategyTranslation.fromProto(
     windowingStrategyProto, RehydratedComponents.forComponents(components));
  } catch (InvalidProtocolBufferException e) {
   throw new IllegalStateException(
     String.format(
       "Unable to hydrate windowing strategy %s for %s.",
       windowingStrategyProto, pCollectionId),
     e);
  }
 }
}

/**
 * Converts from {@link RunnerApi.WindowingStrategy} to the SDK's {@link WindowingStrategy} using
 * the provided components to dereferences identifiers found in the proto.
 */
public static WindowingStrategy<?, ?> fromProto(
  RunnerApi.WindowingStrategy proto, RehydratedComponents components)
  throws InvalidProtocolBufferException {
 SdkFunctionSpec windowFnSpec = proto.getWindowFn();
 WindowFn<?, ?> windowFn = windowFnFromProto(windowFnSpec);
 TimestampCombiner timestampCombiner = timestampCombinerFromProto(proto.getOutputTime());
 AccumulationMode accumulationMode = fromProto(proto.getAccumulationMode());
 Trigger trigger = TriggerTranslation.fromProto(proto.getTrigger());
 ClosingBehavior closingBehavior = fromProto(proto.getClosingBehavior());
 Duration allowedLateness = Duration.millis(proto.getAllowedLateness());
 OnTimeBehavior onTimeBehavior = fromProto(proto.getOnTimeBehavior());
 return WindowingStrategy.of(windowFn)
   .withAllowedLateness(allowedLateness)
   .withMode(accumulationMode)
   .withTrigger(trigger)
   .withTimestampCombiner(timestampCombiner)
   .withClosingBehavior(closingBehavior)
   .withOnTimeBehavior(onTimeBehavior);
}

try {
 windowingStrategy =
   WindowingStrategyTranslation.fromProto(windowingStrategyProto, rehydratedComponents);
} catch (InvalidProtocolBufferException e) {
 throw new IllegalStateException(

@Test
public void testToProtoAndBack() throws Exception {
 WindowingStrategy<?, ?> windowingStrategy = toProtoAndBackSpec.getWindowingStrategy();
 SdkComponents components = SdkComponents.create();
 components.registerEnvironment(Environments.createDockerEnvironment("java"));
 WindowingStrategy<?, ?> toProtoAndBackWindowingStrategy =
   WindowingStrategyTranslation.fromProto(
     WindowingStrategyTranslation.toMessageProto(windowingStrategy, components));
 assertThat(
   toProtoAndBackWindowingStrategy,
   equalTo((WindowingStrategy) windowingStrategy.fixDefaults()));
}

try {
 windowingStrategy =
   WindowingStrategyTranslation.fromProto(windowingStrategyProto, rehydratedComponents);
} catch (InvalidProtocolBufferException e) {
 throw new IllegalStateException(

try {
 windowingStrategy =
   WindowingStrategyTranslation.fromProto(windowingStrategyProto, rehydratedComponents);
} catch (InvalidProtocolBufferException e) {
 throw new IllegalStateException(

try {
 windowingStrategy =
   WindowingStrategyTranslation.fromProto(windowingStrategyProto, rehydratedComponents);
} catch (InvalidProtocolBufferException e) {
 throw new IllegalStateException(

 windowingStrategy =
   (WindowingStrategy<Object, BoundedWindow>)
     WindowingStrategyTranslation.fromProto(windowingStrategyProto, rehydratedComponents);
} catch (InvalidProtocolBufferException e) {
 throw new IllegalStateException(

 windowingStrategy =
   (WindowingStrategy<Object, BoundedWindow>)
     WindowingStrategyTranslation.fromProto(windowingStrategyProto, rehydratedComponents);
} catch (InvalidProtocolBufferException e) {
 throw new IllegalStateException(

 @Test
 public void testToProtoAndBackWithComponents() throws Exception {
  WindowingStrategy<?, ?> windowingStrategy = toProtoAndBackSpec.getWindowingStrategy();
  SdkComponents components = SdkComponents.create();
  components.registerEnvironment(Environments.createDockerEnvironment("java"));
  RunnerApi.WindowingStrategy proto =
    WindowingStrategyTranslation.toProto(windowingStrategy, components);
  RehydratedComponents protoComponents =
    RehydratedComponents.forComponents(components.toComponents());

  assertThat(
    WindowingStrategyTranslation.fromProto(proto, protoComponents).fixDefaults(),
    equalTo(windowingStrategy.fixDefaults()));

  protoComponents.getCoder(
    components.registerCoder(windowingStrategy.getWindowFn().windowCoder()));
  assertThat(
    proto.getAssignsToOneWindow(),
    equalTo(windowingStrategy.getWindowFn().assignsToOneWindow()));
 }
}