FakeSpan.kt

package io.embrace.android.embracesdk.fakes

import io.embrace.android.embracesdk.internal.otel.sdk.id.OtelIds
import io.embrace.android.embracesdk.internal.otel.spans.getEmbraceSpan
import io.opentelemetry.api.common.AttributeKey
import io.opentelemetry.api.common.Attributes
import io.opentelemetry.api.trace.Span
import io.opentelemetry.api.trace.SpanContext
import io.opentelemetry.api.trace.StatusCode
import io.opentelemetry.api.trace.TraceFlags
import io.opentelemetry.api.trace.TraceState
import java.util.concurrent.TimeUnit

class FakeSpan(
    val fakeSpanBuilder: FakeSpanBuilder,
) : Span {

    private val spanContext: SpanContext =
        SpanContext.create(
            fakeSpanBuilder.parentContext.getEmbraceSpan()?.traceId ?: OtelIds.generateTraceId(),
            OtelIds.generateSpanId(),
            TraceFlags.getDefault(),
            TraceState.getDefault()
        )

    private var isRecording = true
    private var status: StatusCode = StatusCode.UNSET
    private var statusDescription: String = ""

    override fun <T : Any> setAttribute(key: AttributeKey<T>, value: T?): Span {
        if (value != null) {
            fakeSpanBuilder.setAttribute(key, value)
        }

        return this
    }

    override fun addEvent(name: String, attributes: Attributes): Span {
        TODO("Not yet implemented")
    }

    override fun addEvent(name: String, attributes: Attributes, timestamp: Long, unit: TimeUnit): Span {
        TODO("Not yet implemented")
    }

    override fun setStatus(statusCode: StatusCode, description: String): Span {
        status = statusCode
        statusDescription = description
        return this
    }

    override fun recordException(exception: Throwable, additionalAttributes: Attributes): Span {
        TODO("Not yet implemented")
    }

    override fun updateName(name: String): Span {
        TODO("Not yet implemented")
    }

    override fun end() {
        isRecording = false
    }

    override fun end(timestamp: Long, unit: TimeUnit): Unit = end()

    override fun getSpanContext(): SpanContext = spanContext

    override fun isRecording(): Boolean = isRecording
}