f-14: code: add db usage in vpc creation #14

Signed-off-by: GnomeZworc <nicolas.boufidjeline@g3e.fr>

cmd/vpc/main.go

@@ -5,21 +5,39 @@ import (
 	"fmt"
 	"os"
 
+	configuration "git.g3e.fr/syonad/two/internal/config/agent"
 	"git.g3e.fr/syonad/two/internal/vpc"
+	"git.g3e.fr/syonad/two/pkg/db/kv"
+	"github.com/dgraph-io/badger/v4"
 )
 
 var (
 	netns     = flag.String("netns", "", "Network namespace à faire")
 	name      = flag.String("name", "", "interface name")
 	action    = flag.String("action", "", "Action a faire")
+	conf_file = flag.String("conf", "/etc/two/agent.yml", "configuration file")
 )
 
+var DB *badger.DB
+
 func main() {
 	flag.Parse()
 
+	conf, err := configuration.LoadConfig(*conf_file)
+	if err != nil {
+		fmt.Println(err)
+		return
+	}
+
+	DB = kv.InitDB(kv.Config{
+		Path: conf.Database.Path,
+	}, false)
+	defer DB.Close()
+
 	switch *action {
 	case "create":
-		if err := vpc.CreateVPC(*name); err != nil {
+		kv.AddInDB(DB, "vpc/"+*name+"/state", "creating")
+		if err := vpc.CreateVPC(DB, *name); err != nil {
 			fmt.Println(err)
 		}
 	case "delete":

internal/vpc/create.go

@@ -3,14 +3,20 @@ package vpc
 import (
 	"git.g3e.fr/syonad/two/internal/netif"
 	"git.g3e.fr/syonad/two/internal/netns"
+	"git.g3e.fr/syonad/two/pkg/db/kv"
+
+	"github.com/dgraph-io/badger/v4"
 )
 
-func CreateVPC(name string) error {
+func CreateVPC(db *badger.DB, name string) error {
 	// missing
 	// search data in db
 	//  change state in db
 
 	// create netns
+	if state, err := kv.GetFromDB(db, "vpc/"+name+"/state"); err != nil {
+		return err
+	} else if state == "creating" {
 		if err := netns.Create(name); err != nil {
 			return err
 		}
@@ -49,6 +55,7 @@ func CreateVPC(name string) error {
 		}); err != nil {
 			return err
 		}
-
+		kv.AddInDB(db, "vpc/"+name+"/state", "created")
+	}
 	return nil
 }