69 lines
1.8 KiB
Go
69 lines
1.8 KiB
Go
package vm
|
|
|
|
import (
|
|
"fmt"
|
|
"time"
|
|
|
|
configuration "git.g3e.fr/syonad/two/internal/config/agent"
|
|
"git.g3e.fr/syonad/two/internal/iptables"
|
|
"git.g3e.fr/syonad/two/internal/metadata"
|
|
"git.g3e.fr/syonad/two/internal/netif"
|
|
"git.g3e.fr/syonad/two/internal/netns"
|
|
"git.g3e.fr/syonad/two/internal/qmp"
|
|
"git.g3e.fr/syonad/two/pkg/db/kv"
|
|
|
|
"github.com/dgraph-io/badger/v4"
|
|
)
|
|
|
|
func StopVM(db *badger.DB, name string, cfg *configuration.Config) error {
|
|
state, err := kv.GetFromDB(db, "vm/"+name+"/state")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if state != "stopping" {
|
|
return nil
|
|
}
|
|
|
|
d, err := loadVM(db, name)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
socketPath := fmt.Sprintf("/tmp/%s.qmp-sock", name)
|
|
|
|
if _, err := qmp.Send(socketPath, []string{`{"execute":"system_powerdown"}`}); err != nil {
|
|
return fmt.Errorf("qmp system_powerdown: %w", err)
|
|
}
|
|
|
|
// attendre l'arrêt effectif de la VM ; forcer via quit après timeout
|
|
timeout := time.After(time.Duration(cfg.Dispatcher.TimeoutSeconds) * time.Second)
|
|
poll := time.Duration(cfg.Dispatcher.PollSeconds) * time.Second
|
|
stopped := false
|
|
for !stopped {
|
|
select {
|
|
case <-timeout:
|
|
qmp.Send(socketPath, []string{`{"execute":"quit"}`})
|
|
stopped = true
|
|
case <-time.After(poll):
|
|
if _, err := qmp.Send(socketPath, nil); err != nil {
|
|
stopped = true
|
|
}
|
|
}
|
|
}
|
|
|
|
if err := netns.Call(d.vpcName, func() error {
|
|
return iptables.DeleteMetadataRedirect(d.ip, d.gatewayIP, d.metadataPort)
|
|
}); err != nil {
|
|
return fmt.Errorf("delete metadata redirect: %w", err)
|
|
}
|
|
|
|
if err := metadata.StopMetadata(name, db, false); err != nil {
|
|
return fmt.Errorf("stop metadata: %w", err)
|
|
}
|
|
|
|
if err := netif.DeleteTap(d.tapID, d.vpcName); err != nil {
|
|
return fmt.Errorf("delete tap: %w", err)
|
|
}
|
|
|
|
return kv.AddInDB(db, "vm/"+name+"/state", "stopped")
|
|
}
|