cmd/microcloud: Trigger StopSession API calls to the joiner members i…

…f the initiator does not finish properly after initiating a session

Signed-off-by: Gabriel Mougard <gabriel.mougard@canonical.com>

cmd/microcloud/main_init.go

@@ -214,13 +214,32 @@ func (c *initConfig) RunInteractive(cmd *cobra.Command, args []string) error {
 		services[s.Type()] = version
 	}
 
+	var reverter *revert.Reverter
 	if c.setupMany {
 		err = c.runSession(context.Background(), s, types.SessionInitiating, c.sessionTimeout, func(gw *cloudClient.WebsocketGateway) error {
 			return c.initiatingSession(gw, s, services, "", nil)
 		})
 		if err != nil {
 			return err
 		}
+
+		reverter = revert.New()
+		defer reverter.Fail()
+
+		reverter.Add(func() {
+			// Stop each joiner member session.
+			cloud := s.Services[types.MicroCloud].(*service.CloudService)
+			for peer, system := range c.systems {
+				if system.ServerInfo.Name == "" || system.ServerInfo.Name == c.name {
+					continue
+				}
+
+				err = cloud.StopJoinerSession(context.Background(), system.ServerInfo.Address, "Initiator cluster member aborted")
+				if err != nil {
+					logger.Error("Failed to stop joiner session", logger.Ctx{"joiner": peer, "error": err})
+				}
+			}
+		})
 	}
 
 	state, err := s.CollectSystemInformation(context.Background(), multicast.ServerInfo{Name: c.name, Address: c.address, Services: services})
@@ -282,6 +301,10 @@ func (c *initConfig) RunInteractive(cmd *cobra.Command, args []string) error {
 		return err
 	}
 
+	if c.setupMany {
+		reverter.Success()
+	}
+
 	return nil
 }