feat: support registry for primitive types (#1)

leboncoin · Nov 21, 2024 · 837e337 · 837e337
1 parent 9804442
commit 837e337
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/...pache/flink/formats/avro/registry/confluent/ConfluentRegistryAvroSerializationSchema.java b/...pache/flink/formats/avro/registry/confluent/ConfluentRegistryAvroSerializationSchema.java
@@ -140,4 +140,19 @@ public static ConfluentRegistryAvroSerializationSchema<GenericRecord> forGeneric
                         DEFAULT_IDENTITY_MAP_CAPACITY,
                         registryConfigs));
     }
+
+    public static ConfluentRegistryAvroSerializationSchema<Object> forPrimitiveType(
+            String subject,
+            Schema schema,
+            String schemaRegistryUrl,
+            @Nullable Map<String, ?> registryConfigs) {
+        return new ConfluentRegistryAvroSerializationSchema<>(
+                Object.class,
+                schema,
+                new CachedSchemaCoderProvider(
+                        subject,
+                        schemaRegistryUrl,
+                        DEFAULT_IDENTITY_MAP_CAPACITY,
+                        registryConfigs));
+    }
 }