mirror of
https://github.com/genuinetools/reg.git
synced 2024-08-19 21:30:24 -04:00
50 lines
1.1 KiB
Go
50 lines
1.1 KiB
Go
package broadcaster // import "github.com/docker/docker/pkg/broadcaster"
|
|
|
|
import (
|
|
"io"
|
|
"sync"
|
|
)
|
|
|
|
// Unbuffered accumulates multiple io.WriteCloser by stream.
|
|
type Unbuffered struct {
|
|
mu sync.Mutex
|
|
writers []io.WriteCloser
|
|
}
|
|
|
|
// Add adds new io.WriteCloser.
|
|
func (w *Unbuffered) Add(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 *Unbuffered) 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 *Unbuffered) Clean() error {
|
|
w.mu.Lock()
|
|
for _, sw := range w.writers {
|
|
sw.Close()
|
|
}
|
|
w.writers = nil
|
|
w.mu.Unlock()
|
|
return nil
|
|
}
|