1
0
mirror of https://github.com/uptrace/go-clickhouse.git synced 2025-06-08 23:26:11 +02:00
go-clickhouse/ch/query_raw.go

70 lines
1.2 KiB
Go
Raw Normal View History

2022-07-14 10:41:16 +03:00
package ch
import (
"context"
"database/sql"
"github.com/uptrace/go-clickhouse/ch/chschema"
)
type RawQuery struct {
baseQuery
query string
args []any
}
func (db *DB) Raw(query string, args ...any) *RawQuery {
return &RawQuery{
baseQuery: baseQuery{
db: db,
},
query: query,
args: args,
}
}
func (q *RawQuery) Scan(ctx context.Context, dest ...any) error {
return q.scan(ctx, dest, false)
}
func (q *RawQuery) ScanColumns(ctx context.Context, dest ...any) error {
return q.scan(ctx, dest, true)
}
func (q *RawQuery) scan(ctx context.Context, dest []any, columnar bool) error {
if q.err != nil {
return q.err
}
model, err := q.newModel(dest...)
if err != nil {
return err
}
query := q.db.FormatQuery(q.query, q.args...)
ctx, evt := q.db.beforeQuery(ctx, q, query, nil, model)
res, err := q.baseQuery.query(ctx, model, query)
q.db.afterQuery(ctx, evt, res, err)
if err != nil {
return err
}
if !columnar && useQueryRowModel(model) {
if res.affected == 0 {
return sql.ErrNoRows
}
}
return nil
}
func (q *RawQuery) AppendQuery(fmter chschema.Formatter, b []byte) ([]byte, error) {
return fmter.AppendQuery(b, q.query, q.args), nil
}
func (q *RawQuery) Operation() string {
return "SELECT"
}