c/bmaas/bmdb/model: split query files
This was starting to become slightly unwieldy.
Change-Id: I8f9d63ff6f844f1452fca2a98d253a9582a3ee8d
Reviewed-on: https://review.monogon.dev/c/monogon/+/1085
Tested-by: Jenkins CI
Reviewed-by: Leopold Schabel <leo@monogon.tech>
diff --git a/cloud/bmaas/bmdb/model/queries_base.sql b/cloud/bmaas/bmdb/model/queries_base.sql
new file mode 100644
index 0000000..60d1737
--- /dev/null
+++ b/cloud/bmaas/bmdb/model/queries_base.sql
@@ -0,0 +1,44 @@
+-- name: NewMachine :one
+INSERT INTO machines (
+ machine_created_at
+) VALUES (
+ now()
+)
+RETURNING *;
+
+-- name: NewSession :one
+INSERT INTO sessions (
+ session_component_name, session_runtime_info, session_created_at, session_interval_seconds, session_deadline
+) VALUES (
+ $1, $2, now(), $3, (now() + $3 * interval '1 second')
+)
+RETURNING *;
+
+-- name: SessionPoke :exec
+-- Update a given session with a new deadline. Must be called in the same
+-- transaction as SessionCheck to ensure the session is still alive.
+UPDATE sessions
+SET session_deadline = now() + session_interval_seconds * interval '1 second'
+WHERE session_id = $1;
+
+-- name: SessionCheck :many
+-- SessionCheck returns a session by ID if that session is still valid (ie. its
+-- deadline hasn't expired).
+SELECT *
+FROM sessions
+WHERE session_id = $1
+AND session_deadline > now();
+
+-- name: StartWork :exec
+INSERT INTO work (
+ machine_id, session_id, process
+) VALUES (
+ $1, $2, $3
+);
+
+-- name: FinishWork :exec
+DELETE FROM work
+WHERE machine_id = $1
+ AND session_id = $2
+ AND process = $3;
+