internal/stmt/stmt.go - master - githubmirror/ydb-go-sql

internal/stmt/stmt.go
85 строк2 KB
asmyasnikov
fix linter issues
18 дек 2021, 17:18
18 дек 2021, 17:189798075
package stmt

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

	"github.com/ydb-platform/ydb-go-sdk/v3/table"

	"github.com/ydb-platform/ydb-go-sql/internal/check"
	"github.com/ydb-platform/ydb-go-sql/internal/errors"
	"github.com/ydb-platform/ydb-go-sql/internal/mode"
	"github.com/ydb-platform/ydb-go-sql/internal/nop"
	"github.com/ydb-platform/ydb-go-sql/internal/rows"
	"github.com/ydb-platform/ydb-go-sql/internal/x"
)

type Stmt interface {
	driver.Stmt
	driver.StmtQueryContext
	driver.StmtExecContext

	driver.NamedValueChecker
}

type stmt struct {
	stmt             table.Statement
	defaultTxControl *table.TransactionControl
}

func (s *stmt) QueryContext(ctx context.Context, args []driver.NamedValue) (driver.Rows, error) {
	switch m := x.QueryMode(ctx); m {
	case mode.DataQuery:
		_, res, err := s.stmt.Execute(ctx, x.TxControl(ctx, s.defaultTxControl), x.ToQueryParams(args), x.DataQueryOptions(ctx)...)
		if err != nil {
			return nil, errors.Map(err)
		}
		return rows.Result(res), errors.Map(res.Err())
	default:
		return nil, fmt.Errorf("unsupported query mode %s type for execute statement query", m)
	}
}

func (s *stmt) ExecContext(ctx context.Context, args []driver.NamedValue) (driver.Result, error) {
	switch m := x.QueryMode(ctx); m {
	case mode.DataQuery:
		_, res, err := s.stmt.Execute(ctx, x.TxControl(ctx, s.defaultTxControl), x.ToQueryParams(args), x.DataQueryOptions(ctx)...)
		if err != nil {
			return nil, errors.Map(err)
		}
		return nop.Result(), errors.Map(res.Err())
	default:
		return nil, fmt.Errorf("unsupported query mode %s type for execute query", m)
	}
}

func New(
	s table.Statement,
	defaultTxControl *table.TransactionControl,
) Stmt {
	return &stmt{
		stmt:             s,
		defaultTxControl: defaultTxControl,
	}
}

func (s *stmt) NumInput() int {
	return s.stmt.NumInput()
}

func (s *stmt) Close() error {
	return nil
}

func (s stmt) Exec(args []driver.Value) (driver.Result, error) {
	return nil, errors.ErrDeprecated
}

func (s stmt) Query(args []driver.Value) (driver.Rows, error) {
	return nil, errors.ErrDeprecated
}

func (s *stmt) CheckNamedValue(v *driver.NamedValue) error {
	return check.NamedValue(v)
}