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

Properly read prepared functions from the embedded FS

Browse source 
Yeah, reading that wrong was bad.
This commit is contained in:
Eli Ribble 2025-12-02 22:28:10 +00:00
parent 97ec2c767d
commit eac240af51
No known key found for this signature in database
3 changed files with 31 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

26
db/prepared.go
View file

@ -10,6 +10,8 @@ import (
//"github.com/stephenafamo/bob"
//"github.com/stephenafamo/bob/dialect/psql"
"github.com/rs/zerolog/log"
"github.com/stephenafamo/bob"
"github.com/stephenafamo/bob/dialect/psql"
)
//go:embed prepared_functions/*.sql
@ -20,19 +22,21 @@ var sqlFiles embed.FS
// preparing statements that will be used later.
func prepareStatements(ctx context.Context) error {
// Get a list of all embedded SQL files
entries, err := sqlFiles.ReadDir(".")
entries, err := sqlFiles.ReadDir("prepared_functions")
if err != nil {
return fmt.Errorf("failed to read SQL directory: %w", err)
}
log.Info().Int("len", len(entries)).Msg("Reading prepared functions")
// Process each SQL file
for _, entry := range entries {
if entry.IsDir() || !strings.HasSuffix(entry.Name(), ".sql") {
log.Info().Str("name", entry.Name()).Msg("Skipping")
continue
}
// Read the SQL file content
content, err := sqlFiles.ReadFile(entry.Name())
content, err := sqlFiles.ReadFile(filepath.Join("prepared_functions", entry.Name()))
if err != nil {
return fmt.Errorf("failed to read SQL file %s: %w", entry.Name(), err)
}
@ -59,5 +63,23 @@ 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)
if err != nil {
return fmt.Errorf("Failed to exectue test function: %w", err)
}
/*insert_id, err := result.LastInsertId()
if err != nil {
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)
}
//log.Info().Int64("insert id", insert_id).Int64("rows", rows_affected).Msg("bah")
log.Info().Int64("rows", rows_affected).Msg("got rows")
return nil
}