126 lines
3.3 KiB
Go
126 lines
3.3 KiB
Go
package dispatcher
|
|
|
|
import (
|
|
"fmt"
|
|
"math/rand"
|
|
"strconv"
|
|
"strings"
|
|
"time"
|
|
|
|
configuration "git.g3e.fr/syonad/two/internal/config/agent"
|
|
"git.g3e.fr/syonad/two/internal/vm"
|
|
"git.g3e.fr/syonad/two/pkg/db/kv"
|
|
"github.com/dgraph-io/badger/v4"
|
|
)
|
|
|
|
type StartVMCommand struct {
|
|
Name string
|
|
Subnet string
|
|
IP string
|
|
VolumePath string
|
|
Memory int
|
|
CPUs int
|
|
UEFI bool
|
|
Password string
|
|
SSHKey string
|
|
}
|
|
|
|
func (c StartVMCommand) Prepare(db *badger.DB, _ *configuration.Config) error {
|
|
if _, err := kv.GetFromDB(db, "vm/"+c.Name+"/state"); err == nil {
|
|
return fmt.Errorf("vm %q already exists", c.Name)
|
|
}
|
|
subnetState, err := kv.GetFromDB(db, "subnet/"+c.Subnet+"/state")
|
|
if err != nil {
|
|
return fmt.Errorf("subnet %q not found", c.Subnet)
|
|
}
|
|
if subnetState == "deleting" || subnetState == "deleted" {
|
|
return fmt.Errorf("subnet %q is %s", c.Subnet, subnetState)
|
|
}
|
|
port, err := allocateMetadataPort(db)
|
|
if err != nil {
|
|
return fmt.Errorf("allocate metadata port: %w", err)
|
|
}
|
|
kv.AddInDB(db, "vm/"+c.Name+"/state", "starting")
|
|
kv.AddInDB(db, "vm/"+c.Name+"/subnet", c.Subnet)
|
|
kv.AddInDB(db, "vm/"+c.Name+"/ip", c.IP)
|
|
kv.AddInDB(db, "vm/"+c.Name+"/metadata_port", strconv.Itoa(port))
|
|
kv.AddInDB(db, "vm/"+c.Name+"/volume_path", c.VolumePath)
|
|
kv.AddInDB(db, "vm/"+c.Name+"/memory", strconv.Itoa(c.Memory))
|
|
kv.AddInDB(db, "vm/"+c.Name+"/cpus", strconv.Itoa(c.CPUs))
|
|
if c.UEFI {
|
|
kv.AddInDB(db, "vm/"+c.Name+"/uefi", "true")
|
|
}
|
|
if c.Password != "" {
|
|
kv.AddInDB(db, "vm/"+c.Name+"/password", c.Password)
|
|
}
|
|
if c.SSHKey != "" {
|
|
kv.AddInDB(db, "vm/"+c.Name+"/sshkey", c.SSHKey)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func allocateMetadataPort(db *badger.DB) (int, error) {
|
|
entries, err := kv.ListByPrefix(db, "vm/")
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
used := make(map[int]struct{})
|
|
for key, value := range entries {
|
|
if strings.HasSuffix(key, "/metadata_port") {
|
|
if p, err := strconv.Atoi(value); err == nil {
|
|
used[p] = struct{}{}
|
|
}
|
|
}
|
|
}
|
|
for range 100 {
|
|
p := rand.Intn(9000) + 1000
|
|
if _, taken := used[p]; !taken {
|
|
return p, nil
|
|
}
|
|
}
|
|
return 0, fmt.Errorf("no free metadata port available in [1000, 9999]")
|
|
}
|
|
|
|
func (c StartVMCommand) Execute(db *badger.DB, cfg *configuration.Config) error {
|
|
timeout := time.After(time.Duration(cfg.Dispatcher.TimeoutSeconds) * time.Second)
|
|
for {
|
|
state, err := kv.GetFromDB(db, "subnet/"+c.Subnet+"/state")
|
|
if err != nil {
|
|
return fmt.Errorf("subnet %q not found while waiting", c.Subnet)
|
|
}
|
|
if state == "created" {
|
|
break
|
|
}
|
|
select {
|
|
case <-timeout:
|
|
return fmt.Errorf("timed out waiting for subnet %q to be created", c.Subnet)
|
|
case <-time.After(time.Duration(cfg.Dispatcher.PollSeconds) * time.Second):
|
|
}
|
|
}
|
|
return vm.StartVM(db, c.Name, cfg)
|
|
}
|
|
|
|
type StopVMCommand struct {
|
|
Name string
|
|
}
|
|
|
|
func (c StopVMCommand) Prepare(db *badger.DB, _ *configuration.Config) error {
|
|
if _, err := kv.GetFromDB(db, "vm/"+c.Name+"/state"); err != nil {
|
|
return fmt.Errorf("vm %q not found", c.Name)
|
|
}
|
|
return kv.AddInDB(db, "vm/"+c.Name+"/state", "stopping")
|
|
}
|
|
|
|
func (c StopVMCommand) Execute(db *badger.DB, cfg *configuration.Config) error {
|
|
if err := vm.StopVM(db, c.Name, cfg); err != nil {
|
|
return err
|
|
}
|
|
state, err := kv.GetFromDB(db, "vm/"+c.Name+"/state")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if state == "stopped" {
|
|
kv.DeleteInDB(db, "vm/"+c.Name)
|
|
}
|
|
return nil
|
|
}
|