main.com.connectrpc.extensions.GoogleJavaLiteProtobufStrategy.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of connect-kotlin-google-javalite-ext Show documentation

Simple, reliable, interoperable. A better RPC.

There is a newer version: 0.7.1

Show newest version

// Copyright 2022-2023 The Connect 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
//
//      http://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 com.connectrpc.extensions

import com.connectrpc.CODEC_NAME_PROTO
import com.connectrpc.Codec
import com.connectrpc.ErrorDetailParser
import com.connectrpc.SerializationStrategy
import com.google.protobuf.ExtensionRegistryLite
import com.google.protobuf.MessageLite
import kotlin.reflect.KClass
import kotlin.reflect.full.isSubclassOf

/**
 * The Google Java-lite Protobuf serialization strategy.
 */
class GoogleJavaLiteProtobufStrategy(
    private val registry: ExtensionRegistryLite = ExtensionRegistryLite.getEmptyRegistry(),
) : SerializationStrategy {
    override fun serializationName(): String {
        return CODEC_NAME_PROTO
    }

    override fun  codec(clazz: KClass): Codec {
        if (!clazz.isSubclassOf(MessageLite::class)) {
            throw RuntimeException("class ${clazz.qualifiedName} does not extend MessageLite")
        }
        @Suppress("UNCHECKED_CAST") // we just checked above, so it's safe
        val messageClass = clazz as KClass

        @Suppress("UNCHECKED_CAST") // messageClass is actually KClass, so it's safe
        val adapter = GoogleLiteProtoAdapter(messageClass, registry) as Codec
        return adapter
    }

    override fun errorDetailParser(): ErrorDetailParser {
        return JavaLiteErrorParser
    }
}