diff options
| author | Stefan Majewsky <majewsky@gmx.net> | 2026-08-01 14:50:01 +0200 |
|---|---|---|
| committer | Stefan Majewsky <majewsky@gmx.net> | 2026-08-01 14:50:01 +0200 |
| commit | 6c174bebade1627a837e70bec827753446d6498a (patch) | |
| tree | 35a229d9601565685f2ea2e24a268bb29be6bc55 /testing/pgruntime/helpers_test.go | |
| parent | e4a5b966decd910213de4ecbdc1b1ed2c4daf947 (diff) | |
| parent | 6d72c85766a8355e1d660ff0c19b43e93faad4d8 (diff) | |
| download | go-gg-6c174bebade1627a837e70bec827753446d6498a.tar.gz | |
Merge branch 'pgruntime'
Diffstat (limited to 'testing/pgruntime/helpers_test.go')
| -rw-r--r-- | testing/pgruntime/helpers_test.go | 47 |
1 files changed, 47 insertions, 0 deletions
diff --git a/testing/pgruntime/helpers_test.go b/testing/pgruntime/helpers_test.go new file mode 100644 index 0000000..ca4ffd2 --- /dev/null +++ b/testing/pgruntime/helpers_test.go @@ -0,0 +1,47 @@ +// SPDX-FileCopyrightText: 2026 Stefan Majewsky <majewsky@gmx.net> +// SPDX-License-Identifier: Apache-2.0 + +package pgruntime_test + +import ( + "context" + "database/sql" + + "go.xyrillian.de/gg/errext" + "go.xyrillian.de/gg/gsql" +) + +// Convenience function for executing a one-off SQL query returning no rows. +// TODO: move to gsql +func execQuery(ctx context.Context, db gsql.Handle, query string, args []any) (sql.Result, error) { + stmt, err := db.GSQLPrepare(ctx, query, false) + if err != nil { + return nil, err + } + result, err := stmt.Exec(ctx, args) + return result, errext.WithCleanup(err, "stmt.Close", stmt.Close()) +} + +// Convenience function for executing a one-off SQL query returning one row. +func queryRow(ctx context.Context, db gsql.Handle, query string, args, slots []any) error { + stmt, err := db.GSQLPrepare(ctx, query, false) + if err != nil { + return err + } + err = stmt.QueryRow(ctx, args, slots) + return errext.WithCleanup(err, "stmt.Close", stmt.Close()) +} + +// Convenience function for executing a one-off SQL query returning one value. +// TODO: move to gsql +func selectOneValue[T any](ctx context.Context, db gsql.Handle, query string, args ...any) (T, error) { + stmt, err := db.GSQLPrepare(ctx, query, false) + if err != nil { + var none T + return none, err + } + + var result T + err = stmt.QueryRow(ctx, args, []any{&result}) + return result, errext.WithCleanup(err, "stmt.Close", stmt.Close()) +} |
