f-8: code: add module to load nocloud in db #8
Signed-off-by: GnomeZworc <nicolas.boufidjeline@g3e.fr>
This commit is contained in:
parent
b8796d0810
commit
13a5f4d3f9
6 changed files with 82 additions and 0 deletions
50
internal/load_db/nocloud/render.go
Normal file
50
internal/load_db/nocloud/render.go
Normal file
|
|
@ -0,0 +1,50 @@
|
|||
package nocloud
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"embed"
|
||||
"text/template"
|
||||
|
||||
"git.g3e.fr/syonad/two/pkg/db/kv"
|
||||
"github.com/dgraph-io/badger/v4"
|
||||
)
|
||||
|
||||
//go:embed templates/*.tmpl
|
||||
var templateFS embed.FS
|
||||
|
||||
func renderConfig(path string, cfg Config) (string, error) {
|
||||
tpl, err := template.ParseFS(templateFS, path)
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
|
||||
var buf bytes.Buffer
|
||||
if err := tpl.Execute(&buf, cfg); err != nil {
|
||||
return "", err
|
||||
}
|
||||
|
||||
return buf.String(), nil
|
||||
}
|
||||
|
||||
var DB *badger.DB
|
||||
|
||||
func LoadNcCloudInDB(config Config, db *badger.DB) {
|
||||
meta_data, _ := renderConfig("templates/meta-data.tmpl", config)
|
||||
user_data, _ := renderConfig("templates/user-data.tmpl", config)
|
||||
network_config, _ := renderConfig("templates/network-config.tmpl", config)
|
||||
vendor_data, _ := renderConfig("templates/vendor-data.tmpl", config)
|
||||
|
||||
DB = db
|
||||
|
||||
kv.AddInDB(DB, "metadata/"+config.Name+"/meta-data", meta_data)
|
||||
kv.AddInDB(DB, "metadata/"+config.Name+"/user-data", user_data)
|
||||
kv.AddInDB(DB, "metadata/"+config.Name+"/network-config", network_config)
|
||||
kv.AddInDB(DB, "metadata/"+config.Name+"/vendor-data", vendor_data)
|
||||
kv.AddInDB(DB, "metadata/"+config.Name+"/vpc", config.VpcName)
|
||||
kv.AddInDB(DB, "metadata/"+config.Name+"/bind_ip", config.BindIP)
|
||||
kv.AddInDB(DB, "metadata/"+config.Name+"/bind_port", config.BindPort)
|
||||
}
|
||||
|
||||
func UnLoadNoCloudInDB(vm_name string, db *badger.DB) {
|
||||
kv.DeleteInDB(DB, "metadata/"+vm_name)
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue