|
73 | 73 | import org.slf4j.Logger; |
74 | 74 | import org.slf4j.LoggerFactory; |
75 | 75 |
|
76 | | -@SuppressWarnings({"rawtypes", "nullness"}) |
| 76 | +@SuppressWarnings({"rawtypes", "nullness", "AutoValueSubclassLeaked"}) |
77 | 77 | public class BigQueryIOTranslation { |
78 | 78 |
|
79 | 79 | private static final Logger LOG = LoggerFactory.getLogger(BigQueryIOTranslation.class); |
@@ -217,7 +217,7 @@ public TypedRead<?> fromConfigRow(Row configRow, PipelineOptions options) { |
217 | 217 | (updateCompatibilityBeamVersion != null) ? updateCompatibilityBeamVersion : "2.53.0"; |
218 | 218 |
|
219 | 219 | try { |
220 | | - BigQueryIO.TypedRead.Builder builder = BigQueryIO.TypedRead.builder(); |
| 220 | + BigQueryIO.TypedRead.Builder builder = new AutoValue_BigQueryIO_TypedRead.Builder<>(); |
221 | 221 |
|
222 | 222 | String jsonTableRef = configRow.getString("json_table_ref"); |
223 | 223 | if (jsonTableRef != null) { |
@@ -378,9 +378,7 @@ public static class ReadRegistrar implements TransformPayloadTranslatorRegistrar |
378 | 378 | public Map<? extends Class<? extends PTransform>, ? extends TransformPayloadTranslator> |
379 | 379 | getTransformPayloadTranslators() { |
380 | 380 | return ImmutableMap.<Class<? extends PTransform>, TransformPayloadTranslator>builder() |
381 | | - .put( |
382 | | - BigQueryIO.read(BigQueryIO.TableRowParser.INSTANCE).getClass(), |
383 | | - new BigQueryIOReadTranslator()) |
| 381 | + .put(AutoValue_BigQueryIO_TypedRead.class, new BigQueryIOReadTranslator()) |
384 | 382 | .build(); |
385 | 383 | } |
386 | 384 | } |
@@ -617,7 +615,7 @@ public Write<?> fromConfigRow(Row configRow, PipelineOptions options) { |
617 | 615 | (updateCompatibilityBeamVersion != null) ? updateCompatibilityBeamVersion : "2.53.0"; |
618 | 616 |
|
619 | 617 | try { |
620 | | - BigQueryIO.Write.Builder builder = BigQueryIO.Write.builder(); |
| 618 | + BigQueryIO.Write.Builder builder = new AutoValue_BigQueryIO_Write.Builder<>(); |
621 | 619 |
|
622 | 620 | String jsonTableRef = configRow.getString("json_table_ref"); |
623 | 621 | if (jsonTableRef != null) { |
@@ -930,7 +928,7 @@ public static class WriteRegistrar implements TransformPayloadTranslatorRegistra |
930 | 928 | public Map<? extends Class<? extends PTransform>, ? extends TransformPayloadTranslator> |
931 | 929 | getTransformPayloadTranslators() { |
932 | 930 | return ImmutableMap.<Class<? extends PTransform>, TransformPayloadTranslator>builder() |
933 | | - .put(BigQueryIO.write().getClass(), new BigQueryIOWriteTranslator()) |
| 931 | + .put(AutoValue_BigQueryIO_Write.class, new BigQueryIOWriteTranslator()) |
934 | 932 | .build(); |
935 | 933 | } |
936 | 934 | } |
|
0 commit comments