Issue #67: used = TRUE after randomization

ttscience · Feb 22, 2024 · 0d6cdf4 · 0d6cdf4
1 parent e94c746
commit 0d6cdf4
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 6 deletions.
diff --git a/R/api_randomize.R b/R/api_randomize.R
@@ -89,7 +89,9 @@ api__randomize_patient <- function(study_id, current_state, req, res) {
     dplyr::select("arm_id" = "id", "name", "ratio") |>
     dplyr::collect()
 
-  randomized_patient <- unbiased:::save_patient(study_id, arm$arm_id)
+  randomized_patient <-
+    unbiased:::save_patient(study_id, arm$arm_id, used = TRUE) |>
+    select(-used)
 
   if (!is.null(randomized_patient$error)) {
     res$status <- 503

diff --git a/R/db.R b/R/db.R
@@ -147,15 +147,15 @@ create_study <- function(
   r
 }
 
-save_patient <- function(study_id, arm_id) {
+save_patient <- function(study_id, arm_id, used) {
   r <- tryCatch(
     {
       randomized_patient <- DBI::dbGetQuery(
         db_connection_pool,
-        "INSERT INTO patient (arm_id, study_id)
-                    VALUES ($1, $2)
-                    RETURNING id, arm_id",
-        list(arm_id, study_id)
+        "INSERT INTO patient (arm_id, study_id, used)
+                    VALUES ($1, $2, $3)
+                    RETURNING id, arm_id, used",
+        list(arm_id, study_id, used)
       )
     },
     error = function(cond) {