Compare commits

..

No commits in common. "a1f7c438881e666b5be10ae35789b524aff52a8b" and "105cc0b0f6457e67cafaaa909e1626f4e2d7e79e" have entirely different histories.

4 changed files with 7 additions and 44 deletions

View file

@ -1,54 +1,17 @@
package main
import (
"flag"
"fmt"
"os"
"strings"
configuration "git.g3e.fr/syonad/two/internal/config/agent"
"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))
})
}
var (
bin_name = os.Args[0]
)
func main() {
conf_file := flag.String("conf", "/etc/two/agent.yml", "configuration file")
vm_name := flag.String("vm_name", "", "Nom de la vm")
vpc := flag.String("vpc_name", "", "vpc name")
bind_ip := flag.String("ip", "", "bind ip")
bind_port := flag.String("port", "", "bind port")
ssh_key := flag.String("key", "", "Clef ssh")
password := flag.String("pass", "", "password user")
start := flag.Bool("start", false, "start metadata server")
stop := flag.Bool("stop", false, "stop metadata server")
flag.Parse()
conf, err := configuration.LoadConfig(*conf_file)
if err != nil {
fmt.Println(err)
return
}
fmt.Print(conf)
DB = kv.InitDB(kv.Config{
Path: conf.Database.Path,
})
defer DB.Close()
fmt.Printf("conf metadata for %s\n - this key %s\n - this password %s\n", *vm_name, *ssh_key, *password)
fmt.Printf("Start %s conf\n", bin_name)
os.Exit(0)
}

View file

@ -1,2 +0,0 @@
database:
path: "./data/"

View file

@ -17,7 +17,9 @@ func LoadConfig(path string) (*Config, error) {
v.SetDefault("database.path", "/var/lib/two/data/")
v.ReadInConfig()
if err := v.ReadInConfig(); err != nil {
return nil, err
}
var cfg Config
if err := v.Unmarshal(&cfg); err != nil {