109 lines
2.9 KiB
Go
109 lines
2.9 KiB
Go
package dispatcher
|
|
|
|
import (
|
|
"fmt"
|
|
"net"
|
|
"strings"
|
|
"time"
|
|
|
|
configuration "git.g3e.fr/syonad/two/internal/config/agent"
|
|
"git.g3e.fr/syonad/two/internal/state"
|
|
"git.g3e.fr/syonad/two/internal/vpc"
|
|
"git.g3e.fr/syonad/two/pkg/db/kv"
|
|
"github.com/dgraph-io/badger/v4"
|
|
)
|
|
|
|
type CreateVPCCommand struct {
|
|
Name string
|
|
CIDR string
|
|
}
|
|
|
|
func (c CreateVPCCommand) Key() string { return "vpc/" + c.Name }
|
|
|
|
func (c CreateVPCCommand) Prepare(db *badger.DB, _ *configuration.Config) error {
|
|
if _, err := kv.GetFromDB(db, "vpc/"+c.Name+"/state"); err == nil {
|
|
return fmt.Errorf("vpc %q already exists", c.Name)
|
|
}
|
|
if _, _, err := net.ParseCIDR(c.CIDR); err != nil {
|
|
return fmt.Errorf("invalid cidr %q: %w", c.CIDR, err)
|
|
}
|
|
if err := kv.AddInDB(db, "vpc/"+c.Name+"/cidr", c.CIDR); err != nil {
|
|
return err
|
|
}
|
|
return state.Set(db, c.Key(), state.Creating)
|
|
}
|
|
|
|
func (c CreateVPCCommand) Execute(db *badger.DB, _ *configuration.Config) error {
|
|
return vpc.CreateVPC(db, c.Name)
|
|
}
|
|
|
|
type DeleteVPCCommand struct {
|
|
Name string
|
|
}
|
|
|
|
func (c DeleteVPCCommand) Key() string { return "vpc/" + c.Name }
|
|
|
|
func (c DeleteVPCCommand) Prepare(db *badger.DB, _ *configuration.Config) error {
|
|
current, err := state.Get(db, c.Key())
|
|
if err != nil {
|
|
return fmt.Errorf("vpc %q not found", c.Name)
|
|
}
|
|
if !state.CanDelete(current) {
|
|
return fmt.Errorf("vpc %q cannot be deleted while %s", c.Name, current)
|
|
}
|
|
entries, err := kv.ListByPrefix(db, "subnet/")
|
|
if err != nil {
|
|
return fmt.Errorf("failed to list subnets: %w", err)
|
|
}
|
|
for key, value := range entries {
|
|
if !strings.HasSuffix(key, "/vpc") || value != c.Name {
|
|
continue
|
|
}
|
|
subnetName := strings.Split(key, "/")[1]
|
|
s, err := state.Get(db, "subnet/"+subnetName)
|
|
if err != nil || (s != state.Deleting && s != state.Deleted) {
|
|
return fmt.Errorf("subnet %q must be deleted before deleting vpc %q", subnetName, c.Name)
|
|
}
|
|
}
|
|
return state.Set(db, c.Key(), state.Deleting)
|
|
}
|
|
|
|
func (c DeleteVPCCommand) Execute(db *badger.DB, cfg *configuration.Config) error {
|
|
timeout := time.After(time.Duration(cfg.Dispatcher.TimeoutSeconds) * time.Second)
|
|
for {
|
|
entries, err := kv.ListByPrefix(db, "subnet/")
|
|
if err != nil {
|
|
return fmt.Errorf("failed to list subnets: %w", err)
|
|
}
|
|
pending := false
|
|
for key, value := range entries {
|
|
if strings.HasSuffix(key, "/vpc") && value == c.Name {
|
|
subnetName := strings.Split(key, "/")[1]
|
|
s, _ := state.Get(db, "subnet/"+subnetName)
|
|
if s == state.Deleting {
|
|
pending = true
|
|
break
|
|
}
|
|
}
|
|
}
|
|
if !pending {
|
|
break
|
|
}
|
|
select {
|
|
case <-timeout:
|
|
return fmt.Errorf("timed out waiting for subnets of vpc %q to be deleted", c.Name)
|
|
case <-time.After(time.Duration(cfg.Dispatcher.PollSeconds) * time.Second):
|
|
}
|
|
}
|
|
if err := vpc.DeleteVPC(db, c.Name); err != nil {
|
|
return err
|
|
}
|
|
current, err := state.Get(db, c.Key())
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if current == state.Deleted {
|
|
kv.DeleteInDB(db, "vpc/"+c.Name)
|
|
}
|
|
return nil
|
|
}
|