cmd/swarm-snapshot: if 2 nodes to create snapshot use connectChain (ethersphere#1709)

aperonam · acud · commit f13e647ea6ac · 2019-08-30T20:01:05.000+05:30
diff --git a/cmd/swarm-snapshot/create.go b/cmd/swarm-snapshot/create.go
@@ -84,7 +84,12 @@ func createSnapshot(filename string, nodes int, services []string) (err error) {
 		return fmt.Errorf("add nodes: %v", err)
 	}
 
-	err = sim.Net.ConnectNodesRing(ids)
+	if nodes == 2 {
+		err = sim.Net.ConnectNodesChain(ids)
+	} else {
+		err = sim.Net.ConnectNodesRing(ids)
+	}
+
 	if err != nil {
 		return fmt.Errorf("connect nodes: %v", err)
 	}

Original file line number	Diff line number	Diff line change
`@@ -84,7 +84,12 @@ func createSnapshot(filename string, nodes int, services []string) (err error) {`
`84`	`84`	`return fmt.Errorf("add nodes: %v", err)`
`85`	`85`	`}`
`86`	`86`
`87`		`- err = sim.Net.ConnectNodesRing(ids)`
	`87`	`+ if nodes == 2 {`
	`88`	`+ err = sim.Net.ConnectNodesChain(ids)`
	`89`	`+ } else {`
	`90`	`+ err = sim.Net.ConnectNodesRing(ids)`
	`91`	`+ }`
	`92`	`+`
`88`	`93`	`if err != nil {`
`89`	`94`	`return fmt.Errorf("connect nodes: %v", err)`
`90`	`95`	`}`