@@ -11,8 +11,6 @@ import (
1111"strconv"
1212"time"
1313
14- "github.com/coder/coder/v2/coderd/prebuilds"
15-
1614"github.com/dustin/go-humanize"
1715"github.com/go-chi/chi/v5"
1816"github.com/google/uuid"
@@ -636,17 +634,17 @@ func createWorkspace(
636634}
637635
638636var (
639- provisionerJob * database.ProvisionerJob
640- workspaceBuild * database.WorkspaceBuild
641- provisionerDaemons []database.GetEligibleProvisionerDaemonsByProvisionerJobIDsRow
642- agentTokensByAgentID map [uuid.UUID ]string
637+ provisionerJob * database.ProvisionerJob
638+ workspaceBuild * database.WorkspaceBuild
639+ provisionerDaemons []database.GetEligibleProvisionerDaemonsByProvisionerJobIDsRow
643640)
644641prebuildsClaimer := * api .PrebuildsClaimer .Load ()
645642
646643err = api .Database .InTx (func (db database.Store )error {
647644var (
648- workspaceID uuid.UUID
649- claimedWorkspace * database.Workspace
645+ workspaceID uuid.UUID
646+ claimedWorkspace * database.Workspace
647+ agentTokensByAgentID map [uuid.UUID ]string
650648)
651649
652650// If a template preset was chosen, try claim a prebuilt workspace.
@@ -685,12 +683,13 @@ func createWorkspace(
685683}else {
686684// Prebuild found!
687685workspaceID = claimedWorkspace .ID
688- initiatorID = prebuildClaimer .Initiator ()
686+ initiatorID = prebuildsClaimer .Initiator ()
689687agents ,err := db .GetWorkspaceAgentsInLatestBuildByWorkspaceID (ctx ,claimedWorkspace .ID )
690688if err != nil {
691689api .Logger .Error (ctx ,"failed to retrieve running agents of claimed prebuilt workspace" ,
692690slog .F ("workspace_id" ,claimedWorkspace .ID ),slog .Error (err ))
693691}
692+ agentTokensByAgentID = make (map [uuid.UUID ]string ,len (agents ))
694693for _ ,agent := range agents {
695694agentTokensByAgentID [agent .ID ]= agent .AuthToken .String ()
696695}