3a41b3f1ce
Maps rely on the keys being comparable. Using an interface type as the map key is dangerous, because some interface types are not comparable. I talked about this in my "Stupid Gopher Tricks" talk: https://talks.golang.org/2015/tricks.slide In this case, if the user-provided writer is backed by a slice (such as io.MultiWriter) then the code will panic at run time. Signed-off-by: Andrew Gerrand <adg@golang.org>
54 lines
1.1 KiB
Go
54 lines
1.1 KiB
Go
package broadcastwriter
|
|
|
|
import (
|
|
"io"
|
|
"sync"
|
|
)
|
|
|
|
// BroadcastWriter accumulate multiple io.WriteCloser by stream.
|
|
type BroadcastWriter struct {
|
|
mu sync.Mutex
|
|
writers []io.WriteCloser
|
|
}
|
|
|
|
// AddWriter adds new io.WriteCloser.
|
|
func (w *BroadcastWriter) AddWriter(writer io.WriteCloser) {
|
|
w.mu.Lock()
|
|
w.writers = append(w.writers, writer)
|
|
w.mu.Unlock()
|
|
}
|
|
|
|
// Write writes bytes to all writers. Failed writers will be evicted during
|
|
// this call.
|
|
func (w *BroadcastWriter) Write(p []byte) (n int, err error) {
|
|
w.mu.Lock()
|
|
var evict []int
|
|
for i, sw := range w.writers {
|
|
if n, err := sw.Write(p); err != nil || n != len(p) {
|
|
// On error, evict the writer
|
|
evict = append(evict, i)
|
|
}
|
|
}
|
|
for n, i := range evict {
|
|
w.writers = append(w.writers[:i-n], w.writers[i-n+1:]...)
|
|
}
|
|
w.mu.Unlock()
|
|
return len(p), nil
|
|
}
|
|
|
|
// Clean closes and removes all writers. Last non-eol-terminated part of data
|
|
// will be saved.
|
|
func (w *BroadcastWriter) Clean() error {
|
|
w.mu.Lock()
|
|
for _, sw := range w.writers {
|
|
sw.Close()
|
|
}
|
|
w.writers = nil
|
|
w.mu.Unlock()
|
|
return nil
|
|
}
|
|
|
|
// New creates a new BroadcastWriter.
|
|
func New() *BroadcastWriter {
|
|
return &BroadcastWriter{}
|
|
}
|