Examples with SchemaReference org.springframework.cloud.stream.schema.SchemaReference used on opensource projects

Example 1 with SchemaReference

use of org.springframework.cloud.stream.schema.SchemaReference in project spring-cloud-stream by spring-cloud.

the class AvroMessageConverterSerializationTests method extractSchemaReference.

private SchemaReference extractSchemaReference(MimeType mimeType) {
    SchemaReference schemaReference = null;
    Matcher schemaMatcher = this.versionedSchema.matcher(mimeType.toString());
    if (schemaMatcher.find()) {
        String subject = schemaMatcher.group(1);
        Integer version = Integer.parseInt(schemaMatcher.group(2));
        schemaReference = new SchemaReference(subject, version, AvroSchemaRegistryClientMessageConverter.AVRO_FORMAT);
    }
    return schemaReference;
}

Example 2 with SchemaReference

use of org.springframework.cloud.stream.schema.SchemaReference in project spring-cloud-stream by spring-cloud.

the class AvroMessageConverterSerializationTests method sourceWriteSameVersion.

@Test
public void sourceWriteSameVersion() throws Exception {
    User specificRecord = new User();
    specificRecord.setName("joe");
    Schema v1 = new Schema.Parser().parse(AvroMessageConverterSerializationTests.class.getClassLoader().getResourceAsStream("schemas/user.avsc"));
    GenericRecord genericRecord = new GenericData.Record(v1);
    genericRecord.put("name", "joe");
    SchemaRegistryClient client = new DefaultSchemaRegistryClient();
    AvroSchemaRegistryClientMessageConverter converter = new AvroSchemaRegistryClientMessageConverter(client, new NoOpCacheManager());
    converter.setSubjectNamingStrategy(new DefaultSubjectNamingStrategy());
    converter.setDynamicSchemaGenerationEnabled(false);
    converter.afterPropertiesSet();
    Message specificMessage = converter.toMessage(specificRecord, new MutableMessageHeaders(Collections.<String, Object>emptyMap()), MimeTypeUtils.parseMimeType("application/*+avro"));
    SchemaReference specificRef = extractSchemaReference(MimeTypeUtils.parseMimeType(specificMessage.getHeaders().get("contentType").toString()));
    Message genericMessage = converter.toMessage(genericRecord, new MutableMessageHeaders(Collections.<String, Object>emptyMap()), MimeTypeUtils.parseMimeType("application/*+avro"));
    SchemaReference genericRef = extractSchemaReference(MimeTypeUtils.parseMimeType(genericMessage.getHeaders().get("contentType").toString()));
    Assert.assertEquals(genericRef, specificRef);
    Assert.assertEquals(1, genericRef.getVersion());
}

Example 3 with SchemaReference

use of org.springframework.cloud.stream.schema.SchemaReference in project spring-cloud-stream by spring-cloud.

the class AvroSchemaRegistryClientMessageConverter method resolveSchemaForWriting.

@Override
protected Schema resolveSchemaForWriting(Object payload, MessageHeaders headers, MimeType hintedContentType) {
    Schema schema;
    schema = extractSchemaForWriting(payload);
    ParsedSchema parsedSchema = this.cacheManager.getCache(REFERENCE_CACHE_NAME).get(schema, ParsedSchema.class);
    if (parsedSchema == null) {
        parsedSchema = new ParsedSchema(schema);
        this.cacheManager.getCache(REFERENCE_CACHE_NAME).putIfAbsent(schema, parsedSchema);
    }
    if (parsedSchema.getRegistration() == null) {
        SchemaRegistrationResponse response = this.schemaRegistryClient.register(toSubject(schema), AVRO_FORMAT, parsedSchema.getRepresentation());
        parsedSchema.setRegistration(response);
    }
    SchemaReference schemaReference = parsedSchema.getRegistration().getSchemaReference();
    DirectFieldAccessor dfa = new DirectFieldAccessor(headers);
    @SuppressWarnings("unchecked") Map<String, Object> _headers = (Map<String, Object>) dfa.getPropertyValue("headers");
    _headers.put(MessageHeaders.CONTENT_TYPE, "application/" + this.prefix + "." + schemaReference.getSubject() + ".v" + schemaReference.getVersion() + "+" + AVRO_FORMAT);
    return schema;
}

Example 4 with SchemaReference

use of org.springframework.cloud.stream.schema.SchemaReference in project spring-cloud-stream by spring-cloud.

the class AvroSchemaRegistryClientMessageConverter method resolveWriterSchemaForDeserialization.

@Override
protected Schema resolveWriterSchemaForDeserialization(MimeType mimeType) {
    if (this.readerSchema == null) {
        Schema schema = null;
        ParsedSchema parsedSchema = null;
        SchemaReference schemaReference = extractSchemaReference(mimeType);
        if (schemaReference != null) {
            parsedSchema = cacheManager.getCache(REFERENCE_CACHE_NAME).get(schemaReference, ParsedSchema.class);
            if (parsedSchema == null) {
                String schemaContent = this.schemaRegistryClient.fetch(schemaReference);
                schema = new Schema.Parser().parse(schemaContent);
                parsedSchema = new ParsedSchema(schema);
                cacheManager.getCache(REFERENCE_CACHE_NAME).putIfAbsent(schemaReference, parsedSchema);
            }
        }
        return parsedSchema.getSchema();
    } else {
        return this.readerSchema;
    }
}

Example 5 with SchemaReference

use of org.springframework.cloud.stream.schema.SchemaReference in project spring-cloud-stream by spring-cloud.

the class DefaultSchemaRegistryClient method register.

@SuppressWarnings({ "rawtypes", "unchecked" })
@Override
public SchemaRegistrationResponse register(String subject, String format, String schema) {
    Map<String, String> requestBody = new HashMap<>();
    requestBody.put("subject", subject);
    requestBody.put("format", format);
    requestBody.put("definition", schema);
    ResponseEntity<Map> responseEntity = this.restTemplate.postForEntity(this.endpoint, requestBody, Map.class);
    if (responseEntity.getStatusCode().is2xxSuccessful()) {
        SchemaRegistrationResponse registrationResponse = new SchemaRegistrationResponse();
        Map<String, Object> responseBody = (Map<String, Object>) responseEntity.getBody();
        registrationResponse.setId((Integer) responseBody.get("id"));
        registrationResponse.setSchemaReference(new SchemaReference(subject, (Integer) responseBody.get("version"), responseBody.get("format").toString()));
        return registrationResponse;
    }
    throw new RuntimeException("Failed to register schema: " + responseEntity.toString());
}