Merge pull request #182 from bstasyszyn/181

fix: Call extension AddPayload when receiving a block from Gossip
trustbloc · Apr 20, 2020 · ba9a412 · ba9a412
2 parents 8bc4d7f + da133f2
commit ba9a412
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/gossip/state/state.go b/gossip/state/state.go
@@ -323,7 +323,7 @@ func (s *GossipStateProviderImpl) receiveAndQueueGossipMessages(ch <-chan *proto
 
 			dataMsg := msg.GetDataMsg()
 			if dataMsg != nil {
-				if err := s.addPayload(dataMsg.GetPayload(), nonBlocking); err != nil {
+				if err := s.extension.AddPayload(s.addPayload)(dataMsg.GetPayload(), s.blockingMode); err != nil {
 					logger.Warningf("Block [%d] received from gossip wasn't added to payload buffer: %v", dataMsg.Payload.SeqNum, err)
 					return
 				}