syonad/two
2
0
Fork 0
Code Issues 3 Pull requests 1 Projects Releases 2 Packages Wiki Activity Actions

feature-8 #9

Merged
nicolas.boufideline merged 27 commits from feature-8 into main 2026-01-04 16:51:23 +00:00
Conversation 0 Commits 27 Files changed 27 +21 -14
1 changed files with 21 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files
Showing only changes of commit df95d8b4a7 - Show all commits

f-8: code: use flags in db binary #8

Prev Next 
Signed-off-by: GnomeZworc <nicolas.boufidjeline@g3e.fr>
GnomeZworc 2025-12-26 23:40:03 +01:00
commit df95d8b4a7
Signed by: nicolas.boufideline
GPG key ID: 4406BBBF8845D632

35
cmd/db/main.go
View file

@ -1,6 +1,7 @@
package main
import (
"flag"
"fmt"
"os"
"strings"
@ -125,7 +126,13 @@ func printDB() {
}
func main() {
conf, err := configuration.LoadConfig("/etc/two/agent.yml")
conf_file := flag.String("conf", "/etc/two/agent.yml", "configuration file")
flag.Parse()
args := flag.Args()
conf, err := configuration.LoadConfig(*conf_file)
if err != nil {
fmt.Println(err)
return
@ -136,53 +143,53 @@ func main() {
})
defer DB.Close()
if len(os.Args) < 2 {
if len(args) < 1 {
fmt.Println("Usage: db <cmd> [args...]")
return
}
cmd := os.Args[1]
cmd := args[0]
switch cmd {
case "check_in_db":
if len(os.Args) != 4 {
if len(args) != 3 {
fmt.Println("Usage: check_in_db <db_name> <id>")
os.Exit(1)
}
ret := CheckInDB(os.Args[2], os.Args[3])
ret := CheckInDB(args[1], args[2])
os.Exit(ret)
case "add_in_db":
if len(os.Args) < 4 {
if len(args) < 3 {
fmt.Println("Usage: add_in_db <db_name> <line...>")
os.Exit(1)
}
line := strings.Join(os.Args[3:], ";")
if err := AddInDB(os.Args[2], line); err != nil {
line := strings.Join(args[2:], ";")
if err := AddInDB(args[1], line); err != nil {
fmt.Println("Error:", err)
os.Exit(1)
}
case "delete_in_db":
if len(os.Args) != 4 {
if len(args) != 3 {
fmt.Println("Usage: delete_in_db <db_name> <id>")
os.Exit(1)
}
if err := DeleteInDB(os.Args[2], os.Args[3]); err != nil {
if err := DeleteInDB(args[1], args[2]); err != nil {
fmt.Println("Error:", err)
os.Exit(1)
}
case "count_in_db":
if len(os.Args) != 4 {
if len(args) != 3 {
fmt.Println("Usage: count_in_db <db_name> <id>")
os.Exit(1)
}
count := CountInDB(os.Args[2], os.Args[3])
count := CountInDB(args[1], args[2])
fmt.Println(count)
case "get_from_db":
if len(os.Args) != 4 {
if len(args) != 3 {
fmt.Println("Usage: get_from_db <db_name> <id>")
os.Exit(1)
}
line, _ := GetFromDB(os.Args[2], os.Args[3])
line, _ := GetFromDB(args[1], args[2])
fmt.Println(line)
case "print":
printDB()