aboutsummaryrefslogtreecommitdiff
path: root/internal/unsorted/database.go
diff options
context:
space:
mode:
authorRunxi Yu <me@runxiyu.org>2025-04-06 01:55:21 +0800
committerRunxi Yu <me@runxiyu.org>2025-04-06 02:08:58 +0800
commitfaa5ca8fab23176d390e9522f1485d467851545b (patch)
treed3b1d081e0ea5e7f71a94dc1d301e2540a8abcc8 /internal/unsorted/database.go
parentSlight refactor on NewServer (diff)
downloadforge-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.go43
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
+}