代码清理
This commit is contained in:
@@ -7,12 +7,12 @@ import (
|
||||
|
||||
type CountWaitGroup struct {
|
||||
wg sync.WaitGroup
|
||||
num atomic.Uint64
|
||||
num atomic.Int64
|
||||
}
|
||||
|
||||
func (c *CountWaitGroup) Add(delta uint64) {
|
||||
func (c *CountWaitGroup) Add(delta uint) {
|
||||
c.wg.Add(int(delta))
|
||||
c.num.Add(delta)
|
||||
c.num.Add(int64(delta))
|
||||
}
|
||||
|
||||
func (c *CountWaitGroup) Done() {
|
||||
@@ -25,5 +25,5 @@ func (c *CountWaitGroup) Wait() {
|
||||
}
|
||||
|
||||
func (c *CountWaitGroup) Count() uint64 {
|
||||
return c.num.Load()
|
||||
return uint64(c.num.Load())
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user