spans/scripting.go - master - githubmirror/ydb-go-sdk

spans/scripting.go
104 строки2 KB
Aleksey Myasnikov
Harden trace handlers (spans/log/metrics) and add fuzz tests (#2194)
07 июн 2026, 00:54
07 июн 2026, 00:54c974edf
package spans

import (
	"github.com/ydb-platform/ydb-go-sdk/v3/internal/kv"
	"github.com/ydb-platform/ydb-go-sdk/v3/trace"
)

//nolint:funlen
func scripting(adapter Adapter) (t trace.Scripting) {
	t.OnExecute = func(info trace.ScriptingExecuteStartInfo) func(trace.ScriptingExecuteDoneInfo) {
		if adapter.Details()&trace.ScriptingEvents == 0 {
			return nil
		}

		start := childSpanWithReplaceCtx(
			adapter,
			info.Context,
			safeCall(info.Call),
			kv.String("query", info.Query),
		)

		return func(info trace.ScriptingExecuteDoneInfo) {
			if info.Error == nil {
				finish(
					start,
					safeErr(info.Result),
				)
			} else {
				finish(
					start,
					info.Error,
				)
			}
		}
	}
	t.OnStreamExecute = func(
		info trace.ScriptingStreamExecuteStartInfo,
	) func(
		trace.ScriptingStreamExecuteIntermediateInfo,
	) func(
		trace.ScriptingStreamExecuteDoneInfo,
	) {
		if adapter.Details()&trace.ScriptingEvents == 0 {
			return nil
		}

		start := childSpanWithReplaceCtx(
			adapter,
			info.Context,
			safeCall(info.Call),
			kv.String("query", info.Query),
		)

		return func(
			info trace.ScriptingStreamExecuteIntermediateInfo,
		) func(
			trace.ScriptingStreamExecuteDoneInfo,
		) {
			if info.Error != nil {
				start.Warn(info.Error)
			}

			return func(info trace.ScriptingStreamExecuteDoneInfo) {
				if info.Error != nil {
					start.Error(info.Error)
				}
				start.End()
			}
		}
	}
	t.OnExplain = func(info trace.ScriptingExplainStartInfo) func(trace.ScriptingExplainDoneInfo) {
		if adapter.Details()&trace.ScriptingEvents == 0 {
			return nil
		}

		start := childSpanWithReplaceCtx(
			adapter,
			info.Context,
			safeCall(info.Call),
			kv.String("query", info.Query),
		)

		return func(info trace.ScriptingExplainDoneInfo) {
			finish(start, info.Error)
		}
	}
	t.OnClose = func(info trace.ScriptingCloseStartInfo) func(trace.ScriptingCloseDoneInfo) {
		if adapter.Details()&trace.ScriptingEvents == 0 {
			return nil
		}

		start := childSpanWithReplaceCtx(
			adapter,
			info.Context,
			safeCall(info.Call),
		)

		return func(info trace.ScriptingCloseDoneInfo) {
			finish(start, info.Error)
		}
	}

	return t
}