Просмотр исходного кода

Fix race condition on MockBroker.history

Sebastien Launay 10 лет назад
Родитель
Сommit
7340cab460
1 измененных файлов с 2 добавлено и 0 удалено
  1. 2 0
      mockbroker.go

+ 2 - 0
mockbroker.go

@@ -226,7 +226,9 @@ func (b *MockBroker) handleRequests(conn net.Conn, idx int, wg *sync.WaitGroup)
 			b.serverError(err)
 			b.serverError(err)
 			break
 			break
 		}
 		}
+		b.lock.Lock()
 		requestResponse.ResponseSize = len(resHeader) + len(encodedRes)
 		requestResponse.ResponseSize = len(resHeader) + len(encodedRes)
+		b.lock.Unlock()
 	}
 	}
 	Logger.Printf("*** mockbroker/%d/%d: connection closed, err=%v", b.BrokerID(), idx, err)
 	Logger.Printf("*** mockbroker/%d/%d: connection closed, err=%v", b.BrokerID(), idx, err)
 }
 }