Add Data Sampling capability URN to Java SDK (apache#25593)

* Add Data Sampling capability URN to Java SDK * trigger tests * Update runners/core-construction-java/src/main/java/org/apache/beam/runners/core/construction/Environments.java --------- Co-authored-by: Sam Rohde <srohde@google.com> Co-authored-by: Lukasz Cwik <lcwik@google.com>
akvelon · Mar 10, 2023 · fd6f732 · fd6f732
1 parent ed2cc6a
commit fd6f732
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/...nstruction-java/src/main/java/org/apache/beam/runners/core/construction/Environments.java b/...nstruction-java/src/main/java/org/apache/beam/runners/core/construction/Environments.java
@@ -423,6 +423,7 @@ public static Set<String> getJavaCapabilities() {
     capabilities.add("beam:version:sdk_base:" + JAVA_SDK_HARNESS_CONTAINER_URL);
     capabilities.add(BeamUrns.getUrn(SplittableParDoComponents.TRUNCATE_SIZED_RESTRICTION));
     capabilities.add(BeamUrns.getUrn(Primitives.TO_STRING));
+    capabilities.add(BeamUrns.getUrn(StandardProtocols.Enum.DATA_SAMPLING));
     return capabilities.build();
   }
 

diff --git a/...uction-java/src/test/java/org/apache/beam/runners/core/construction/EnvironmentsTest.java b/...uction-java/src/test/java/org/apache/beam/runners/core/construction/EnvironmentsTest.java
@@ -204,6 +204,9 @@ public void testCapabilities() {
     assertThat(
         Environments.getJavaCapabilities(),
         hasItem(BeamUrns.getUrn(RunnerApi.StandardProtocols.Enum.STATE_CACHING)));
+    assertThat(
+        Environments.getJavaCapabilities(),
+        hasItem(BeamUrns.getUrn(RunnerApi.StandardProtocols.Enum.DATA_SAMPLING)));
     // Check that SDF truncation is supported
     assertThat(
         Environments.getJavaCapabilities(),