mirror of
https://github.com/coder/coder.git
synced 2025-07-03 16:13:58 +00:00
9b5d49967c
This PR extracts dynamic parameter rendering logic from coderd/parameters.go into a new coderd/dynamicparameters package. Partly for organization and maintainability, but primarily to be reused in `wsbuilder` to be leveraged as validation.
26 lines
846 B
Go
26 lines
846 B
Go
package coderdtest
|
|
|
|
import "github.com/coder/coder/v2/codersdk/wsjson"
|
|
|
|
// 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 cases.
|
|
// 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
|
|
}
|
|
}
|