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

Commit31a3d36

Browse files
committed
use subscribewitherr
1 parentd8cd584 commit31a3d36

File tree

5 files changed

+49
-33
lines changed

5 files changed

+49
-33
lines changed

‎coderd/agentapi/stats_test.go

Lines changed: 10 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -13,8 +13,6 @@ import (
1313
"go.uber.org/mock/gomock"
1414
"google.golang.org/protobuf/types/known/durationpb"
1515

16-
"cdr.dev/slog/sloggers/slogtest"
17-
1816
agentproto"github.com/coder/coder/v2/agent/proto"
1917
"github.com/coder/coder/v2/coderd/agentapi"
2018
"github.com/coder/coder/v2/coderd/database"
@@ -152,10 +150,12 @@ func TestUpdateStates(t *testing.T) {
152150

153151
// Ensure that pubsub notifications are sent.
154152
notifyDescription:=make(chanstruct{})
155-
ps.Subscribe(wspubsub.WorkspaceEventChannel(workspace.OwnerID),
153+
ps.SubscribeWithErr(wspubsub.WorkspaceEventChannel(workspace.OwnerID),
156154
wspubsub.HandleWorkspaceEvent(
157-
slogtest.Make(t,nil),
158-
func(_ context.Context,e wspubsub.WorkspaceEvent) {
155+
func(_ context.Context,e wspubsub.WorkspaceEvent,errerror) {
156+
iferr!=nil {
157+
return
158+
}
159159
ife.Kind==wspubsub.WorkspaceEventKindStatsUpdate&&e.WorkspaceID==workspace.ID {
160160
gofunc() {
161161
notifyDescription<-struct{}{}
@@ -490,10 +490,12 @@ func TestUpdateStates(t *testing.T) {
490490

491491
// Ensure that pubsub notifications are sent.
492492
notifyDescription:=make(chanstruct{})
493-
ps.Subscribe(wspubsub.WorkspaceEventChannel(workspace.OwnerID),
493+
ps.SubscribeWithErr(wspubsub.WorkspaceEventChannel(workspace.OwnerID),
494494
wspubsub.HandleWorkspaceEvent(
495-
slogtest.Make(t,nil),
496-
func(_ context.Context,e wspubsub.WorkspaceEvent) {
495+
func(_ context.Context,e wspubsub.WorkspaceEvent,errerror) {
496+
iferr!=nil {
497+
return
498+
}
497499
ife.Kind==wspubsub.WorkspaceEventKindStatsUpdate&&e.WorkspaceID==workspace.ID {
498500
gofunc() {
499501
notifyDescription<-struct{}{}

‎coderd/provisionerdserver/provisionerdserver_test.go

Lines changed: 20 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -296,10 +296,12 @@ func TestAcquireJob(t *testing.T) {
296296

297297
startPublished:=make(chanstruct{})
298298
varclosedbool
299-
closeStartSubscribe,err:=ps.Subscribe(wspubsub.WorkspaceEventChannel(workspace.OwnerID),
299+
closeStartSubscribe,err:=ps.SubscribeWithErr(wspubsub.WorkspaceEventChannel(workspace.OwnerID),
300300
wspubsub.HandleWorkspaceEvent(
301-
slogtest.Make(t,nil),
302-
func(_ context.Context,e wspubsub.WorkspaceEvent) {
301+
func(_ context.Context,e wspubsub.WorkspaceEvent,errerror) {
302+
iferr!=nil {
303+
return
304+
}
303305
ife.Kind==wspubsub.WorkspaceEventKindStateChange&&e.WorkspaceID==workspace.ID {
304306
if!closed {
305307
close(startPublished)
@@ -404,10 +406,12 @@ func TestAcquireJob(t *testing.T) {
404406
})
405407

406408
stopPublished:=make(chanstruct{})
407-
closeStopSubscribe,err:=ps.Subscribe(wspubsub.WorkspaceEventChannel(workspace.OwnerID),
409+
closeStopSubscribe,err:=ps.SubscribeWithErr(wspubsub.WorkspaceEventChannel(workspace.OwnerID),
408410
wspubsub.HandleWorkspaceEvent(
409-
slogtest.Make(t,nil),
410-
func(_ context.Context,e wspubsub.WorkspaceEvent) {
411+
func(_ context.Context,e wspubsub.WorkspaceEvent,errerror) {
412+
iferr!=nil {
413+
return
414+
}
411415
ife.Kind==wspubsub.WorkspaceEventKindStateChange&&e.WorkspaceID==workspace.ID {
412416
close(stopPublished)
413417
}
@@ -925,10 +929,12 @@ func TestFailJob(t *testing.T) {
925929
require.NoError(t,err)
926930

927931
publishedWorkspace:=make(chanstruct{})
928-
closeWorkspaceSubscribe,err:=ps.Subscribe(wspubsub.WorkspaceEventChannel(workspace.OwnerID),
932+
closeWorkspaceSubscribe,err:=ps.SubscribeWithErr(wspubsub.WorkspaceEventChannel(workspace.OwnerID),
929933
wspubsub.HandleWorkspaceEvent(
930-
slogtest.Make(t,nil),
931-
func(_ context.Context,e wspubsub.WorkspaceEvent) {
934+
func(_ context.Context,e wspubsub.WorkspaceEvent,errerror) {
935+
iferr!=nil {
936+
return
937+
}
932938
ife.Kind==wspubsub.WorkspaceEventKindStateChange&&e.WorkspaceID==workspace.ID {
933939
close(publishedWorkspace)
934940
}
@@ -1321,10 +1327,12 @@ func TestCompleteJob(t *testing.T) {
13211327
require.NoError(t,err)
13221328

13231329
publishedWorkspace:=make(chanstruct{})
1324-
closeWorkspaceSubscribe,err:=ps.Subscribe(wspubsub.WorkspaceEventChannel(workspace.OwnerID),
1330+
closeWorkspaceSubscribe,err:=ps.SubscribeWithErr(wspubsub.WorkspaceEventChannel(workspace.OwnerID),
13251331
wspubsub.HandleWorkspaceEvent(
1326-
slogtest.Make(t,nil),
1327-
func(_ context.Context,e wspubsub.WorkspaceEvent) {
1332+
func(_ context.Context,e wspubsub.WorkspaceEvent,errerror) {
1333+
iferr!=nil {
1334+
return
1335+
}
13281336
ife.Kind==wspubsub.WorkspaceEventKindStateChange&&e.WorkspaceID==workspace.ID {
13291337
close(publishedWorkspace)
13301338
}

‎coderd/workspaceagents.go

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -417,10 +417,12 @@ func (api *API) workspaceAgentLogs(rw http.ResponseWriter, r *http.Request) {
417417
notifyCh<-struct{}{}
418418

419419
// Subscribe to workspace to detect new builds.
420-
closeSubscribeWorkspace,err:=api.Pubsub.Subscribe(wspubsub.WorkspaceEventChannel(workspace.OwnerID),
420+
closeSubscribeWorkspace,err:=api.Pubsub.SubscribeWithErr(wspubsub.WorkspaceEventChannel(workspace.OwnerID),
421421
wspubsub.HandleWorkspaceEvent(
422-
logger,
423-
func(_ context.Context,e wspubsub.WorkspaceEvent) {
422+
func(_ context.Context,e wspubsub.WorkspaceEvent,errerror) {
423+
iferr!=nil {
424+
return
425+
}
424426
ife.Kind==wspubsub.WorkspaceEventKindStateChange&&e.WorkspaceID==workspace.ID {
425427
select {
426428
caseworkspaceNotifyCh<-struct{}{}:

‎coderd/workspaces.go

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1702,10 +1702,12 @@ func (api *API) watchWorkspace(rw http.ResponseWriter, r *http.Request) {
17021702
})
17031703
}
17041704

1705-
cancelWorkspaceSubscribe,err:=api.Pubsub.Subscribe(wspubsub.WorkspaceEventChannel(workspace.OwnerID),
1705+
cancelWorkspaceSubscribe,err:=api.Pubsub.SubscribeWithErr(wspubsub.WorkspaceEventChannel(workspace.OwnerID),
17061706
wspubsub.HandleWorkspaceEvent(
1707-
api.Logger,
1708-
func(ctx context.Context,payload wspubsub.WorkspaceEvent) {
1707+
func(ctx context.Context,payload wspubsub.WorkspaceEvent,errerror) {
1708+
iferr!=nil {
1709+
return
1710+
}
17091711
ifpayload.WorkspaceID!=workspace.ID {
17101712
return
17111713
}

‎coderd/wspubsub/wspubsub.go

Lines changed: 9 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -5,8 +5,6 @@ import (
55
"encoding/json"
66
"fmt"
77

8-
"cdr.dev/slog"
9-
108
"github.com/google/uuid"
119
"golang.org/x/xerrors"
1210
)
@@ -17,18 +15,22 @@ func WorkspaceEventChannel(ownerID uuid.UUID) string {
1715
returnfmt.Sprintf("workspace_owner:%s",ownerID)
1816
}
1917

20-
funcHandleWorkspaceEvent(logger slog.Logger,cbfunc(ctx context.Context,payloadWorkspaceEvent))func(ctx context.Context,message []byte) {
21-
returnfunc(ctx context.Context,message []byte) {
18+
funcHandleWorkspaceEvent(cbfunc(ctx context.Context,payloadWorkspaceEvent,errerror))func(ctx context.Context,message []byte,errerror) {
19+
returnfunc(ctx context.Context,message []byte,errerror) {
20+
iferr!=nil {
21+
cb(ctx,WorkspaceEvent{},xerrors.Errorf("workspace event pubsub: %w",err))
22+
return
23+
}
2224
varpayloadWorkspaceEvent
2325
iferr:=json.Unmarshal(message,&payload);err!=nil {
24-
logger.Warn(ctx,"failed tounmarshal workspace event",slog.Error(err))
26+
cb(ctx,WorkspaceEvent{},xerrors.Errorf("unmarshal workspace event"))
2527
return
2628
}
2729
iferr:=payload.Validate();err!=nil {
28-
logger.Warn(ctx,"invalid workspace event",slog.Error(err))
30+
cb(ctx,payload,xerrors.Errorf("validate workspace event"))
2931
return
3032
}
31-
cb(ctx,payload)
33+
cb(ctx,payload,err)
3234
}
3335
}
3436

0 commit comments

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