private byte[] serializeAvroRecord(final GenericData.Record avroRecord) { final Map<String, String> props = new HashMap<>(); props.put("schema.registry.url", "localhost:9092"); final KafkaAvroSerializer avroSerializer = new KafkaAvroSerializer(schemaRegistryClient, props); return avroSerializer.serialize("topic", avroRecord); }
private GenericRow serializeDeserializeAvroRecord(final Schema schema, final String topicName, final SchemaRegistryClient schemaRegistryClient, final GenericRecord avroRecord) { final Map<String, Object> map = new HashMap<>(); map.put(AbstractKafkaAvroSerDeConfig.AUTO_REGISTER_SCHEMAS, true); map.put(AbstractKafkaAvroSerDeConfig.SCHEMA_REGISTRY_URL_CONFIG, ""); final KafkaAvroSerializer kafkaAvroSerializer = new KafkaAvroSerializer(schemaRegistryClient, map); final byte[] bytes = kafkaAvroSerializer.serialize(topicName, avroRecord); final Deserializer<GenericRow> deserializer = new KsqlAvroTopicSerDe(KsqlConstants.DEFAULT_AVRO_SCHEMA_FULL_NAME).getGenericRowSerde( schema, ksqlConfig, false, () -> schemaRegistryClient, "loggerName").deserializer(); return deserializer.deserialize(topicName, bytes); }
.set("timestamp", 1467176316L) .build(); GenericRecord output = secorSchemaRegistryClient.decodeMessage("test-avr-topic", avroSerializer.serialize("test-avr-topic", record1)); assertEquals(secorSchemaRegistryClient.getSchema("test-avr-topic"), schemaV1); assertEquals(output.get("data_field_1"), 1); assertEquals(output.get("timestamp"), 1467176315L); output = secorSchemaRegistryClient.decodeMessage("test-avr-topic", avroSerializer.serialize("test-avr-topic", record2)); assertEquals(secorSchemaRegistryClient.getSchema("test-avr-topic"), schemaV2); assertEquals(output.get("data_field_1"), 1);
@Override public byte[] serialize(String topic, T record) { return inner.serialize(topic, record); }
@Override public byte[] serialize(String topic, GenericRecord record) { return inner.serialize(topic, record); }
@Test public void testConfluentSerDes() throws Exception { org.apache.avro.Schema schema = new org.apache.avro.Schema.Parser().parse(GENERIC_TEST_RECORD_SCHEMA); GenericRecord record = new GenericRecordBuilder(schema).set("field1", "some value").set("field2", "some other value").build(); Map<String, Object> config = new HashMap<>(); config.put(AbstractKafkaAvroSerDeConfig.SCHEMA_REGISTRY_URL_CONFIG, rootTarget.getUri().toString()); KafkaAvroSerializer kafkaAvroSerializer = new KafkaAvroSerializer(); kafkaAvroSerializer.configure(config, false); byte[] bytes = kafkaAvroSerializer.serialize("topic", record); KafkaAvroDeserializer kafkaAvroDeserializer = new KafkaAvroDeserializer(); kafkaAvroDeserializer.configure(config, false); GenericRecord result = (GenericRecord) kafkaAvroDeserializer.deserialize("topic", bytes); LOG.info(result.toString()); }
byte[] bytes = kafkaAvroSerializer.serialize("topic", record);
@Test public void testAvroDecoderFailure() { String topic = "testAvro"; Object avroRecordV1 = createAvroRecordVersion1(); byte[] payloadV1 = avroSerializer.serialize(topic, avroRecordV1); AvroMessageDecoder decoder = createAvroDecoder(topic, true, schemaRegistry); decoder.decode(payloadV1); Object avroRecordV2 = createAvroRecordVersion2(); byte[] payloadV2 = avroSerializer.serialize(topic, avroRecordV2); try { decoder.decode(payloadV2); fail("AvroMessageDecoder should not be able to decode Avro record with new schema version"); } catch (MessageDecoderException e) { assertEquals(e.getMessage(), "Producer schema is newer than the schema known to Camus"); } } }
@Test public void testAvroDecoderCompatible() { String topic = "testAvro"; Object avroRecordV1 = createAvroRecordVersion1(); byte[] payloadV1 = avroSerializer.serialize(topic, avroRecordV1); Object avroRecordV2 = createAvroRecordVersion2(); byte[] payloadV2 = avroSerializer.serialize(topic, avroRecordV2); AvroMessageDecoder decoder = createAvroDecoder(topic, true, schemaRegistry); try { decoder.decode(payloadV1).getRecord(); } catch (MessageDecoderException e) { fail("Backward compatible schema should be able to decode Avro records with old schema"); } Object recordV2 = decoder.decode(payloadV2).getRecord(); assertEquals(avroRecordV2, recordV2); }
@Test public void testAvroDecoder() { String topic = "testAvro"; Object avroRecord = createAvroRecordVersion1(); byte[] payload = avroSerializer.serialize(topic, avroRecord); AvroMessageDecoder decoder = createAvroDecoder(topic, true, schemaRegistry); Object record = decoder.decode(payload).getRecord(); assertEquals(avroRecord, record); payload = avroEncoder.toBytes(avroRecord); AvroMessageDecoder decoder2 = createAvroDecoder(topic, false, schemaRegistry); record = decoder2.decode(payload).getRecord(); assertEquals(avroRecord, record); }