IR: threadLocal
To be used for per-file state in global structures during parallel lowering.
This commit is contained in:
@@ -0,0 +1,26 @@
|
||||
/*
|
||||
* Copyright 2010-2020 JetBrains s.r.o. and Kotlin Programming Language contributors.
|
||||
* Use of this source code is governed by the Apache 2.0 license that can be found in the license/LICENSE.txt file.
|
||||
*/
|
||||
|
||||
package org.jetbrains.kotlin.utils
|
||||
|
||||
import java.util.concurrent.ConcurrentHashMap
|
||||
import kotlin.properties.ReadWriteProperty
|
||||
import kotlin.reflect.KProperty
|
||||
|
||||
fun <T> threadLocal(initializer: () -> T): ReadWriteProperty<Any?, T> = ThreadLocalDelegate(initializer)
|
||||
|
||||
private class ThreadLocalDelegate<T>(private val initializer: () -> T) : ReadWriteProperty<Any?, T> {
|
||||
private val map = ConcurrentHashMap<Thread, T>()
|
||||
|
||||
override operator fun getValue(thisRef: Any?, property: KProperty<*>): T {
|
||||
return map.getOrPut(Thread.currentThread()) {
|
||||
initializer()
|
||||
}
|
||||
}
|
||||
|
||||
override operator fun setValue(thisRef: Any?, property: KProperty<*>, value: T) {
|
||||
map[Thread.currentThread()] = value
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user