sql_conn_prepare_context.go

// (c) Copyright IBM Corp. 2023

package instana

import (
	"context"
	"database/sql/driver"
)

type wConnPrepareContext struct {
	driver.ConnPrepareContext
	connDetails DbConnDetails
	sensor      TracerLogger
}

func (conn *wConnPrepareContext) PrepareContext(ctx context.Context, query string) (driver.Stmt, error) {
	stmt, err := conn.ConnPrepareContext.PrepareContext(ctx, query)

	if err != nil {
		conn.connDetails.Error = err
		return stmt, err
	}

	if stmtAlreadyWrapped(stmt) {
		return stmt, nil
	}

	return wrapStmt(stmt, query, conn.connDetails, conn.sensor), nil
}