feat: add Raw

ch/query_raw.go

@@ -0,0 +1,69 @@
+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"
+}