ENT-9837 Use a cache to avoid resolving contract class name for a contract state repeatedly. (#7348)

core-deterministic/build.gradle

@@ -82,6 +82,7 @@ def patchCore = tasks.register('patchCore', Zip) {
         exclude 'net/corda/core/serialization/internal/CheckpointSerializationFactory*.class'
         exclude 'net/corda/core/internal/rules/*.class'
         exclude 'net/corda/core/internal/utilities/PrivateInterner*.class'
+        exclude 'net/corda/core/internal/ContractStateClassCache*.class'
     }
 
     reproducibleFileOrder = true

core-deterministic/src/main/kotlin/net/corda/core/internal/ContractStateClassCache.kt

@@ -0,0 +1,16 @@
+package net.corda.core.internal
+
+import net.corda.core.contracts.ContractState
+
+@Suppress("unused")
+object ContractStateClassCache {
+    @Suppress("UNUSED_PARAMETER")
+    fun contractClassName(key: Class<ContractState>): String? {
+        return null
+    }
+
+    @Suppress("UNUSED_PARAMETER")
+    fun cacheContractClassName(key: Class<ContractState>, contractClassName: String?): String? {
+        return contractClassName
+    }
+}

core/src/main/kotlin/net/corda/core/internal/ConstraintsUtils.kt

@@ -28,15 +28,19 @@ val Attachment.contractVersion: Version get() = if (this is ContractAttachment)
  * one and it inherits from [Contract].
  */
 val ContractState.requiredContractClassName: String? get() {
-    val annotation = javaClass.getAnnotation(BelongsToContract::class.java)
-    if (annotation != null) {
-        return annotation.value.java.typeName.removePrefix(DJVM_SANDBOX_PREFIX)
-    }
-    val enclosingClass = javaClass.enclosingClass ?: return null
-    return if (Contract::class.java.isAssignableFrom(enclosingClass)) {
-        enclosingClass.typeName.removePrefix(DJVM_SANDBOX_PREFIX)
-    } else {
-        null
+    return ContractStateClassCache.contractClassName(this.javaClass) ?: let {
+        val annotation = javaClass.getAnnotation(BelongsToContract::class.java)
+        val className = if (annotation != null) {
+            annotation.value.java.typeName.removePrefix(DJVM_SANDBOX_PREFIX)
+        } else {
+            val enclosingClass = javaClass.enclosingClass ?: return null
+            if (Contract::class.java.isAssignableFrom(enclosingClass)) {
+                enclosingClass.typeName.removePrefix(DJVM_SANDBOX_PREFIX)
+            } else {
+                null
+            }
+        }
+        ContractStateClassCache.cacheContractClassName(this.javaClass, className)
     }
 }
 

core/src/main/kotlin/net/corda/core/internal/ContractStateClassCache.kt

@@ -0,0 +1,18 @@
+package net.corda.core.internal
+
+import com.google.common.collect.MapMaker
+import net.corda.core.contracts.ContractState
+
+object ContractStateClassCache {
+    private val classToString = MapMaker().weakKeys().makeMap<Class<*>, String>()
+
+    fun contractClassName(key: Class<ContractState>): String? {
+        return classToString[key]
+    }
+
+    fun cacheContractClassName(key: Class<ContractState>, contractClassName: String?): String? {
+        if (contractClassName == null) return null
+        classToString.putIfAbsent(key, contractClassName)
+        return contractClassName
+    }
+}