io.github.majusko.pulsar2.solon.utils.SchemaUtils Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of pulsar2-solon-plugin Show documentation

There is a newer version: 3.0.5-M3

package io.github.majusko.pulsar2.solon.utils;

import com.google.protobuf.GeneratedMessageV3;

import io.github.majusko.pulsar2.solon.constant.Serialization;
import io.github.majusko.pulsar2.solon.error.exception.ProducerInitException;

import org.apache.pulsar.client.api.Schema;

import java.lang.reflect.Method;

public class SchemaUtils {

    private SchemaUtils() {
    }

    private static  Schema getGenericSchema(Serialization serialization, Class clazz) throws RuntimeException {
        switch (serialization) {
            case JSON: {
                return Schema.JSON(clazz);
            }
            case AVRO: {
                return Schema.AVRO(clazz);
            }
            case STRING: {
                return Schema.STRING;
            }
            default: {
                throw new ProducerInitException("Unknown producer schema.");
            }
        }
    }

    private static  Schema getProtoSchema(Serialization serialization, Class clazz) throws RuntimeException {
        if (serialization == Serialization.PROTOBUF) {
            return Schema.PROTOBUF(clazz);
        }
        throw new ProducerInitException("Unknown producer schema.");
    }

    public static Schema getSchema(Serialization serialisation, Class clazz) {
        if (clazz == byte[].class) {
            return Schema.BYTES;
        }

        if (isProto(serialisation)) {
            return getProtoSchema(serialisation, (Class) clazz);
        }

        return getGenericSchema(serialisation, clazz);
    }

    public static boolean isProto(Serialization serialization) {
        return serialization == Serialization.PROTOBUF;
    }

    public static Class getParameterType(Method method) {
        return method.getParameterTypes()[0];
    }

}