org.gradle.configurationcache.serialization.codecs.jos.JavaSerializationEncodingLookup.kt Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2021 the original author or 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 org.gradle.configurationcache.serialization.codecs.jos

import org.gradle.configurationcache.serialization.codecs.Encoding
import org.gradle.internal.service.scopes.Scopes
import org.gradle.internal.service.scopes.ServiceScope
import java.io.ObjectOutputStream
import java.lang.reflect.Method
import java.lang.reflect.Modifier
import java.util.concurrent.ConcurrentHashMap


@ServiceScope(Scopes.BuildTree::class)
internal
class JavaSerializationEncodingLookup {
    private
    val encodings = ConcurrentHashMap, EncodingDetails>()

    fun encodingFor(type: Class<*>): Encoding? {
        return encodings.computeIfAbsent(type) { t -> calculateEncoding(t) }.encoding
    }

    private
    fun calculateEncoding(type: Class<*>): EncodingDetails {
        val candidates = type.allMethods()
        val encoding = writeReplaceEncodingFor(candidates)
            ?: readResolveEncodingFor(candidates)
            ?: writeObjectEncodingFor(candidates)
            ?: readObjectEncodingFor(candidates)
        return EncodingDetails(encoding)
    }

    private
    fun writeReplaceEncodingFor(candidates: List) =
        writeReplaceMethodFrom(candidates)
            ?.let(JavaObjectSerializationCodec::WriteReplaceEncoding)

    private
    fun readResolveEncodingFor(candidates: List) =
        readResolveMethodFrom(candidates)
            ?.let { JavaObjectSerializationCodec.ReadResolveEncoding }

    private
    fun writeObjectEncodingFor(candidates: List): Encoding? =
        writeObjectMethodHierarchyFrom(candidates)
            .takeIf { it.isNotEmpty() }
            ?.let(JavaObjectSerializationCodec::WriteObjectEncoding)

    private
    fun readObjectEncodingFor(candidates: List): Encoding? =
        readObjectMethodHierarchyFrom(candidates)
            .takeIf { it.isNotEmpty() }
            ?.let { JavaObjectSerializationCodec.ReadObjectEncoding }

    private
    fun writeReplaceMethodFrom(candidates: List) =
        candidates.firstAccessibleMatchingMethodOrNull {
            !Modifier.isStatic(modifiers)
                && parameterCount == 0
                && returnType == java.lang.Object::class.java
                && name == "writeReplace"
        }

    private
    fun writeObjectMethodHierarchyFrom(candidates: List) = candidates
        .serializationMethodHierarchy("writeObject", ObjectOutputStream::class.java)

    private
    fun readResolveMethodFrom(candidates: List) =
        candidates.find {
            it.isReadResolve()
        }

    private
    class EncodingDetails(
        val encoding: Encoding?
    )
}