From e83bcc1097d49b46b79131e546f1270b9192cc05 Mon Sep 17 00:00:00 2001
From: Jimmy Hu <jimmy.hu@dexon.org>
Date: Wed, 27 Feb 2019 10:41:01 +0800
Subject: core: sync to latest core (#214)

* vendor: sync to latest core
* fix for single chain
---
 dex/peer.go | 37 +++++++++++++++++--------------------
 1 file changed, 17 insertions(+), 20 deletions(-)

(limited to 'dex/peer.go')

diff --git a/dex/peer.go b/dex/peer.go
index a157709f0..8ade9f152 100644
--- a/dex/peer.go
+++ b/dex/peer.go
@@ -119,9 +119,8 @@ const (
 )
 
 type peerLabel struct {
-	set     setType
-	chainID uint32
-	round   uint64
+	set   setType
+	round uint64
 }
 
 func (p peerLabel) String() string {
@@ -130,7 +129,7 @@ func (p peerLabel) String() string {
 	case dkgset:
 		t = fmt.Sprintf("DKGSet round: %d", p.round)
 	case notaryset:
-		t = fmt.Sprintf("NotarySet round: %d chain: %d", p.round, p.chainID)
+		t = fmt.Sprintf("NotarySet round: %d", p.round)
 	}
 	return t
 }
@@ -962,24 +961,22 @@ func (ps *peerSet) BuildConnection(round uint64) {
 		}
 	}
 
-	for chainID := uint32(0); chainID < ps.gov.GetNumChains(round); chainID++ {
-		notaryLabel := peerLabel{set: notaryset, chainID: chainID, round: round}
-		if _, ok := ps.label2Nodes[notaryLabel]; !ok {
-			notaryPKs, err := ps.gov.NotarySet(round, chainID)
-			if err != nil {
-				log.Error("get notary set fail",
-					"round", round, "chainID", chainID, "err", err)
-				continue
-			}
+	notaryLabel := peerLabel{set: notaryset, round: round}
+	if _, ok := ps.label2Nodes[notaryLabel]; !ok {
+		notaryPKs, err := ps.gov.NotarySet(round)
+		if err != nil {
+			log.Error("get notary set fail",
+				"round", round, "err", err)
+			return
+		}
 
-			nodes := ps.pksToNodes(notaryPKs)
-			ps.label2Nodes[notaryLabel] = nodes
+		nodes := ps.pksToNodes(notaryPKs)
+		ps.label2Nodes[notaryLabel] = nodes
 
-			if _, exists := nodes[ps.srvr.Self().ID().String()]; exists {
-				ps.buildDirectConn(notaryLabel)
-			} else {
-				ps.buildGroupConn(notaryLabel)
-			}
+		if _, exists := nodes[ps.srvr.Self().ID().String()]; exists {
+			ps.buildDirectConn(notaryLabel)
+		} else {
+			ps.buildGroupConn(notaryLabel)
 		}
 	}
 }
-- 
cgit v1.2.3