github.com/cloud-foundations/dominator@v0.0.0-20221004181915-6e4fee580046/cmd/vm-control/snapshotVm.go (about)

     1  package main
     2  
     3  import (
     4  	"fmt"
     5  	"net"
     6  
     7  	"github.com/Cloud-Foundations/Dominator/lib/errors"
     8  	"github.com/Cloud-Foundations/Dominator/lib/log"
     9  	proto "github.com/Cloud-Foundations/Dominator/proto/hypervisor"
    10  )
    11  
    12  func snapshotVmSubcommand(args []string, logger log.DebugLogger) error {
    13  	if err := snapshotVm(args[0], logger); err != nil {
    14  		return fmt.Errorf("Error snapshotting VM: %s", err)
    15  	}
    16  	return nil
    17  }
    18  
    19  func snapshotVm(vmHostname string, logger log.DebugLogger) error {
    20  	if vmIP, hypervisor, err := lookupVmAndHypervisor(vmHostname); err != nil {
    21  		return err
    22  	} else {
    23  		return snapshotVmOnHypervisor(hypervisor, vmIP, logger)
    24  	}
    25  }
    26  
    27  func snapshotVmOnHypervisor(hypervisor string, ipAddr net.IP,
    28  	logger log.DebugLogger) error {
    29  	request := proto.SnapshotVmRequest{ipAddr, *forceIfNotStopped,
    30  		*snapshotRootOnly}
    31  	client, err := dialHypervisor(hypervisor)
    32  	if err != nil {
    33  		return err
    34  	}
    35  	defer client.Close()
    36  	var reply proto.SnapshotVmResponse
    37  	err = client.RequestReply("Hypervisor.SnapshotVm", request, &reply)
    38  	if err != nil {
    39  		return err
    40  	}
    41  	return errors.New(reply.Error)
    42  }