40 lines
710 B
Go
40 lines
710 B
Go
|
package sync
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
store "github.com/micro/go-micro/store"
|
||
|
|
mem_store "github.com/micro/go-micro/store/memory"
|
||
|
|
mem_lock "github.com/micro/go-micro/sync/lock/memory"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestIterate(t *testing.T) {
|
||
|
|
s1 := mem_store.NewStore()
|
||
|
|
s2 := mem_store.NewStore()
|
||
|
|
recA := &store.Record{
|
||
|
|
Key: "A",
|
||
|
|
Value: nil,
|
||
|
|
}
|
||
|
|
recB := &store.Record{
|
||
|
|
Key: "B",
|
||
|
|
Value: nil,
|
||
|
|
}
|
||
|
|
s1.Write(recA)
|
||
|
|
s1.Write(recB)
|
||
|
|
s2.Write(recB)
|
||
|
|
s2.Write(recA)
|
||
|
|
|
||
|
|
f := func(key, val interface{}) error {
|
||
|
|
time.Sleep(1 * time.Millisecond)
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
l := mem_lock.NewLock()
|
||
|
|
m1 := NewMap(WithStore(s1), WithLock(l))
|
||
|
|
m2 := NewMap(WithStore(s2), WithLock(l))
|
||
|
|
go func() {
|
||
|
|
m2.Iterate(f)
|
||
|
|
}()
|
||
|
|
m1.Iterate(f)
|
||
|
|
}
|