io.micronaut.pulsar.schemas.json.JsonSchema Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2017-2022 original authors
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * https://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package io.micronaut.pulsar.schemas.json;

import io.micronaut.json.JsonMapper;
import org.apache.pulsar.client.api.schema.SchemaDefinition;
import org.apache.pulsar.client.api.schema.SchemaReader;
import org.apache.pulsar.client.api.schema.SchemaWriter;
import org.apache.pulsar.client.impl.schema.AvroBaseStructSchema;
import org.apache.pulsar.client.impl.schema.SchemaDefinitionBuilderImpl;
import org.apache.pulsar.client.impl.schema.util.SchemaUtil;
import org.apache.pulsar.common.schema.SchemaInfo;
import org.apache.pulsar.common.schema.SchemaType;

import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;

/**
 * JSON Schema to allow using {@link JsonMapper} from Micronaut instead of shaded one in Pulsar library.
 *
 * @param  POJO type to send and receive via Pulsar.
 * @author Haris Secic
 * @since 1.0
 */
public class JsonSchema extends AvroBaseStructSchema {

    private static final Map> SCHEMAS = new ConcurrentHashMap<>(10);

    public JsonSchema(SchemaInfo schemaInfo, SchemaReader reader, SchemaWriter writer) {
        super(schemaInfo);
        this.setWriter(writer);
        this.setReader(reader);
    }

    /**
     * @param pojo       The pojo class to map
     * @param jsonMapper The json mapper to use for mapping
     * @param         The pojo class to map
     * @return The parsed json schema
     * @since 1.1.0
     */
    @SuppressWarnings("unchecked")
    public static  JsonSchema of(Class pojo, JsonMapper jsonMapper) {
        return (JsonSchema) SCHEMAS.computeIfAbsent(pojo.hashCode(), x -> {
            final SchemaReader reader = new JsonReader<>(jsonMapper, pojo);
            final SchemaWriter writer = new JsonWriter<>(jsonMapper);
            final SchemaDefinition schemaDefinition = new SchemaDefinitionBuilderImpl().withPojo(pojo)
                .withSchemaReader(reader)
                .withSchemaWriter(writer)
                .build();
            return new JsonSchema<>(SchemaUtil.parseSchemaInfo(schemaDefinition, SchemaType.JSON), reader, writer);
        });
    }
}