io.sentry.kotlin.SentryContext.kt Maven / Gradle / Ivy

Go to download
package io.sentry.kotlin

import io.sentry.IScopes
import io.sentry.Sentry
import kotlinx.coroutines.CopyableThreadContextElement
import kotlin.coroutines.AbstractCoroutineContextElement
import kotlin.coroutines.CoroutineContext

/**
 * Sentry context element for [CoroutineContext].
 */
public class SentryContext(private val scopes: IScopes = Sentry.forkedCurrentScope("coroutine")) :
    CopyableThreadContextElement, AbstractCoroutineContextElement(Key) {

    private companion object Key : CoroutineContext.Key

    @SuppressWarnings("deprecation")
    override fun copyForChild(): CopyableThreadContextElement {
        return SentryContext(scopes.forkedCurrentScope("coroutine.child"))
    }

    @SuppressWarnings("deprecation")
    override fun mergeForChild(overwritingElement: CoroutineContext.Element): CoroutineContext {
        return overwritingElement[Key] ?: SentryContext(scopes.forkedCurrentScope("coroutine.child"))
    }

    override fun updateThreadContext(context: CoroutineContext): IScopes {
        val oldState = Sentry.getCurrentScopes()
        Sentry.setCurrentScopes(scopes)
        return oldState
    }

    override fun restoreThreadContext(context: CoroutineContext, oldState: IScopes) {
        Sentry.setCurrentScopes(oldState)
    }
}