Merge branch 'master' of ssh://github.com/micro/go-micro

This commit is contained in:
Asim Aslam 2019-11-03 16:12:24 +00:00
commit 81e9298be6
3 changed files with 45 additions and 1 deletions

View File

@ -1,4 +1,4 @@
// Package memoy provides a sync.Mutex implementation of the lock for local use // Package memory provides a sync.Mutex implementation of the lock for local use
package memory package memory
import ( import (

View File

@ -5,6 +5,7 @@ import (
"encoding/base64" "encoding/base64"
"encoding/json" "encoding/json"
"fmt" "fmt"
"sort"
"github.com/micro/go-micro/store" "github.com/micro/go-micro/store"
ckv "github.com/micro/go-micro/store/etcd" ckv "github.com/micro/go-micro/store/etcd"
@ -94,6 +95,10 @@ func (m *syncMap) Iterate(fn func(key, val interface{}) error) error {
return err return err
} }
sort.Slice(keyvals, func(i, j int) bool {
return keyvals[i].Key < keyvals[j].Key
})
for _, keyval := range keyvals { for _, keyval := range keyvals {
// lock // lock
if err := m.opts.Lock.Acquire(keyval.Key); err != nil { if err := m.opts.Lock.Acquire(keyval.Key); err != nil {

39
sync/map_test.go Normal file
View File

@ -0,0 +1,39 @@
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)
}