Counter

gistfile1.go

package counter

import "io"

type Counter struct {
	w io.Writer
	c int
}

func New(w io.Writer) (h *Counter) {
	h = new(Counter)
	h.w = w
	h.c = 0
	return
}

func (h *Counter) Write(p []byte) (n int, err error) {
	n, err = h.w.Write(p)
	h.c += n
	return
}

// Count: returns the total number of bytes that were 
// written to the underlying writer since the last call 
// to Clear
func (h *Counter) Count() (int) {
	return h.c
}

// Clear: zeroes the write byte counter
func (h *Counter) Clear() {
	h.c = 0
}