commonMain.in.shabinder.soundbound.diagnostics.DiagnosticContext.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of soundbound-extensions-lib Show documentation

SoundBound Extensions Stub

The newest version!

package `in`.shabinder.soundbound.diagnostics

import `in`.shabinder.soundbound.models.ProviderExceptions
import `in`.shabinder.soundbound.models.toThrowableWrapper
import kotlin.coroutines.CoroutineContext
import kotlin.coroutines.coroutineContext
import kotlinx.coroutines.withContext

// DiagnosticContext: A CoroutineContext Element to manage DiagnosticHelper
class DiagnosticContext private constructor(val manager: DiagnosticManager) :
  CoroutineContext.Element {

  override val key: CoroutineContext.Key get() = Key

  companion object Key : CoroutineContext.Key {
    suspend fun  withDiagnostics(block: suspend DiagnosticManager.() -> T): T {
      val currentContext = coroutineContext[Key]
      return if (currentContext != null) {
        currentContext.manager.tryOrThrowWithDiagnostics { block() }
      } else {
        val newDiagnostics = DiagnosticManager()
        val newContext = DiagnosticContext(newDiagnostics)
        withContext(coroutineContext + newContext) {
          newDiagnostics.tryOrThrowWithDiagnostics { block() }
        }
      }
    }

    suspend fun getDiagnosticsContext(): DiagnosticManager {
      val currentContext = coroutineContext[Key]
      return currentContext?.manager ?: throw IllegalStateException("No DiagnosticContext found.")
    }

    private inline fun  DiagnosticManager.tryOrThrowWithDiagnostics(
      block: DiagnosticManager.() -> T
    ): T = try {
      block()
    } catch (e: Throwable) {
      if (e is ProviderExceptions.ExceptionWithDiagnostics) {
        // Already has diagnostics, rethrow
        throw e
      }

      appendDiagnostics("ErrorCaughtWithDiagnostics", e)
      throw ProviderExceptions.ExceptionWithDiagnostics(
        cause = e.toThrowableWrapper(),
        diagnostics = getDiagnostics()
      )
    }
  }
}