Compare commits
9 commits
main
...
feature-15
| Author | SHA1 | Date | |
|---|---|---|---|
|
bf00d74a26 |
|||
|
825fabb64a |
|||
|
40b29657ae |
|||
|
ff17645c62 |
|||
|
d9f2ec4e67 |
|||
|
74d7fc1425 |
|||
|
beed163b02 |
|||
|
049cabc489 |
|||
|
1e7575bc7d |
8 changed files with 69 additions and 31 deletions
|
|
@ -35,15 +35,19 @@ func main() {
|
||||||
defer db.Close()
|
defer db.Close()
|
||||||
|
|
||||||
if *start {
|
if *start {
|
||||||
metadata.StartMetadata(metadata.NoCloudConfig{
|
if err := metadata.StartMetadata(metadata.NoCloudConfig{
|
||||||
VpcName: *vpc,
|
VpcName: *vpc,
|
||||||
Name: *vm_name,
|
Name: *vm_name,
|
||||||
BindIP: *bind_ip,
|
BindIP: *bind_ip,
|
||||||
BindPort: *bind_port,
|
BindPort: *bind_port,
|
||||||
Password: *password,
|
Password: *password,
|
||||||
SSHKEY: *ssh_key,
|
SSHKEY: *ssh_key,
|
||||||
}, db, *dryrun)
|
}, db, *dryrun); err != nil {
|
||||||
|
fmt.Println(err)
|
||||||
|
}
|
||||||
} else if *stop {
|
} else if *stop {
|
||||||
metadata.StopMetadata(*vm_name, db, *dryrun)
|
if err := metadata.StopMetadata(*vm_name, db, *dryrun); err != nil {
|
||||||
|
fmt.Println(err)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
||||||
|
|
@ -1,26 +1,40 @@
|
||||||
package metadata
|
package metadata
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"fmt"
|
||||||
|
|
||||||
"git.g3e.fr/syonad/two/pkg/systemd"
|
"git.g3e.fr/syonad/two/pkg/systemd"
|
||||||
"github.com/dgraph-io/badger/v4"
|
"github.com/dgraph-io/badger/v4"
|
||||||
)
|
)
|
||||||
|
|
||||||
func StartMetadata(config NoCloudConfig, db *badger.DB, dryrun bool) {
|
func StartMetadata(config NoCloudConfig, db *badger.DB, dryrun bool) error {
|
||||||
service, _ := systemd.New()
|
service, err := systemd.New()
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("failed to connect to systemd: %w", err)
|
||||||
|
}
|
||||||
defer service.Close()
|
defer service.Close()
|
||||||
|
|
||||||
LoadNcCloudInDB(config, db)
|
LoadNcCloudInDB(config, db)
|
||||||
if !dryrun {
|
if !dryrun {
|
||||||
service.Start("metadata@" + config.Name)
|
if err := service.Start("metadata@" + config.Name + ".service"); err != nil {
|
||||||
|
return fmt.Errorf("failed to start metadata@%s: %w", config.Name, err)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func StopMetadata(vm_name string, db *badger.DB, dryrun bool) {
|
func StopMetadata(vm_name string, db *badger.DB, dryrun bool) error {
|
||||||
service, _ := systemd.New()
|
service, err := systemd.New()
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("failed to connect to systemd: %w", err)
|
||||||
|
}
|
||||||
defer service.Close()
|
defer service.Close()
|
||||||
|
|
||||||
UnLoadNoCloudInDB(vm_name, db)
|
UnLoadNoCloudInDB(vm_name, db)
|
||||||
if !dryrun {
|
if !dryrun {
|
||||||
service.Stop("metadata@" + vm_name)
|
if err := service.Stop("metadata@" + vm_name + ".service"); err != nil {
|
||||||
|
return fmt.Errorf("failed to stop metadata@%s: %w", vm_name, err)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
return nil
|
||||||
}
|
}
|
||||||
|
|
|
||||||
|
|
@ -2,6 +2,6 @@
|
||||||
|
|
||||||
package netns
|
package netns
|
||||||
|
|
||||||
func call(name string, fn func() error) error {
|
func call(_ string, fn func() error) error {
|
||||||
return fn()
|
return fn()
|
||||||
}
|
}
|
||||||
|
|
|
||||||
|
|
@ -3,12 +3,17 @@
|
||||||
package netns
|
package netns
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"fmt"
|
||||||
"os"
|
"os"
|
||||||
|
"runtime"
|
||||||
|
|
||||||
"golang.org/x/sys/unix"
|
"golang.org/x/sys/unix"
|
||||||
)
|
)
|
||||||
|
|
||||||
func create(name string) error {
|
func create(name string) error {
|
||||||
|
runtime.LockOSThread()
|
||||||
|
defer runtime.UnlockOSThread()
|
||||||
|
|
||||||
base := "/var/run/netns"
|
base := "/var/run/netns"
|
||||||
path := base + "/" + name
|
path := base + "/" + name
|
||||||
|
|
||||||
|
|
@ -16,6 +21,12 @@ func create(name string) error {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// si le fichier existe déjà, le démonter d'abord
|
||||||
|
if _, err := os.Stat(path); err == nil {
|
||||||
|
unix.Unmount(path, unix.MNT_DETACH)
|
||||||
|
os.Remove(path)
|
||||||
|
}
|
||||||
|
|
||||||
// fichier cible
|
// fichier cible
|
||||||
f, err := os.Create(path)
|
f, err := os.Create(path)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
|
|
@ -35,9 +46,12 @@ func create(name string) error {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
// bind mount du netns courant vers /var/run/netns/<name>
|
// bind mount du netns du thread courant vers /var/run/netns/<name>
|
||||||
|
// /proc/self/ns/net pointe vers le ns du processus (thread principal),
|
||||||
|
// pas du thread courant — il faut utiliser le tid explicitement
|
||||||
|
threadNsPath := fmt.Sprintf("/proc/self/task/%d/ns/net", unix.Gettid())
|
||||||
if err := unix.Mount(
|
if err := unix.Mount(
|
||||||
"/proc/self/ns/net",
|
threadNsPath,
|
||||||
path,
|
path,
|
||||||
"",
|
"",
|
||||||
unix.MS_BIND,
|
unix.MS_BIND,
|
||||||
|
|
|
||||||
|
|
@ -22,7 +22,7 @@ func CreateVPC(db *badger.DB, name string) error {
|
||||||
}
|
}
|
||||||
|
|
||||||
// create veth public for this netns
|
// create veth public for this netns
|
||||||
if err := netif.CreateVethToNetns("veth"+name+"ext", "vethpublicint", "/var/run/netns/"+name, 9000); err != nil {
|
if err := netif.CreateVethToNetns("vp-"+name+"-e", "vp-public-i", "/var/run/netns/"+name, 9000); err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
@ -34,24 +34,24 @@ func CreateVPC(db *badger.DB, name string) error {
|
||||||
}
|
}
|
||||||
|
|
||||||
// set veth to ext public bridge
|
// set veth to ext public bridge
|
||||||
if err := netif.BridgeSetMaster("veth"+name+"ext", "br-public"); err != nil {
|
if err := netif.BridgeSetMaster("vp-"+name+"-e", "br-public"); err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
// set veth to int public bridge
|
// set veth to int public bridge
|
||||||
if err := netns.Call(name, func() error {
|
if err := netns.Call(name, func() error {
|
||||||
return netif.BridgeSetMaster("vethpublicint", "br-public")
|
return netif.BridgeSetMaster("vp-public-i", "br-public")
|
||||||
}); err != nil {
|
}); err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
// set set ext veth up
|
// set set ext veth up
|
||||||
if err := netif.LinkSetUp("veth" + name + "ext"); err != nil {
|
if err := netif.LinkSetUp("vp-" + name + "-e"); err != nil {
|
||||||
return nil
|
return err
|
||||||
}
|
}
|
||||||
// set set int veth up
|
// set set int veth up
|
||||||
if err := netns.Call(name, func() error {
|
if err := netns.Call(name, func() error {
|
||||||
return netif.LinkSetUp("vethpublicint")
|
return netif.LinkSetUp("vp-public-i")
|
||||||
}); err != nil {
|
}); err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
|
||||||
|
|
@ -12,7 +12,7 @@ func DeleteVPC(db *badger.DB, name string) error {
|
||||||
if state, err := kv.GetFromDB(db, "vpc/"+name+"/state"); err != nil {
|
if state, err := kv.GetFromDB(db, "vpc/"+name+"/state"); err != nil {
|
||||||
return err
|
return err
|
||||||
} else if state == "deleting" {
|
} else if state == "deleting" {
|
||||||
if err := netif.DeleteLink(name + "-ext"); err != nil {
|
if err := netif.DeleteLink("vp-" + name + "-e"); err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -6,7 +6,8 @@ import (
|
||||||
|
|
||||||
func InitDB(conf Config, readonly bool) *badger.DB {
|
func InitDB(conf Config, readonly bool) *badger.DB {
|
||||||
opts := badger.DefaultOptions(conf.Path).
|
opts := badger.DefaultOptions(conf.Path).
|
||||||
WithReadOnly(readonly)
|
WithReadOnly(readonly).
|
||||||
|
WithBypassLockGuard(readonly)
|
||||||
opts.Logger = nil
|
opts.Logger = nil
|
||||||
opts.ValueLogFileSize = 10 << 20 // 10 Mo par fichier vlog
|
opts.ValueLogFileSize = 10 << 20 // 10 Mo par fichier vlog
|
||||||
opts.NumMemtables = 1
|
opts.NumMemtables = 1
|
||||||
|
|
|
||||||
|
|
@ -11,6 +11,7 @@ import (
|
||||||
|
|
||||||
const (
|
const (
|
||||||
defaultTimeout = 5 * time.Second
|
defaultTimeout = 5 * time.Second
|
||||||
|
jobTimeout = 30 * time.Second
|
||||||
jobMode = "replace"
|
jobMode = "replace"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
@ -28,10 +29,7 @@ type ServiceStatus struct {
|
||||||
|
|
||||||
// New crée une connexion D-Bus systemd (scope système)
|
// New crée une connexion D-Bus systemd (scope système)
|
||||||
func New() (*Manager, error) {
|
func New() (*Manager, error) {
|
||||||
ctx, cancel := context.WithTimeout(context.Background(), defaultTimeout)
|
conn, err := dbus.NewSystemConnectionContext(context.Background())
|
||||||
defer cancel()
|
|
||||||
|
|
||||||
conn, err := dbus.NewSystemConnectionContext(ctx)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
|
@ -57,17 +55,17 @@ func (m *Manager) Stop(service string) error {
|
||||||
}
|
}
|
||||||
|
|
||||||
func (m *Manager) job(method, service string) error {
|
func (m *Manager) job(method, service string) error {
|
||||||
ctx, cancel := context.WithTimeout(context.Background(), defaultTimeout)
|
callCtx, callCancel := context.WithTimeout(context.Background(), defaultTimeout)
|
||||||
defer cancel()
|
defer callCancel()
|
||||||
|
|
||||||
ch := make(chan string, 1)
|
ch := make(chan string, 1)
|
||||||
|
|
||||||
var err error
|
var err error
|
||||||
switch method {
|
switch method {
|
||||||
case "StartUnit":
|
case "StartUnit":
|
||||||
_, err = m.conn.StartUnitContext(ctx, service, jobMode, ch)
|
_, err = m.conn.StartUnitContext(callCtx, service, jobMode, ch)
|
||||||
case "StopUnit":
|
case "StopUnit":
|
||||||
_, err = m.conn.StopUnitContext(ctx, service, jobMode, ch)
|
_, err = m.conn.StopUnitContext(callCtx, service, jobMode, ch)
|
||||||
default:
|
default:
|
||||||
return errors.New("unsupported job method")
|
return errors.New("unsupported job method")
|
||||||
}
|
}
|
||||||
|
|
@ -76,9 +74,16 @@ func (m *Manager) job(method, service string) error {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
result := <-ch
|
waitCtx, waitCancel := context.WithTimeout(context.Background(), jobTimeout)
|
||||||
if result != "done" {
|
defer waitCancel()
|
||||||
return fmt.Errorf("%s %s failed: %s", method, service, result)
|
|
||||||
|
select {
|
||||||
|
case result := <-ch:
|
||||||
|
if result != "done" {
|
||||||
|
return fmt.Errorf("%s %s failed: %s", method, service, result)
|
||||||
|
}
|
||||||
|
case <-waitCtx.Done():
|
||||||
|
return fmt.Errorf("%s %s timed out after %s", method, service, jobTimeout)
|
||||||
}
|
}
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue