47 lines
958 B
Go
47 lines
958 B
Go
|
// Copyright © 2016 Steve Francia <spf@spf13.com>.
|
||
|
|
//
|
||
|
|
// Use of this source code is governed by an MIT-style
|
||
|
|
// license that can be found in the LICENSE file.
|
||
|
|
|
||
|
|
package jwalterweatherman
|
||
|
|
|
||
|
|
import (
|
||
|
|
"io"
|
||
|
|
"sync/atomic"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Counter is an io.Writer that increments a counter on Write.
|
||
|
|
type Counter struct {
|
||
|
|
count uint64
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c *Counter) incr() {
|
||
|
|
atomic.AddUint64(&c.count, 1)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Reset resets the counter.
|
||
|
|
func (c *Counter) Reset() {
|
||
|
|
atomic.StoreUint64(&c.count, 0)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Count returns the current count.
|
||
|
|
func (c *Counter) Count() uint64 {
|
||
|
|
return atomic.LoadUint64(&c.count)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c *Counter) Write(p []byte) (n int, err error) {
|
||
|
|
c.incr()
|
||
|
|
return len(p), nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// LogCounter creates a LogListener that counts log statements >= the given threshold.
|
||
|
|
func LogCounter(counter *Counter, t1 Threshold) LogListener {
|
||
|
|
return func(t2 Threshold) io.Writer {
|
||
|
|
if t2 < t1 {
|
||
|
|
// Not interested in this threshold.
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
return counter
|
||
|
|
}
|
||
|
|
}
|