diff options
author | Runxi Yu <me@runxiyu.org> | 2025-04-06 01:55:21 +0800 |
---|---|---|
committer | Runxi Yu <me@runxiyu.org> | 2025-04-06 02:08:58 +0800 |
commit | faa5ca8fab23176d390e9522f1485d467851545b (patch) | |
tree | d3b1d081e0ea5e7f71a94dc1d301e2540a8abcc8 /internal/unsorted/database.go | |
parent | Slight refactor on NewServer (diff) | |
download | forge-d0b51bf88914c72ad8405c1dfccd7ada12bfeaf8.tar.gz forge-d0b51bf88914c72ad8405c1dfccd7ada12bfeaf8.tar.zst forge-d0b51bf88914c72ad8405c1dfccd7ada12bfeaf8.zip |
Move stuff into internal/unsortedv0.1.28
Diffstat (limited to 'internal/unsorted/database.go')
-rw-r--r-- | internal/unsorted/database.go | 43 |
1 files changed, 43 insertions, 0 deletions
diff --git a/internal/unsorted/database.go b/internal/unsorted/database.go new file mode 100644 index 0000000..222b0c4 --- /dev/null +++ b/internal/unsorted/database.go @@ -0,0 +1,43 @@ +// SPDX-License-Identifier: AGPL-3.0-only +// SPDX-FileCopyrightText: Copyright (c) 2025 Runxi Yu <https://runxiyu.org> + +package unsorted + +import ( + "context" + + "github.com/jackc/pgx/v5" +) + +// TODO: All database handling logic in all request handlers must be revamped. +// We must ensure that each request has all logic in one transaction (subject +// to exceptions if appropriate) so they get a consistent view of the database +// at a single point. A failure to do so may cause things as serious as +// privilege escalation. + +// queryNameDesc is a helper function that executes a query and returns a +// list of nameDesc results. The query must return two string arguments, i.e. a +// name and a description. +func (s *Server) queryNameDesc(ctx context.Context, query string, args ...any) (result []nameDesc, err error) { + var rows pgx.Rows + + if rows, err = s.database.Query(ctx, query, args...); err != nil { + return nil, err + } + defer rows.Close() + + for rows.Next() { + var name, description string + if err = rows.Scan(&name, &description); err != nil { + return nil, err + } + result = append(result, nameDesc{name, description}) + } + return result, rows.Err() +} + +// nameDesc holds a name and a description. +type nameDesc struct { + Name string + Description string +} |