mirror of
https://github.com/opentofu/opentofu.git
synced 2024-12-26 00:41:27 -06:00
5ac311e2a9
We use a third-party library "colorable" to translate VT100 color sequences into Windows console attribute-setting calls when Terraform is running on Windows. colorable is not concurrency-safe for multiple writes to the same console, because it writes to the console one character at a time and so two concurrent writers get their characters interleaved, creating unreadable garble. Here we wrap around it a synchronization mechanism to ensure that there can be only one Write call outstanding across both stderr and stdout, mimicking the usual behavior we expect (when stderr/stdout are a normal file handle) of each Write being completed atomically.
32 lines
637 B
Go
32 lines
637 B
Go
package main
|
|
|
|
import (
|
|
"io"
|
|
"sync"
|
|
)
|
|
|
|
type synchronizedWriter struct {
|
|
io.Writer
|
|
mutex *sync.Mutex
|
|
}
|
|
|
|
// synchronizedWriters takes a set of writers and returns wrappers that ensure
|
|
// that only one write can be outstanding at a time across the whole set.
|
|
func synchronizedWriters(targets ...io.Writer) []io.Writer {
|
|
mutex := &sync.Mutex{}
|
|
ret := make([]io.Writer, len(targets))
|
|
for i, target := range targets {
|
|
ret[i] = &synchronizedWriter{
|
|
Writer: target,
|
|
mutex: mutex,
|
|
}
|
|
}
|
|
return ret
|
|
}
|
|
|
|
func (w *synchronizedWriter) Write(p []byte) (int, error) {
|
|
w.mutex.Lock()
|
|
defer w.mutex.Unlock()
|
|
return w.Writer.Write(p)
|
|
}
|