more mutex

synced-pool.go

@@ -28,6 +28,7 @@ type RemoteHost struct {
 }
 
 type SyncedPool struct {
+	mu          sync.RWMutex
 	Discovery   Discovery
 	listener    net.Listener
 	Hostname    string
@@ -296,7 +297,9 @@ func (p *SyncedPool) handleConnection(conn net.Conn) {
 				if r.Host == string(idAndHostParts[1]) {
 					found = true
 					log.Printf("Remote grain %s changed to %s\n", idAndHostParts[0], idAndHostParts[1])
+					p.mu.Lock()
 					p.remoteIndex[ToCartId(idAndHostParts[0])] = r.Pool
+					p.mu.Unlock()
 				}
 			}
 
@@ -446,7 +449,9 @@ func (p *SyncedPool) Process(id CartId, messages ...Message) ([]byte, error) {
 	_, ok := p.local.grains[id]
 	if !ok {
 		// check if remote grain exists
+		p.mu.RLock()
 		remoteGrain, ok := p.remoteIndex[id]
+		p.mu.RUnlock()
 		if ok {
 			remoteLookupCount.Inc()
 			return remoteGrain.Process(id, messages...)
@@ -465,7 +470,9 @@ func (p *SyncedPool) Get(id CartId) ([]byte, error) {
 	_, ok := p.local.grains[id]
 	if !ok {
 		// check if remote grain exists
+		p.mu.RLock()
 		remoteGrain, ok := p.remoteIndex[id]
+		p.mu.RUnlock()
 		if ok {
 			remoteLookupCount.Inc()
 			return remoteGrain.Get(id)