sqldb

type DB interface {
	Begin() (*Tx, error)
	BeginTx(ctx context.Context, opts *sql.TxOptions) (*Tx, error)
	Close() error
	Conn(ctx context.Context) (*sql.Conn, error)
	Exec(query string, args ...any) (sql.Result, error)
	ExecContext(ctx context.Context, query string, args ...any) (sql.Result, error)
	Ping() error
	PingContext(ctx context.Context) error
	Prepare(query string) (*sql.Stmt, error)
	PrepareContext(ctx context.Context, query string) (*sql.Stmt, error)
	Query(query string, args ...any) (*sql.Rows, error)
	QueryContext(ctx context.Context, query string, args ...any) (*sql.Rows, error)
	QueryRow(query string, args ...any) *sql.Row
	QueryRowContext(ctx context.Context, query string, args ...any) *sql.Row
	SetConnMaxIdleTime(d time.Duration)
	SetConnMaxLifetime(d time.Duration)
	Stats() sql.DBStats
	WithHook(h Hook)
	SetMaxIdleConns(n int)
	SetMaxOpenConns(n int)
}

type Hook interface {
	AfterQuery(
		context context.Context,
		sid string,
		method string,
		query string,
		args []interface{},
		start_time time.Time,
		end_time time.Time,
		err error,
	)
	AfterBegin(
		context context.Context,
		sid string,
		start_time,
		end_time time.Time,
		err error,
	)
	AfterCommit(
		context context.Context,
		sid string,
		start_time,
		end_time time.Time,
		err error,
	)
	AfterRollback(
		context context.Context,
		start_time, end_time time.Time,
		err error,
	)
}

type Options struct {
	MaxIdleConns   int
	MaxOpenConns   int
	PanicablePings bool
	Name           string
	Hook           Hook
}

type Trx interface {
	TraceDependency(
		ctx context.Context,
		spanId string,
		dependencyType string,
		serviceName string,
		commandName string,
		success bool,
		startTimestamp time.Time,
		eventTimestamp time.Time,
		fields map[string]string,
	)
	TraceException(
		ctx context.Context,
		err interface{},
		skip int,
		fields map[string]string,
	)
}

type Tx struct {
	*sql.Tx
	// contains filtered or unexported fields
}