Improve utils
parent
60f1fb1f70
commit
3b173dc6fc
GPG Key ID:
676DEE768C17A9D7
@ -1,37 +0,0 @@
|
|||||||
package org.koitharu.kotatsu.parsers.util
|
|
||||||
|
|
||||||
import kotlinx.coroutines.sync.Mutex
|
|
||||||
import kotlinx.coroutines.sync.withLock
|
|
||||||
import java.lang.ref.SoftReference
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Like a [SuspendLazy] but with [SoftReference] under the hood
|
|
||||||
*/
|
|
||||||
public class SoftSuspendLazy<T : Any>(
|
|
||||||
private val initializer: suspend () -> T,
|
|
||||||
) {
|
|
||||||
|
|
||||||
private val mutex = Mutex()
|
|
||||||
private var cachedValue: SoftReference<T>? = null
|
|
||||||
|
|
||||||
public suspend fun get(): T {
|
|
||||||
// fast way
|
|
||||||
cachedValue?.get()?.let {
|
|
||||||
return it
|
|
||||||
}
|
|
||||||
return mutex.withLock {
|
|
||||||
cachedValue?.get()?.let {
|
|
||||||
return it
|
|
||||||
}
|
|
||||||
val result = initializer()
|
|
||||||
cachedValue = SoftReference(result)
|
|
||||||
result
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
public suspend fun tryGet(): Result<T> = runCatchingCancellable { get() }
|
|
||||||
|
|
||||||
public fun peek(): T? {
|
|
||||||
return cachedValue?.get()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@ -1,41 +0,0 @@
|
|||||||
package org.koitharu.kotatsu.parsers.util
|
|
||||||
|
|
||||||
import kotlinx.coroutines.sync.Mutex
|
|
||||||
import kotlinx.coroutines.sync.withLock
|
|
||||||
|
|
||||||
public class SuspendLazy<T>(
|
|
||||||
private val initializer: suspend () -> T,
|
|
||||||
) {
|
|
||||||
|
|
||||||
private val mutex = Mutex()
|
|
||||||
private var cachedValue: Any? = Uninitialized
|
|
||||||
|
|
||||||
@Suppress("UNCHECKED_CAST")
|
|
||||||
public suspend fun get(): T {
|
|
||||||
// fast way
|
|
||||||
cachedValue.let {
|
|
||||||
if (it !== Uninitialized) {
|
|
||||||
return it as T
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return mutex.withLock {
|
|
||||||
cachedValue.let {
|
|
||||||
if (it !== Uninitialized) {
|
|
||||||
return it as T
|
|
||||||
}
|
|
||||||
}
|
|
||||||
val result = initializer()
|
|
||||||
cachedValue = result
|
|
||||||
result
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
public suspend fun tryGet(): Result<T> = runCatchingCancellable { get() }
|
|
||||||
|
|
||||||
@Suppress("UNCHECKED_CAST")
|
|
||||||
public fun peek(): T? {
|
|
||||||
return cachedValue?.takeUnless { it === Uninitialized } as T?
|
|
||||||
}
|
|
||||||
|
|
||||||
private object Uninitialized
|
|
||||||
}
|
|
||||||
@ -0,0 +1,41 @@
|
|||||||
|
package org.koitharu.kotatsu.parsers.util.suspendlazy
|
||||||
|
|
||||||
|
import kotlinx.coroutines.sync.Mutex
|
||||||
|
import kotlinx.coroutines.sync.withLock
|
||||||
|
import kotlinx.coroutines.withContext
|
||||||
|
import java.lang.ref.SoftReference
|
||||||
|
import kotlin.coroutines.CoroutineContext
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Like a [SuspendLazy] but with [SoftReference] under the hood
|
||||||
|
*/
|
||||||
|
internal class SoftSuspendLazyImpl<T : Any>(
|
||||||
|
private val coroutineContext: CoroutineContext,
|
||||||
|
private val initializer: SuspendLazyInitializer<T>,
|
||||||
|
) : SuspendLazy<T> {
|
||||||
|
|
||||||
|
private val mutex: Mutex = Mutex()
|
||||||
|
private var cachedValue: SoftReference<T>? = null
|
||||||
|
|
||||||
|
override val isInitialized: Boolean
|
||||||
|
get() = cachedValue?.get() != null
|
||||||
|
|
||||||
|
override suspend fun get(): T {
|
||||||
|
// fast way
|
||||||
|
cachedValue?.get()?.let {
|
||||||
|
return it
|
||||||
|
}
|
||||||
|
return mutex.withLock {
|
||||||
|
cachedValue?.get()?.let {
|
||||||
|
return it
|
||||||
|
}
|
||||||
|
val result = withContext(coroutineContext) {
|
||||||
|
initializer()
|
||||||
|
}
|
||||||
|
cachedValue = SoftReference(result)
|
||||||
|
result
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
override fun peek(): T? = cachedValue?.get()
|
||||||
|
}
|
||||||
@ -0,0 +1,33 @@
|
|||||||
|
package org.koitharu.kotatsu.parsers.util.suspendlazy
|
||||||
|
|
||||||
|
import org.koitharu.kotatsu.parsers.util.runCatchingCancellable
|
||||||
|
import kotlin.coroutines.CoroutineContext
|
||||||
|
import kotlin.coroutines.EmptyCoroutineContext
|
||||||
|
|
||||||
|
internal typealias SuspendLazyInitializer<T> = suspend () -> T
|
||||||
|
|
||||||
|
public interface SuspendLazy<T> {
|
||||||
|
|
||||||
|
public val isInitialized: Boolean
|
||||||
|
|
||||||
|
public suspend fun get(): T
|
||||||
|
|
||||||
|
public fun peek(): T?
|
||||||
|
}
|
||||||
|
|
||||||
|
public suspend fun <T> SuspendLazy<T>.getOrNull(): T? = runCatchingCancellable { get() }.getOrNull()
|
||||||
|
|
||||||
|
public fun <T> suspendLazy(
|
||||||
|
context: CoroutineContext = EmptyCoroutineContext,
|
||||||
|
initializer: SuspendLazyInitializer<T>,
|
||||||
|
): SuspendLazy<T> = SuspendLazyImpl(context, initializer)
|
||||||
|
|
||||||
|
public fun <T : Any> suspendLazy(
|
||||||
|
context: CoroutineContext = EmptyCoroutineContext,
|
||||||
|
soft: Boolean,
|
||||||
|
initializer: SuspendLazyInitializer<T>,
|
||||||
|
): SuspendLazy<T> = if (soft) {
|
||||||
|
SoftSuspendLazyImpl(context, initializer)
|
||||||
|
} else {
|
||||||
|
SuspendLazyImpl(context, initializer)
|
||||||
|
}
|
||||||
@ -0,0 +1,47 @@
|
|||||||
|
package org.koitharu.kotatsu.parsers.util.suspendlazy
|
||||||
|
|
||||||
|
import kotlinx.coroutines.sync.Mutex
|
||||||
|
import kotlinx.coroutines.sync.withLock
|
||||||
|
import kotlinx.coroutines.withContext
|
||||||
|
import kotlin.coroutines.CoroutineContext
|
||||||
|
|
||||||
|
internal class SuspendLazyImpl<T>(
|
||||||
|
private val coroutineContext: CoroutineContext,
|
||||||
|
private val initializer: SuspendLazyInitializer<T>,
|
||||||
|
) : SuspendLazy<T> {
|
||||||
|
|
||||||
|
private val mutex: Mutex = Mutex()
|
||||||
|
private var cachedValue: Any? = Uninitialized
|
||||||
|
|
||||||
|
override val isInitialized: Boolean
|
||||||
|
get() = cachedValue !== Uninitialized
|
||||||
|
|
||||||
|
@Suppress("UNCHECKED_CAST")
|
||||||
|
override suspend fun get(): T {
|
||||||
|
// fast way
|
||||||
|
cachedValue.let {
|
||||||
|
if (it !== Uninitialized) {
|
||||||
|
return it as T
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return mutex.withLock {
|
||||||
|
cachedValue.let {
|
||||||
|
if (it !== Uninitialized) {
|
||||||
|
return it as T
|
||||||
|
}
|
||||||
|
}
|
||||||
|
val result = withContext(coroutineContext) {
|
||||||
|
initializer()
|
||||||
|
}
|
||||||
|
cachedValue = result
|
||||||
|
result
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@Suppress("UNCHECKED_CAST")
|
||||||
|
override fun peek(): T? {
|
||||||
|
return cachedValue?.takeUnless { it === Uninitialized } as T?
|
||||||
|
}
|
||||||
|
|
||||||
|
private object Uninitialized
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue