diff options
Diffstat (limited to 'simulation')
| -rw-r--r-- | simulation/app.go | 7 |
1 files changed, 3 insertions, 4 deletions
diff --git a/simulation/app.go b/simulation/app.go index 0330f07..a6f9a57 100644 --- a/simulation/app.go +++ b/simulation/app.go @@ -58,6 +58,7 @@ func (a *simApp) addBlock(block *types.Block) { a.blockByHashMutex.Lock() defer a.blockByHashMutex.Unlock() + // TODO(jimmy-dexon) : Remove block in this hash if it's no longer needed. a.blockByHash[block.Hash] = block } @@ -95,11 +96,11 @@ func (a *simApp) StronglyAcked(blockHash common.Hash) { // TotalOrderingDeliver is called when blocks are delivered by the total // ordering algorithm. func (a *simApp) TotalOrderingDeliver(blockHashes common.Hashes, early bool) { - a.blockByHashMutex.Lock() - defer a.blockByHashMutex.Unlock() now := time.Now() blocks := make([]*types.Block, len(blockHashes)) + a.blockByHashMutex.RLock() + defer a.blockByHashMutex.RUnlock() for idx := range blockHashes { blocks[idx] = a.blockByHash[blockHashes[idx]] } @@ -115,8 +116,6 @@ func (a *simApp) TotalOrderingDeliver(blockHashes common.Hashes, early bool) { confirmLatency = append(confirmLatency, now.Sub(block.Timestamps[a.ValidatorID])) } - // TODO(jimmy-dexon) : Remove block in this hash if it's no longer needed. - a.blockByHash[block.Hash] = block for hash := range block.Acks { for _, blockHash := range a.getAckedBlocks(hash) { payload = append(payload, TimestampMessage{ |