@@ -11,8 +11,6 @@ import (
11
11
"strconv"
12
12
"time"
13
13
14
- "github.com/coder/coder/v2/coderd/prebuilds"
15
-
16
14
"github.com/dustin/go-humanize"
17
15
"github.com/go-chi/chi/v5"
18
16
"github.com/google/uuid"
@@ -636,17 +634,17 @@ func createWorkspace(
636
634
}
637
635
638
636
var (
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
643
640
)
644
641
prebuildsClaimer := * api .PrebuildsClaimer .Load ()
645
642
646
643
err = api .Database .InTx (func (db database.Store ) error {
647
644
var (
648
- workspaceID uuid.UUID
649
- claimedWorkspace * database.Workspace
645
+ workspaceID uuid.UUID
646
+ claimedWorkspace * database.Workspace
647
+ agentTokensByAgentID map [uuid.UUID ]string
650
648
)
651
649
652
650
// If a template preset was chosen, try claim a prebuilt workspace.
@@ -685,12 +683,13 @@ func createWorkspace(
685
683
} else {
686
684
// Prebuild found!
687
685
workspaceID = claimedWorkspace .ID
688
- initiatorID = prebuildClaimer .Initiator ()
686
+ initiatorID = prebuildsClaimer .Initiator ()
689
687
agents , err := db .GetWorkspaceAgentsInLatestBuildByWorkspaceID (ctx , claimedWorkspace .ID )
690
688
if err != nil {
691
689
api .Logger .Error (ctx , "failed to retrieve running agents of claimed prebuilt workspace" ,
692
690
slog .F ("workspace_id" , claimedWorkspace .ID ), slog .Error (err ))
693
691
}
692
+ agentTokensByAgentID = make (map [uuid.UUID ]string , len (agents ))
694
693
for _ , agent := range agents {
695
694
agentTokensByAgentID [agent .ID ] = agent .AuthToken .String ()
696
695
}
0 commit comments