Compare commits
No commits in common. "a1f7c438881e666b5be10ae35789b524aff52a8b" and "105cc0b0f6457e67cafaaa909e1626f4e2d7e79e" have entirely different histories.
a1f7c43888
...
105cc0b0f6
4 changed files with 7 additions and 44 deletions
|
|
@ -1,54 +1,17 @@
|
||||||
package main
|
package main
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"flag"
|
|
||||||
"fmt"
|
"fmt"
|
||||||
"os"
|
"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
|
var (
|
||||||
|
bin_name = os.Args[0]
|
||||||
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")
|
|
||||||
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()
|
fmt.Printf("Start %s conf\n", bin_name)
|
||||||
|
|
||||||
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)
|
|
||||||
|
|
||||||
os.Exit(0)
|
os.Exit(0)
|
||||||
}
|
}
|
||||||
|
|
|
||||||
|
|
@ -1,2 +0,0 @@
|
||||||
database:
|
|
||||||
path: "./data/"
|
|
||||||
|
|
@ -17,7 +17,9 @@ func LoadConfig(path string) (*Config, error) {
|
||||||
|
|
||||||
v.SetDefault("database.path", "/var/lib/two/data/")
|
v.SetDefault("database.path", "/var/lib/two/data/")
|
||||||
|
|
||||||
v.ReadInConfig()
|
if err := v.ReadInConfig(); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
var cfg Config
|
var cfg Config
|
||||||
if err := v.Unmarshal(&cfg); err != nil {
|
if err := v.Unmarshal(&cfg); err != nil {
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue