vm_shutdown.go

package ovirtclient

import (
	"fmt"
	"time"
)

func (o *oVirtClient) ShutdownVM(id VMID, force bool, retries ...RetryStrategy) (err error) {
	retries = defaultRetries(retries, defaultWriteTimeouts(o))
	err = retry(
		fmt.Sprintf("shutting down VM %s", id),
		o.logger,
		retries,
		func() error {
			_, err := o.conn.SystemService().VmsService().VmService(string(id)).Shutdown().Force(force).Send()
			return err
		})
	return
}

func (m *mockClient) ShutdownVM(id VMID, force bool, _ ...RetryStrategy) error {
	m.lock.Lock()
	defer m.lock.Unlock()
	if item, ok := m.vms[id]; ok {
		if (item.status == VMStatusSavingState || item.status == VMStatusRestoringState) && !force {
			return newError(EConflict, "VM is currently backing up or restoring.")
		}
		if item.status != VMStatusDown {
			item.status = VMStatusPoweringDown
			go func() {
				time.Sleep(2 * time.Second)
				m.lock.Lock()
				defer m.lock.Unlock()
				item.status = VMStatusDown
			}()
		}
		return nil
	}
	return newError(ENotFound, "vm with ID %s not found", id)
}