Merge pull request #11 from travisperson/fix/node-connection-issues

fix: node connection issues

Author: travisperson

cmd/filsnap/cmds/create.go

@@ -2,9 +2,11 @@ package cmds
 
 import (
 	"context"
+	"errors"
 	"fmt"
 	"io"
 	"io/ioutil"
+	"syscall"
 	"time"
 
 	"github.com/filecoin-project/go-state-types/abi"
@@ -153,6 +155,11 @@ var cmdCreate = &cli.Command{
 		for _, addr := range addrs {
 			node, closer, err := CreateLotusClient(ctx, addr)
 			if err != nil {
+				if errors.Is(err, syscall.ECONNREFUSED) {
+					logger.Warnw("failed to dial node", "err", err)
+					continue
+				}
+
 				return err
 			}
 
@@ -161,6 +168,10 @@ var cmdCreate = &cli.Command{
 			nodes = append(nodes, node)
 		}
 
+		if len(nodes) == 0 {
+			return xerrors.Errorf("no nodes")
+		}
+
 		cm := consensus.NewConsensusManager(nodes)
 
 		same, err := cm.CheckGenesis(ctx)

pkg/export/export.go

@@ -128,6 +128,7 @@ func (e *Export) done() {
 }
 
 func (e *Export) Export(ctx context.Context) error {
+	defer e.done()
 	if err := e.node.Shutdown(ctx); err != nil {
 		return err
 	}
@@ -145,8 +146,6 @@ func (e *Export) Export(ctx context.Context) error {
 		return err
 	}
 
-	defer e.done()
-
 	var last bool
 	for b := range stream {
 		last = e.update(len(b)) == 0