Browse Source

address code review comments

Matthew R. Kasun 2 years ago
parent
commit
c6f9b17414
1 changed files with 4 additions and 2 deletions
  1. 4 2
      netclient/functions/connection.go

+ 4 - 2
netclient/functions/connection.go

@@ -24,8 +24,9 @@ func Connect(network string) error {
 	if err = wireguard.ApplyConf(&cfg.Node, cfg.Node.Interface, filePath); err != nil {
 	if err = wireguard.ApplyConf(&cfg.Node, cfg.Node.Interface, filePath); err != nil {
 		return err
 		return err
 	}
 	}
+	err = config.ModNodeConfig(&cfg.Node)
 	daemon.Restart()
 	daemon.Restart()
-	return config.ModNodeConfig(&cfg.Node)
+	return err
 }
 }
 
 
 // Disconnect - attempts to disconnect a node on given network
 // Disconnect - attempts to disconnect a node on given network
@@ -43,6 +44,7 @@ func Disconnect(network string) error {
 	if err = wireguard.ApplyConf(&cfg.Node, cfg.Node.Interface, filePath); err != nil {
 	if err = wireguard.ApplyConf(&cfg.Node, cfg.Node.Interface, filePath); err != nil {
 		return err
 		return err
 	}
 	}
+	err = config.ModNodeConfig(&cfg.Node)
 	daemon.Restart()
 	daemon.Restart()
-	return config.ModNodeConfig(&cfg.Node)
+	return err
 }
 }