io.micronaut.pulsar.schemas.protobuf.ProtobufSchema 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.protobuf;

import io.micronaut.core.type.Argument;
import io.micronaut.protobuf.codec.ProtobufferCodec;
import org.apache.pulsar.client.api.schema.SchemaDefinition;
import org.apache.pulsar.client.impl.schema.AbstractStructSchema;
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 ProtobufferCodec} from Micronaut.
 *
 * @param  POJO type to send and receive via Pulsar.
 * @author Haris Secic
 * @since 1.0
 */
public final class ProtobufSchema extends AbstractStructSchema {

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

    public ProtobufSchema(final SchemaInfo schemaInfo, final ProtobufReader reader, final ProtobufWriter writer) {
        super(schemaInfo);
        this.setReader(reader);
        this.setWriter(writer);
    }

    @SuppressWarnings("unchecked")
    public static  ProtobufSchema of(final Class type, final ProtobufferCodec codec) {
        return (ProtobufSchema) SCHEMAS.computeIfAbsent(type.hashCode(), x -> {
            final ProtobufWriter writer = new ProtobufWriter<>(codec);
            final ProtobufReader reader = new ProtobufReader<>(codec, Argument.of(type));
            final SchemaDefinition schemaDefinition = new SchemaDefinitionBuilderImpl()
                    .withPojo(type)
                    .withSchemaWriter(writer)
                    .withSchemaReader(reader)
                    .build();
            return new ProtobufSchema<>(SchemaUtil.parseSchemaInfo(schemaDefinition, SchemaType.PROTOBUF), reader, writer);
        });
    }
}