f-8: code: move Load in DB to internal lib #8

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

cmd/agent/main.go

@@ -5,10 +5,13 @@ import (
 	"os"
 )
 
+var (
+	bin_name = os.Args[0]
+)
+
 func main() {
-	bin_name := os.Args[0]
 
 	fmt.Printf("%s: Start process\n", bin_name)
 
-	os.Exit(5)
+	os.Exit(0)
 }

cmd/db/main.go

@@ -1,6 +1,7 @@
 package main
 
 import (
+	"flag"
 	"fmt"
 	"os"
 	"strings"
@@ -59,9 +60,7 @@ func AddInDB(dbName string, line string) error {
 func DeleteInDB(dbName, id string) error {
 	key := []byte(dbName + "/" + id + "/bash")
 
-	return DB.Update(func(txn *badger.Txn) error {
-		return txn.Delete(key)
-	})
+	return kv.DeleteInDB(DB, string(key))
 }
 
 func CountInDB(dbName, id string) int {
@@ -125,7 +124,13 @@ func printDB() {
 }
 
 func main() {
-	conf, err := configuration.LoadConfig("/etc/two/agent.yml")
+	conf_file := flag.String("conf", "/etc/two/agent.yml", "configuration file")
+
+	flag.Parse()
+
+	args := flag.Args()
+
+	conf, err := configuration.LoadConfig(*conf_file)
 	if err != nil {
 		fmt.Println(err)
 		return
@@ -136,53 +141,53 @@ func main() {
 	})
 	defer DB.Close()
 
-	if len(os.Args) < 2 {
+	if len(args) < 1 {
 		fmt.Println("Usage: db <cmd> [args...]")
 		return
 	}
 
-	cmd := os.Args[1]
+	cmd := args[0]
 
 	switch cmd {
 	case "check_in_db":
-		if len(os.Args) != 4 {
+		if len(args) != 3 {
 			fmt.Println("Usage: check_in_db <db_name> <id>")
 			os.Exit(1)
 		}
-		ret := CheckInDB(os.Args[2], os.Args[3])
+		ret := CheckInDB(args[1], args[2])
 		os.Exit(ret)
 	case "add_in_db":
-		if len(os.Args) < 4 {
+		if len(args) < 3 {
 			fmt.Println("Usage: add_in_db <db_name> <line...>")
 			os.Exit(1)
 		}
-		line := strings.Join(os.Args[3:], ";")
-		if err := AddInDB(os.Args[2], line); err != nil {
+		line := strings.Join(args[2:], ";")
+		if err := AddInDB(args[1], line); err != nil {
 			fmt.Println("Error:", err)
 			os.Exit(1)
 		}
 	case "delete_in_db":
-		if len(os.Args) != 4 {
+		if len(args) != 3 {
 			fmt.Println("Usage: delete_in_db <db_name> <id>")
 			os.Exit(1)
 		}
-		if err := DeleteInDB(os.Args[2], os.Args[3]); err != nil {
+		if err := DeleteInDB(args[1], args[2]); err != nil {
 			fmt.Println("Error:", err)
 			os.Exit(1)
 		}
 	case "count_in_db":
-		if len(os.Args) != 4 {
+		if len(args) != 3 {
 			fmt.Println("Usage: count_in_db <db_name> <id>")
 			os.Exit(1)
 		}
-		count := CountInDB(os.Args[2], os.Args[3])
+		count := CountInDB(args[1], args[2])
 		fmt.Println(count)
 	case "get_from_db":
-		if len(os.Args) != 4 {
+		if len(args) != 3 {
 			fmt.Println("Usage: get_from_db <db_name> <id>")
 			os.Exit(1)
 		}
-		line, _ := GetFromDB(os.Args[2], os.Args[3])
+		line, _ := GetFromDB(args[1], args[2])
 		fmt.Println(line)
 	case "print":
 		printDB()

cmd/meta_cli/main.go

@@ -3,26 +3,12 @@ package main
 import (
 	"flag"
 	"fmt"
-	"os"
-	"strings"
 
 	configuration "git.g3e.fr/syonad/two/internal/config/agent"
+	"git.g3e.fr/syonad/two/internal/load_db/nocloud"
 	"git.g3e.fr/syonad/two/pkg/db/kv"
-	"github.com/dgraph-io/badger/v4"
 )
 
-var DB *badger.DB
-
-func AddInDB(dbName string, line string) error {
-	// ID = partie avant le premier ';'
-	id := strings.Split(line, ";")[0] + "/bash"
-	key := []byte(dbName + "/" + id)
-
-	return DB.Update(func(txn *badger.Txn) error {
-		return txn.Set(key, []byte(line))
-	})
-}
-
 func main() {
 	conf_file := flag.String("conf", "/etc/two/agent.yml", "configuration file")
 	vm_name := flag.String("vm_name", "", "Nom de la vm")
@@ -43,12 +29,21 @@ func main() {
 	}
 	fmt.Print(conf)
 
-	DB = kv.InitDB(kv.Config{
+	db := kv.InitDB(kv.Config{
 		Path: conf.Database.Path,
 	})
-	defer DB.Close()
+	defer db.Close()
 
-	fmt.Printf("conf metadata for %s\n - this key %s\n - this password %s\n", *vm_name, *ssh_key, *password)
-
-	os.Exit(0)
+	if *start {
+		nocloud.LoadNcCloudInDB(nocloud.Config{
+			VpcName:  *vpc,
+			Name:     *vm_name,
+			BindIP:   *bind_ip,
+			BindPort: *bind_port,
+			Password: *password,
+			SSHKEY:   *ssh_key,
+		}, db)
+	} else if *stop {
+		nocloud.UnLoadNoCloudInDB(*vm_name, db)
+	}
 }

pkg/db/kv/deleteInDB.go

@@ -1,11 +1,44 @@
 package kv
 
 import (
+	"fmt"
+	"log"
+
 	"github.com/dgraph-io/badger/v4"
 )
 
-func DeleteInDB(db *badger.DB, key string) error {
+func deleteKey(db *badger.DB, key string) error {
 	return db.Update(func(txn *badger.Txn) error {
 		return txn.Delete([]byte(key))
 	})
 }
+
+func DeleteInDB(db *badger.DB, key string) error {
+
+	prefix := []byte(key + "/")
+
+	err := db.View(func(txn *badger.Txn) error {
+		opts := badger.DefaultIteratorOptions
+		opts.PrefetchValues = false
+
+		it := txn.NewIterator(opts)
+		defer it.Close()
+
+		for it.Seek(prefix); it.ValidForPrefix(prefix); it.Next() {
+			item := it.Item()
+			key := item.Key()
+
+			k := append([]byte{}, key...)
+			fmt.Println(string(k))
+			if err := deleteKey(db, string(k)); err != nil {
+				return err
+			}
+		}
+		return nil
+	})
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	return deleteKey(db, key)
+}