feature-8 #9
1 changed files with 15 additions and 20 deletions
f-8: code: move Load in DB to internal lib #8
Signed-off-by: GnomeZworc <nicolas.boufidjeline@g3e.fr>
commit
97132550b0
|
|
@ -3,26 +3,12 @@ package main
|
||||||
import (
|
import (
|
||||||
"flag"
|
"flag"
|
||||||
"fmt"
|
"fmt"
|
||||||
"os"
|
|
||||||
"strings"
|
|
||||||
|
|
||||||
configuration "git.g3e.fr/syonad/two/internal/config/agent"
|
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"
|
"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() {
|
func main() {
|
||||||
conf_file := flag.String("conf", "/etc/two/agent.yml", "configuration file")
|
conf_file := flag.String("conf", "/etc/two/agent.yml", "configuration file")
|
||||||
vm_name := flag.String("vm_name", "", "Nom de la vm")
|
vm_name := flag.String("vm_name", "", "Nom de la vm")
|
||||||
|
|
@ -43,12 +29,21 @@ func main() {
|
||||||
}
|
}
|
||||||
fmt.Print(conf)
|
fmt.Print(conf)
|
||||||
|
|
||||||
DB = kv.InitDB(kv.Config{
|
db := kv.InitDB(kv.Config{
|
||||||
Path: conf.Database.Path,
|
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)
|
if *start {
|
||||||
|
nocloud.LoadNcCloudInDB(nocloud.Config{
|
||||||
os.Exit(0)
|
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)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue