github/kubernetes
1
0
Fork 0
mirror of https://github.com/k3s-io/kubernetes.git synced 2025-07-28 05:57:25 +00:00
Code Issues Projects Releases Wiki Activity

address review feedback

Browse Source
This commit is contained in:
James DeFelice 2020-02-13 11:43:09 -06:00 committed by James DeFelice
parent 6f8dfdfb0e
commit a4230055f3
1 changed files with 5 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
pkg/kubelet/dockershim/helpers.go
View File

@ -397,13 +397,13 @@ func (d dockerOpt) GetKV() (string, string) {
return d.key, d.value
}
// writeLimiter limits the total output written across one or more streams.
type writeLimiter struct {
// sharedWriteLimiter limits the total output written across one or more streams.
type sharedWriteLimiter struct {
delegate io.Writer
limit *int64
}
func (w writeLimiter) Write(p []byte) (int, error) {
func (w sharedWriteLimiter) Write(p []byte) (int, error) {
if len(p) == 0 {
return 0, nil
}
@ -418,11 +418,7 @@ func (w writeLimiter) Write(p []byte) (int, error) {
}
n, err := w.delegate.Write(p)
if n > 0 {
newLimit := limit - int64(n)
for !atomic.CompareAndSwapInt64(w.limit, limit, newLimit) {
limit = atomic.LoadInt64(w.limit)
newLimit = limit - int64(n)
}
atomic.AddInt64(w.limit, -1*int64(n))
}
if err == nil && truncated {
err = errMaximumWrite
@ -434,7 +430,7 @@ func limitedWriter(w io.Writer, limit *int64) io.Writer {
if w == nil {
return nil
}
return &writeLimiter{
return &sharedWriteLimiter{
delegate: w,
limit: limit,
}