Sha256: 1c78d61110db6a727de206334844e2a292c95536d14e38c01a2cd06abe074f3c
Contents?: true
Size: 1.65 KB
Versions: 122
Compression:
Stored size: 1.65 KB
Contents
package paasio
import (
"io"
"sync"
)
// NewWriteCounter returns an implementation of WriteCounter. Calls to
// w.Write() are not guaranteed to be synchronized.
func NewWriteCounter(w io.Writer) WriteCounter {
return &writeCounter{
w: w,
counter: newCounter(),
}
}
// NewReadCounter returns an implementation of ReadCounter. Calls to
// r.Read() are not guaranteed to be synchronized.
func NewReadCounter(r io.Reader) ReadCounter {
return &readCounter{
r: r,
counter: newCounter(),
}
}
// NewReadWriteCounter returns an implementation of ReadWriteCounter.
// Calls to rw.Write() and rw.Read() are not guaranteed to be synchronized.
func NewReadWriteCounter(rw io.ReadWriter) ReadWriteCounter {
return &rwCounter{
NewWriteCounter(rw),
NewReadCounter(rw),
}
}
type readCounter struct {
r io.Reader
counter
}
func (rc *readCounter) Read(p []byte) (int, error) {
m, err := rc.r.Read(p)
rc.addBytes(m)
return m, err
}
func (rc *readCounter) ReadCount() (n int64, nops int) {
return rc.count()
}
type writeCounter struct {
w io.Writer
counter
}
func (wc *writeCounter) Write(p []byte) (int, error) {
m, err := wc.w.Write(p)
wc.addBytes(m)
return m, err
}
func (wc *writeCounter) WriteCount() (n int64, nops int) {
return wc.count()
}
type counter struct {
bytes int64
ops int
mutex *sync.Mutex
}
func newCounter() counter {
return counter{mutex: new(sync.Mutex)}
}
func (c *counter) addBytes(n int) {
c.mutex.Lock()
defer c.mutex.Unlock()
c.bytes += int64(n)
c.ops++
}
func (c *counter) count() (int64, int) {
c.mutex.Lock()
defer c.mutex.Unlock()
return c.bytes, c.ops
}
type rwCounter struct {
WriteCounter
ReadCounter
}
Version data entries
122 entries across 122 versions & 1 rubygems