chore: verify if there is connection

babylonlabs-io · Oct 25, 2024 · ad2c6a2 · ad2c6a2
1 parent 205bdfa
commit ad2c6a2
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/finality-provider/cmd/fpd/daemon/daemon_commands.go b/finality-provider/cmd/fpd/daemon/daemon_commands.go
@@ -430,7 +430,7 @@ func runCommandAddFinalitySig(cmd *cobra.Command, args []string) error {
 		return err
 	}
 	defer func() {
-		fmt.Printf("start to clean up")
+		fmt.Printf("\nstart to clean up")
 		if err := cleanUp(); err != nil {
 			fmt.Printf("Failed to clean up grpc client: %v\n", err)
 		}

diff --git a/finality-provider/service/client/rpcclient.go b/finality-provider/service/client/rpcclient.go
@@ -25,6 +25,9 @@ func NewFinalityProviderServiceGRpcClient(remoteAddr string) (client *FinalityPr
 	}
 
 	cleanUp = func() error {
+		if conn == nil {
+			return nil
+		}
 		return conn.Close()
 	}