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

Show more of this group Show more artifacts with this name
Show all versions of sentry-kotlin-extensions Show documentation

SDK for sentry.io

There is a newer version: 8.0.0-alpha.4

Show newest version

package io.sentry.kotlin

import io.sentry.IHub
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 hub: IHub = Sentry.getCurrentHub().clone()) :
    CopyableThreadContextElement, AbstractCoroutineContextElement(Key) {

    private companion object Key : CoroutineContext.Key

    override fun copyForChild(): CopyableThreadContextElement {
        return SentryContext(hub.clone())
    }

    override fun mergeForChild(overwritingElement: CoroutineContext.Element): CoroutineContext {
        return overwritingElement[Key] ?: SentryContext(hub.clone())
    }

    override fun updateThreadContext(context: CoroutineContext): IHub {
        val oldState = Sentry.getCurrentHub()
        Sentry.setCurrentHub(hub)
        return oldState
    }

    override fun restoreThreadContext(context: CoroutineContext, oldState: IHub) {
        Sentry.setCurrentHub(oldState)
    }
}