add AndLock to query name

micromdm · Dec 11, 2024 · f51e34e · f51e34e
1 parent 0cf4452
commit f51e34e
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 9 deletions.
diff --git a/engine/storage/mysql/query.sql b/engine/storage/mysql/query.sql
@@ -127,7 +127,7 @@ FROM
 WHERE
   id = ?;
 
--- name: GetIDCommandsByStepID :many
+-- name: GetIDCommandsByStepIDAndLock :many
 SELECT
   ic.command_uuid,
   ic.request_type,

diff --git a/engine/storage/mysql/sqlc/query.sql.go b/engine/storage/mysql/sqlc/query.sql.go
diff --git a/engine/storage/mysql/storage.go b/engine/storage/mysql/storage.go
@@ -97,7 +97,7 @@ func (s *MySQLStorage) StoreCommandResponseAndRetrieveCompletedStep(ctx context.
 			Commands: []storage.StepCommandResult{*sc},
 		}
 
-		cmdR, err := qtx.GetIDCommandsByStepID(ctx, sqlc.GetIDCommandsByStepIDParams{
+		cmdR, err := qtx.GetIDCommandsByStepIDAndLock(ctx, sqlc.GetIDCommandsByStepIDAndLockParams{
 			EnrollmentID: id,
 			ID:           cmdCt.StepID,
 		})