52 lines
1.7 KiB
Kotlin
52 lines
1.7 KiB
Kotlin
package cash.z.ecc.android.sdk.internal.jni
|
|
|
|
import kotlinx.coroutines.Dispatchers
|
|
import kotlinx.coroutines.sync.Mutex
|
|
import kotlinx.coroutines.sync.withLock
|
|
import kotlinx.coroutines.withContext
|
|
import java.util.concurrent.atomic.AtomicBoolean
|
|
|
|
/**
|
|
* Loads a native library once. This class is thread-safe.
|
|
*
|
|
* To use this class, create a singleton instance for each given [libraryName].
|
|
*
|
|
* @param libraryName Name of the library to load.
|
|
*/
|
|
internal class NativeLibraryLoader(private val libraryName: String) {
|
|
private val isLoaded = AtomicBoolean(false)
|
|
private val mutex = Mutex()
|
|
|
|
/**
|
|
* @param onLoad Callback allowing additional initialization after the native library is loaded. This will only be
|
|
* invoked the first time this method is called for the lifetime of the process.
|
|
*/
|
|
suspend fun load(onLoad: suspend () -> Unit) {
|
|
// Double-checked locking to avoid the Mutex unless necessary, as the hot path is
|
|
// for the library to be loaded since this should only run once for the lifetime
|
|
// of the application
|
|
if (!isLoaded.get()) {
|
|
mutex.withLock {
|
|
if (!isLoaded.get()) {
|
|
loadNativeLibrary()
|
|
onLoad()
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
private suspend fun loadNativeLibrary() {
|
|
runCatching {
|
|
loadLibrarySuspend(libraryName)
|
|
isLoaded.set(true)
|
|
}.onFailure {
|
|
// Fail fast, because this is not a recoverable error
|
|
throw AssertionError("Failed loading native library $libraryName", it)
|
|
}
|
|
}
|
|
}
|
|
|
|
private suspend fun loadLibrarySuspend(libraryName: String) = withContext(Dispatchers.IO) {
|
|
System.loadLibrary(libraryName)
|
|
}
|