github.com/Cloud-Foundations/Dominator@v0.3.4/cmd/hyper-control/connectToVmManager.go (about)

     1  package main
     2  
     3  import (
     4  	"fmt"
     5  	"net"
     6  	"os"
     7  
     8  	"github.com/Cloud-Foundations/Dominator/lib/errors"
     9  	"github.com/Cloud-Foundations/Dominator/lib/log"
    10  	terminalclient "github.com/Cloud-Foundations/Dominator/lib/net/terminal/client"
    11  	proto "github.com/Cloud-Foundations/Dominator/proto/hypervisor"
    12  )
    13  
    14  func connectToVmManagerSubcommand(args []string,
    15  	logger log.DebugLogger) error {
    16  	if err := connectToVmManager(args[0], logger); err != nil {
    17  		return fmt.Errorf("error connecting to VM manager: %s", err)
    18  	}
    19  	return nil
    20  }
    21  
    22  func connectToVmManager(vmHostname string, logger log.DebugLogger) error {
    23  	if vmIP, hypervisor, err := lookupVmAndHypervisor(vmHostname); err != nil {
    24  		return err
    25  	} else {
    26  		return connectToVmManagerOnHypervisor(hypervisor, vmIP, logger)
    27  	}
    28  }
    29  
    30  func connectToVmManagerOnHypervisor(hypervisor string, ipAddr net.IP,
    31  	logger log.DebugLogger) error {
    32  	client, err := dialHypervisor(hypervisor)
    33  	if err != nil {
    34  		return err
    35  	}
    36  	defer client.Close()
    37  	conn, err := client.Call("Hypervisor.ConnectToVmManager")
    38  	if err != nil {
    39  		return err
    40  	}
    41  	defer conn.Close()
    42  	request := proto.ConnectToVmManagerRequest{IpAddress: ipAddr}
    43  	if err := conn.Encode(request); err != nil {
    44  		return err
    45  	}
    46  	if err := conn.Flush(); err != nil {
    47  		return err
    48  	}
    49  	var response proto.ConnectToVmManagerResponse
    50  	if err := conn.Decode(&response); err != nil {
    51  		return err
    52  	}
    53  	if err := errors.New(response.Error); err != nil {
    54  		return err
    55  	}
    56  	if err := terminalclient.StartTerminal(conn); err != nil {
    57  		fmt.Fprintln(os.Stderr, err)
    58  		os.Exit(1)
    59  	}
    60  	fmt.Fprint(os.Stderr, "\r")
    61  	return nil
    62  }