Gleipnir/nidus-sync
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Use a prepared function and get a result

Browse source
This commit is contained in:
Eli Ribble 2025-12-02 22:44:51 +00:00
parent eac240af51
commit b488730177
No known key found for this signature in database
2 changed files with 17 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

23
db/prepared.go
View file

@ -12,6 +12,7 @@ import (
"github.com/rs/zerolog/log"
"github.com/stephenafamo/bob"
"github.com/stephenafamo/bob/dialect/psql"
"github.com/stephenafamo/scan"
)
//go:embed prepared_functions/*.sql
@ -63,8 +64,11 @@ func prepareStatements(ctx context.Context) error {
return nil
}
func TestPreparedQuery(ctx context.Context) error {
query := psql.RawQuery("EXECUTE test_function")
result, err := bob.Exec(ctx, PGInstance.BobDB, query)
type Skn struct {
Result int
}
query := psql.RawQuery("EXECUTE test_function(4)")
result, err := bob.One[Skn](ctx, PGInstance.BobDB, query, scan.StructMapper[Skn]())
if err != nil {
return fmt.Errorf("Failed to exectue test function: %w", err)
}
@ -73,13 +77,16 @@ func TestPreparedQuery(ctx context.Context) error {
log.Error().Err(err).Msg("failed insert id")
return fmt.Errorf("Failed to get insert ID: %w", err)
}*/
rows_affected, err := result.RowsAffected()
if err != nil {
log.Error().Err(err).Msg("failed rows affected")
return fmt.Errorf("Failed to get rows affected: %w", err)
}
/*
rows_affected, err := result.RowsAffected()
if err != nil {
log.Error().Err(err).Msg("failed rows affected")
return fmt.Errorf("Failed to get rows affected: %w", err)
}
*/
//log.Info().Int64("insert id", insert_id).Int64("rows", rows_affected).Msg("bah")
log.Info().Int64("rows", rows_affected).Msg("got rows")
//log.Info().Int64("rows", rows_affected).Msg("got rows")
log.Info().Int("value", result.Result).Msg("got result")
return nil
}

4
db/prepared_functions/test_function.sql
View file

@ -1,3 +1,3 @@
PREPARE test_function AS
SELECT version();
PREPARE test_function(int) AS
SELECT 2*$1 AS result;