Skip to content

Commit 73392ce

Browse files
committed
refactor stream adapter
1 parent dc8e03e commit 73392ce

File tree

1 file changed

+4
-5
lines changed

1 file changed

+4
-5
lines changed

internal/fwserver/server_listresource.go

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -78,11 +78,10 @@ func (s *Server) ListResource(ctx context.Context, fwReq *ListResourceRequest, f
7878
}
7979

8080
func listResourceEventStreamAdapter(stream iter.Seq[list.ListResourceEvent]) iter.Seq[ListResourceEvent] {
81-
return func(yield func(ListResourceEvent) bool) {
82-
for event := range stream {
83-
if !yield(ListResourceEvent(event)) {
84-
break
85-
}
81+
return func(yieldFw func(ListResourceEvent) bool) {
82+
yield := func(event list.ListResourceEvent) bool {
83+
return yieldFw(ListResourceEvent(event))
8684
}
85+
stream(yield)
8786
}
8887
}

0 commit comments

Comments
 (0)