mirror of
https://github.com/tursom/GoCollections.git
synced 2025-03-22 16:00:06 +08:00
56 lines
909 B
Go
56 lines
909 B
Go
/*
|
|
* Copyright (c) 2022 tursom. All rights reserved.
|
|
* Use of this source code is governed by a GPL-3
|
|
* license that can be found in the LICENSE file.
|
|
*/
|
|
|
|
package atomic
|
|
|
|
import (
|
|
"fmt"
|
|
"sync"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestInt32_SetBit(t *testing.T) {
|
|
var i Int32
|
|
for j := 0; j < 32; j++ {
|
|
if i.SetBit(j, true) {
|
|
t.Fail()
|
|
}
|
|
if i.Load() != 1<<j {
|
|
t.Fail()
|
|
}
|
|
fmt.Println(i.Load())
|
|
if !i.SetBit(j, false) {
|
|
t.Fail()
|
|
}
|
|
if i.Load() != 0 {
|
|
t.Fail()
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestInt32_CompareAndSwapBit(t *testing.T) {
|
|
var i Int32
|
|
cond := sync.WaitGroup{}
|
|
cond.Add(1)
|
|
for k := 0; k < 128; k++ {
|
|
id := k
|
|
go func() {
|
|
cond.Wait()
|
|
for j := 0; j < 32; j++ {
|
|
for !i.CompareAndSwapBit(j, false, true) {
|
|
}
|
|
for !i.CompareAndSwapBit(j, true, false) {
|
|
}
|
|
}
|
|
fmt.Printf("%d finished\n", id)
|
|
}()
|
|
}
|
|
cond.Done()
|
|
time.Sleep(1 * time.Second)
|
|
fmt.Println(i.Load())
|
|
}
|