Movatterモバイル変換

[0]ホーム
URL:

Skip to content

Navigation Menu

Sign in
Appearance settings

Search code, repositories, users, issues, pull requests...

Provide feedback

We read every piece of feedback, and take your input very seriously.

Saved searches

Use saved searches to filter your results more quickly

 Sign up
Appearance settings

Commitabac612

Browse files
committed
cleanup
1 parent30cfbe3 commitabac612

File tree

3 files changed

+16
-15
lines changed

3 files changed

+16
-15
lines changed

‎scaletest/workspaceupdates/config.go‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@ import (
1212
)
1313

1414
typeConfigstruct {
15-
// User is the configuration for the user to create or use.
15+
// User is the configuration for the user to create.
1616
User createusers.Config`json:"user"`
1717

1818
// Workspace is the configuration for the workspace to create. The workspace

‎scaletest/workspaceupdates/metrics.go‎

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,6 @@ package workspaceupdates
22

33
import (
44
"strconv"
5-
"sync/atomic"
65
"time"
76

87
"github.com/prometheus/client_golang/prometheus"
@@ -11,8 +10,6 @@ import (
1110
typeMetricsstruct {
1211
WorkspaceUpdatesLatencySeconds prometheus.HistogramVec
1312
WorkspaceUpdatesErrorsTotal prometheus.CounterVec
14-
15-
numErrors atomic.Int64
1613
}
1714

1815
funcNewMetrics(reg prometheus.Registerer)*Metrics {
@@ -41,6 +38,5 @@ func (m *Metrics) RecordCompletion(elapsed time.Duration, username string, owned
4138
}
4239

4340
func (m*Metrics)AddError(usernamestring,ownedWorkspacesint64,actionstring) {
44-
m.numErrors.Add(1)
4541
m.WorkspaceUpdatesErrorsTotal.WithLabelValues(username,strconv.Itoa(int(ownedWorkspaces)),action).Inc()
4642
}

‎scaletest/workspaceupdates/run.go‎

Lines changed: 15 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -71,35 +71,34 @@ func (r *Runner) Run(ctx context.Context, id string, logs io.Writer) error {
7171
r.client.SetLogBodies(true)
7272

7373
r.createUserRunner=createusers.NewRunner(r.client,r.cfg.User)
74-
newUser,err:=r.createUserRunner.RunReturningUser(ctx,id,logs)
74+
newUserAndToken,err:=r.createUserRunner.RunReturningUser(ctx,id,logs)
7575
iferr!=nil {
7676
returnxerrors.Errorf("create user: %w",err)
7777
}
78-
user:=newUser.User
79-
client:=codersdk.New(r.client.URL,
80-
codersdk.WithSessionToken(newUser.SessionToken),
78+
newUser:=newUserAndToken.User
79+
newUserClient:=codersdk.New(r.client.URL,
80+
codersdk.WithSessionToken(newUserAndToken.SessionToken),
8181
codersdk.WithLogger(logger),
8282
codersdk.WithLogBodies())
8383

84-
logger.Info(ctx,fmt.Sprintf("user %q created",user.Username),slog.F("id",user.ID.String()))
84+
logger.Info(ctx,fmt.Sprintf("user %q created",newUser.Username),slog.F("id",newUser.ID.String()))
8585

8686
dialCtx,cancel:=context.WithTimeout(ctx,r.cfg.DialTimeout)
8787
defercancel()
8888

8989
logger.Info(ctx,"connecting to workspace updates stream")
90-
clients,err:=r.dialTailnet(dialCtx,client,user,logger)
90+
clients,err:=r.dialTailnet(dialCtx,newUserClient,newUser,logger)
9191
iferr!=nil {
9292
returnxerrors.Errorf("tailnet dial failed: %w",err)
9393
}
9494
logger.Info(ctx,"connected to workspace updates stream")
95-
deferclients.Closer.Close()
9695

9796
watchCtx,cancelWatch:=context.WithCancel(ctx)
9897
defercancelWatch()
9998

10099
completionCh:=make(chanerror,1)
101100
gofunc() {
102-
completionCh<-r.watchWorkspaceUpdates(watchCtx,clients,user,logger)
101+
completionCh<-r.watchWorkspaceUpdates(watchCtx,clients,newUser,logger)
103102
}()
104103

105104
reachedBarrier=true
@@ -110,14 +109,15 @@ func (r *Runner) Run(ctx context.Context, id string, logs io.Writer) error {
110109
fori:=ranger.cfg.WorkspaceCount {
111110
workspaceName,err:=loadtestutil.GenerateWorkspaceName(id)
112111
iferr!=nil {
112+
clients.Closer.Close()
113113
returnxerrors.Errorf("generate random name for workspace: %w",err)
114114
}
115115
workspaceBuildConfig:=r.cfg.Workspace
116116
workspaceBuildConfig.OrganizationID=r.cfg.User.OrganizationID
117-
workspaceBuildConfig.UserID=user.ID.String()
117+
workspaceBuildConfig.UserID=newUser.ID.String()
118118
workspaceBuildConfig.Request.Name=workspaceName
119119

120-
runner:=workspacebuild.NewRunner(client,workspaceBuildConfig)
120+
runner:=workspacebuild.NewRunner(newUserClient,workspaceBuildConfig)
121121
r.workspacebuildRunners=append(r.workspacebuildRunners,runner)
122122

123123
logger.Info(ctx,fmt.Sprintf("creating workspace %d/%d",i+1,r.cfg.WorkspaceCount))
@@ -128,6 +128,7 @@ func (r *Runner) Run(ctx context.Context, id string, logs io.Writer) error {
128128
}
129129
err=runner.Run(ctx,fmt.Sprintf("%s-%d",id,i),logs)
130130
iferr!=nil {
131+
clients.Closer.Close()
131132
returnxerrors.Errorf("create workspace %d: %w",i,err)
132133
}
133134
}
@@ -143,8 +144,12 @@ func (r *Runner) Run(ctx context.Context, id string, logs io.Writer) error {
143144
returnxerrors.Errorf("workspace updates streaming failed: %w",err)
144145
}
145146
logger.Info(ctx,"workspace updates streaming completed successfully")
147+
clients.Closer.Close()
146148
returnnil
147149
case<-waitUpdatesCtx.Done():
150+
cancelWatch()
151+
clients.Closer.Close()
152+
<-completionCh// ensure watch goroutine exits
148153
ifwaitUpdatesCtx.Err()==context.DeadlineExceeded {
149154
returnxerrors.Errorf("timeout waiting for workspace updates after %v",r.cfg.WorkspaceUpdatesTimeout)
150155
}

0 commit comments

Comments
 (0)
[8]ページ先頭
©2009-2025 Movatter.jp