Function SynchronousStream

coderd/coderdtest/stream.go:12–25

SynchronousStream returns a function that assumes the stream is synchronous. Meaning each request sent assumes exactly one response will be received. The function will block until the response is received or an error occurs. This should not be used in production code, as it does not handle edge cas

(stream *wsjson.Stream[R, W])

// The second function `pop` can be used to retrieve the next response from the
// stream without sending a new request. This is useful for dynamic parameters
func SynchronousStream[R any, W any](stream *wsjson.Stream[R, W]) (do func(W) (R, error), pop func() R) {
	rec := stream.Chan()

	return func(req W) (R, error) {
			err := stream.Send(req)
			if err != nil {
				return *new(R), err
			}

			return <-rec, nil
		}, func() R {
			return <-rec
		}
}