111 lines
1.3 KiB
Go
111 lines
1.3 KiB
Go
// counter
|
|
package main
|
|
|
|
import (
|
|
"sync"
|
|
)
|
|
|
|
type Counter struct {
|
|
mu sync.Mutex
|
|
user int
|
|
log int
|
|
rem int
|
|
err int
|
|
dup int
|
|
wg []int
|
|
}
|
|
|
|
func NewCounter() *Counter {
|
|
return &Counter{
|
|
user: 0,
|
|
log: 0,
|
|
err: 0,
|
|
rem: 0,
|
|
dup: 0,
|
|
wg: make([]int, opts.Concurrent),
|
|
}
|
|
}
|
|
|
|
func (c *Counter) AddUser() {
|
|
c.mu.Lock()
|
|
c.user++
|
|
c.mu.Unlock()
|
|
}
|
|
|
|
func (c *Counter) AddDuplicate() {
|
|
c.mu.Lock()
|
|
c.dup++
|
|
c.mu.Unlock()
|
|
}
|
|
|
|
func (c *Counter) AddLog(add int) {
|
|
c.mu.Lock()
|
|
c.log += add
|
|
c.mu.Unlock()
|
|
}
|
|
|
|
func (c *Counter) AddRem(add int) {
|
|
c.mu.Lock()
|
|
c.rem += add
|
|
c.mu.Unlock()
|
|
}
|
|
|
|
func (c *Counter) AddWG(id int) {
|
|
c.mu.Lock()
|
|
c.wg[id]++
|
|
c.mu.Unlock()
|
|
}
|
|
|
|
func (c *Counter) AddErr() {
|
|
c.mu.Lock()
|
|
c.err++
|
|
c.mu.Unlock()
|
|
}
|
|
|
|
func (c *Counter) RemWG(id int) {
|
|
c.mu.Lock()
|
|
c.wg[id]--
|
|
c.mu.Unlock()
|
|
}
|
|
|
|
func (c *Counter) ValUser() (ret int) {
|
|
c.mu.Lock()
|
|
ret = c.user
|
|
c.mu.Unlock()
|
|
return
|
|
}
|
|
|
|
func (c *Counter) ValDup() (ret int) {
|
|
c.mu.Lock()
|
|
ret = c.dup
|
|
c.mu.Unlock()
|
|
return
|
|
}
|
|
|
|
func (c *Counter) ValLog() (ret int) {
|
|
c.mu.Lock()
|
|
ret = c.log
|
|
c.mu.Unlock()
|
|
return
|
|
}
|
|
|
|
func (c *Counter) ValErr() (ret int) {
|
|
c.mu.Lock()
|
|
ret = c.err
|
|
c.mu.Unlock()
|
|
return
|
|
}
|
|
|
|
func (c *Counter) ValRem() (ret int) {
|
|
c.mu.Lock()
|
|
ret = c.rem
|
|
c.mu.Unlock()
|
|
return
|
|
}
|
|
|
|
func (c *Counter) ValWG(id int) (ret int) {
|
|
c.mu.Lock()
|
|
ret = c.wg[id]
|
|
c.mu.Unlock()
|
|
return
|
|
}
|