steampipe/pkg/query/queryresult/result_streamer.go

package queryresult

type ResultStreamer struct {
	Results            chan *Result
	allResultsReceived chan string
}

func NewResultStreamer() *ResultStreamer {
	return &ResultStreamer{
		// make buffered channel so we can always stream a single result
		Results:            make(chan *Result, 1),
		allResultsReceived: make(chan string, 1),
	}
}

// StreamResult streams result on the Results channel, then waits for them to be read
func (q *ResultStreamer) StreamResult(result *Result) {
	q.Results <- result
	// wait for the result to be read
	<-q.allResultsReceived
}

// Close closes the result stream
func (q *ResultStreamer) Close() {
	close(q.Results)
}

// AllResultsRead is a signal that indicates the all results have been read from the stream
func (q *ResultStreamer) AllResultsRead() {
	q.allResultsReceived <- ""
}