[multi-bls] validator sign and send message in parallel (#3054)

consensus/validator.go

@@ -3,6 +3,7 @@ package consensus
 import (
 	"bytes"
 	"encoding/hex"
+	"sync"
 	"time"
 
 	"github.com/ethereum/go-ethereum/common"
@@ -58,8 +59,12 @@ func (consensus *Consensus) onAnnounce(msg *msg_pb.Message) {
 }
 
 func (consensus *Consensus) prepare() {
+	var wg sync.WaitGroup
+	wg.Add(len(consensus.PubKey.PublicKey))
 	groupID := []nodeconfig.GroupID{nodeconfig.NewGroupIDByShardID(nodeconfig.ShardID(consensus.ShardID))}
 	for i, key := range consensus.PubKey.PublicKey {
+		go func(i int) {
+			defer wg.Done()
 			networkMessage, err := consensus.construct(msg_pb.MessageType_PREPARE, nil, key, consensus.priKey.PrivateKey[i])
 			if err != nil {
 				consensus.getLogger().Err(err).
@@ -81,7 +86,9 @@ func (consensus *Consensus) prepare() {
 						Msg("[OnAnnounce] Sent Prepare Message!!")
 				}
 			}
+		}(i)
 	}
+	wg.Wait()
 	consensus.getLogger().Debug().
 		Str("From", consensus.phase.String()).
 		Str("To", FBFTPrepare.String()).
@@ -207,7 +214,11 @@ func (consensus *Consensus) onPrepared(msg *msg_pb.Message) {
 	groupID := []nodeconfig.GroupID{
 		nodeconfig.NewGroupIDByShardID(nodeconfig.ShardID(consensus.ShardID)),
 	}
+	var wg sync.WaitGroup
+	wg.Add(len(consensus.PubKey.PublicKey))
 	for i, key := range consensus.PubKey.PublicKey {
+		go func(i int) {
+			defer wg.Done()
 			networkMessage, _ := consensus.construct(
 				msg_pb.MessageType_COMMIT,
 				commitPayload,
@@ -227,7 +238,9 @@ func (consensus *Consensus) onPrepared(msg *msg_pb.Message) {
 						Msg("[OnPrepared] Sent Commit Message!!")
 				}
 			}
+		}(i)
 	}
+	wg.Wait()
 	consensus.getLogger().Debug().
 		Str("From", consensus.phase.String()).
 		Str("To", FBFTCommit.String()).