diff options
Diffstat (limited to 'tests/benchmarks/increment-stm/stm.go')
-rw-r--r-- | tests/benchmarks/increment-stm/stm.go | 33 |
1 files changed, 33 insertions, 0 deletions
diff --git a/tests/benchmarks/increment-stm/stm.go b/tests/benchmarks/increment-stm/stm.go new file mode 100644 index 0000000..9f12eb1 --- /dev/null +++ b/tests/benchmarks/increment-stm/stm.go @@ -0,0 +1,33 @@ +package stm + +import ( + "flag" +) + + + +var nFlag = flag.Int( + "n", + 1_000, + "The number of iterations to execute", +) + +func MainTest() { + flag.Parse() + n := *nFlag + + for i := 0; i < n; i++ { + // swap 1000 goroutines that each increment v by 1 + v := NewVar(0) + for i := 0; i < 1000; i++ { + go Atomically(VoidOperation(func(tx *Tx) { + v.Set(tx, v.Get(tx) + 1) + })) + } + + // wait for v to reach 1000 + Atomically(VoidOperation(func(tx *Tx) { + tx.Assert(v.Get(tx) == 1000) + })) + } +} |