meh, make it thread safe

open-toast · Jun 12, 2024 · 37f9638 · 37f9638
1 parent 1421ef4
commit 37f9638
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/protokt-protovalidate/src/main/kotlin/protokt/v1/buf/validate/Validator.kt b/protokt-protovalidate/src/main/kotlin/protokt/v1/buf/validate/Validator.kt
@@ -29,6 +29,8 @@ import protokt.v1.GeneratedMessage
 import protokt.v1.Message
 import protokt.v1.google.protobuf.RuntimeContext
 import protokt.v1.google.protobuf.toDynamicMessage
+import java.util.Collections
+import java.util.concurrent.ConcurrentHashMap
 import kotlin.reflect.full.findAnnotation
 
 @Beta
@@ -43,9 +45,10 @@ class Validator @JvmOverloads constructor(
 
     private val failFast = config.isFailFast
 
-    private val evaluatorsByFullTypeName = mutableMapOf<String, Evaluator>()
-    private val descriptors = mutableSetOf<Descriptor>()
+    private val evaluatorsByFullTypeName = ConcurrentHashMap<String, Evaluator>()
+    private val descriptors = Collections.newSetFromMap(ConcurrentHashMap<Descriptor, Boolean>())
 
+    @Volatile
     private var runtimeContext = RuntimeContext(emptyList())
 
     fun load(descriptor: Descriptor) {