cmd/swarm-snapshot: if 2 nodes to create snapshot use connectChain (#…

…1709)
ethersphere · Aug 30, 2019 · f13e647 · f13e647
1 parent dec28cd
commit f13e647
Showing 1 changed file with 6 additions and 1 deletion.
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)
 	}