English | 简体中文
A High Performance Atomic Counter for Concurrent Write-More-Read-Less Scenario in Go.
Similar to LongAdder in Java, or
ThreadCachedInt in folly,
In scenarios of high concurrent writes but few reads, it can provide dozens of times the write performance than sync/atomic
.
per 100 calls.
Under MacOS with M1 Pro:
goos: darwin
goarch: arm64
pkg: github.com/chen3feng/atomiccounter
BenchmarkNonAtomicAdd-10 47337121 22.14 ns/op
BenchmarkAtomicAdd-10 180942 6861 ns/op
BenchmarkCounter-10 14871549 81.02 ns/op
Under Linux:
goos: linux
goarch: amd64
pkg: github.com/chen3feng/atomiccounter
cpu: Intel(R) Xeon(R) Gold 6133 CPU @ 2.50GHz
BenchmarkNonAtomicAdd-16 9508723 135.3 ns/op
BenchmarkAtomicAdd-16 582798 2070 ns/op
BenchmarkCounter-16 4748263 263.1 ns/op
From top to bottom are writing time-consuming of non-atomic (and thus unsafe), atomic, and atomiccounter
.
It can be seen that in the case of high concurrent writes, atomiccounter
is only a few times more slower
than non-atomic writes, but much faster than atomic writes.
But it is much slower reads:
goos: darwin
goarch: arm64
pkg: github.com/chen3feng/atomiccounter
BenchmarkNonAtomicRead-10 1000000000 0.3112 ns/op
BenchmarkAtomicRead-10 1000000000 0.5336 ns/op
BenchmarkCounterRead-10 54609476 21.20 ns/op
In addition, each atomiccounter.Int64
object needs to consume 8K memory, so please only use it in a small number of
scenarios with a large number of concurrent writes but few reads, such as counting the number of requests.
Data race is one of the biggest performance killers in multi-core programs. For counters with a large number of writes, if ordinary atomic is used, the performance will be severely affected.
In scenarios with few reads, a common solution is to spread the writes across different variables and accumulate them when they are read. Such as Java's LongAdder and folly ThreadCachedInt, and per-cpu in the Linux kernel are all used this this method. Although the implementation details are different, the idea is similar.
At present, there is no well-known implementation for this kind of purpose in go, so I implemented this library.
To reduce memory footprint, multiple Int64
objects may share same memory chunk.
An int64 array of multiple sizes of CPU cache line size becomes a cell. A group of cells is called a chunk.
The size of the cell is an integer multiple of the cache line size of the CPU, and the first and last fields are paded with blanks of the size of the cache line size, thus avoiding false sharing.
The chunk.lastIndex
member is used to record the last unused index for allocating the Int64
object.
Each Int64
object contains 2 fields: the chunk pointer and the index in the cell, so multiple Int64
objects can share the same chunk,
but access elements with different indices in each cell.
The address of the last created chunk is recorded in the global variable lastChunk
. When an Int64
object is created,
its lastIndex
is increased. If it reachs the number of int64 in the cell,
it means that this chunk has been totally allocated and a new chunk needs to be created.
Please first understand Go's GMP scheduling model.
The best performance is to get the current subscript of M
in Go and directly access the corresponding cell
,
so that there will be no conflict between different M
s, and even avoid using atomic operations.
But I haven't found a way to get the M
's subscript.
Therefore, this implementation uses the hash of the address of M
as the subscript to access the cell,
and the measured effect is also quite good.
As long as the number of cells in each chunk is larger than the common number of CPU cores, the impact of hash collisions can be reduced, so that different M will have a high probability of accessing different cells.
When increasing the value of an Int64
object, the hash of current M
's' address is used as the
subscript to obtain the corresponding cell in the chunk.
Then use the Int64.index
member as a subscript to access the int64 array in this cell.
When reading, traverse the value indexed by Int64.index
in all cell arrays and accumulated the value.
import "github.com/chen3feng/atomiccounter"
Package atomiccounter provides an atomic counter for high throughput concurrent writing and rare reading scenario.
Example
package main
import (
"fmt"
"github.com/chen3feng/atomiccounter"
"sync"
)
func main() {
counter := atomiccounter.MakeInt64()
var wg sync.WaitGroup
for i := 0; i < 100; i++ {
wg.Add(1)
go func() {
counter.Inc()
wg.Done()
}()
}
wg.Wait()
fmt.Println(counter.Read())
counter.Set(0)
fmt.Println(counter.Read())
}
100
0
type Int64
Int64 is an int64 atomic counter.
type Int64 struct {
// contains filtered or unexported fields
}
func MakeInt64
func MakeInt64() Int64
MakeInt64 creates a new Int64 object. Int64 objects must be created by this function, simply initialized doesn't work.
func (*Int64) Add
func (c *Int64) Add(n int64)
Add adds n to the counter.
func (*Int64) Inc
func (c *Int64) Inc()
Inc adds 1 to the counter.
func (*Int64) Read
func (c *Int64) Read() int64
Read return the current value. it is a little slow so it should not be called frequently. Th result is not guaranteed to be accurate in race conditions.
func (*Int64) Set
func (c *Int64) Set(n int64)
Set set the value of the counter to n.
func (*Int64) Swap
func (c *Int64) Swap(n int64) int64
Swap returns the current value and swap it with n.
Generated by gomarkdoc