f-8: code: use db for database config #8

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

internal/metadata/server.go

@@ -1,15 +1,15 @@
 package metadata
 
 import (
-	"encoding/json"
 	"fmt"
-	"io/ioutil"
 	"log"
 	"net"
 	"net/http"
+	"strconv"
 	"time"
 
 	"git.g3e.fr/syonad/two/internal/netns"
+	"git.g3e.fr/syonad/two/pkg/db/kv"
 )
 
 var data NoCloudData
@@ -22,6 +22,48 @@ func getIP(r *http.Request) string {
 	return ip
 }
 
+func getFromDB(config Config) NoCloudData {
+	var netns_name string
+	var port int
+	var iface string
+
+	db := kv.InitDB(kv.Config{Path: config.ConfFile})
+
+	metadata, _ := kv.GetFromDB(db, "metadata/"+config.VmName+"/meta-data")
+	userdata, _ := kv.GetFromDB(db, "metadata/"+config.VmName+"/user-data")
+	networkconfig, _ := kv.GetFromDB(db, "metadata/"+config.VmName+"/network-config")
+	vendordata, _ := kv.GetFromDB(db, "metadata/"+config.VmName+"/vendor-data")
+
+	if config.Netns == "" {
+		netns_name, _ = kv.GetFromDB(db, "metadata/"+config.VmName+"/vpc")
+	} else {
+		netns_name = config.Netns
+	}
+
+	if config.Iface == "" {
+		iface, _ = kv.GetFromDB(db, "metadata/"+config.VmName+"/bind_ip")
+	} else {
+		iface = config.Iface
+	}
+
+	if config.Port == 0 {
+		sport, _ := kv.GetFromDB(db, "metadata/"+config.VmName+"/bind_port")
+		port, _ = strconv.Atoi(sport)
+	} else {
+		port = config.Port
+	}
+
+	return NoCloudData{
+		MetaData:      metadata,
+		UserData:      userdata,
+		NetworkConfig: networkconfig,
+		VendorData:    vendordata,
+		NetNs:         netns_name,
+		Iface:         iface,
+		Port:          port,
+	}
+}
+
 func rootHandler(w http.ResponseWriter, r *http.Request) {
 	ip := getIP(r)
 	path := r.URL.Path
@@ -47,24 +89,17 @@ func rootHandler(w http.ResponseWriter, r *http.Request) {
 }
 
 func StartServer(config Config) {
-	if config.Netns != "" {
-		if err := netns.Enter(config.Netns); err != nil {
+	data = getFromDB(config)
+
+	if data.NetNs != "" {
+		if err := netns.Enter(data.NetNs); err != nil {
 			log.Fatalf("Impossible d'entrer dans le netns: %v", err)
 		}
 	}
 
-	raw, err := ioutil.ReadFile(config.File)
-	if err != nil {
-		log.Fatalf("Erreur de lecture du fichier: %v", err)
-	}
-
-	if err := json.Unmarshal(raw, &data); err != nil {
-		log.Fatalf("Erreur de parsing JSON: %v", err)
-	}
-
 	http.HandleFunc("/", rootHandler)
 
-	address := fmt.Sprintf("%s:%d", config.Iface, config.Port)
+	address := fmt.Sprintf("%s:%d", data.Iface, data.Port)
 	log.Printf("Serveur NoCloud démarré sur http://%s/", address)
 	log.Fatal(http.ListenAndServe(address, nil))
 }

internal/metadata/struct.go

@@ -1,15 +1,20 @@
 package metadata
 
 type NoCloudData struct {
-	MetaData      string `json:"meta-data"`
-	UserData      string `json:"user-data"`
-	NetworkConfig string `json:"network-config"`
-	VendorData    string `json:"vendor-data"`
+	MetaData      string
+	UserData      string
+	NetworkConfig string
+	VendorData    string
+	NetNs         string
+	Iface         string
+	Port          int
 }
 
 type Config struct {
-	Netns string
-	File  string
-	Iface string
-	Port  int
+	Netns    string
+	File     string
+	Iface    string
+	Port     int
+	ConfFile string
+	VmName   string
 }