Add usage information

Signed-off-by: Flipez <code@brauser.io>

database.go

@@ -2,6 +2,7 @@ package main
 
 import (
  "context"
+ "encoding/json"
  "log"
 
  "github.com/hetznercloud/hcloud-go/hcloud"
@@ -13,6 +14,7 @@ type Database struct {
  Client *hcloud.Client
  Context context.Context
  Self *hcloud.SSHKey
+ NoInfo bool
 }
 
 func (d *Database) Init() {
@@ -42,6 +44,8 @@ func (d *Database) Fetch() *hcloud.SSHKey {
  log.Fatalf("database %s not found", d.Name)
  }
 
+ checkSize(d)
+
  return d.Self
 }
 
@@ -57,6 +61,9 @@ func (d *Database) Set(key, value string) bool {
 
  d.Store = database.Labels
 
+ log.Println("OK")
+ checkSize(d)
+
  return true
 }
 
@@ -78,3 +85,11 @@ func (d *Database) List() []string {
 
  return keys
 }
+
+func checkSize(db *Database) {
+ if !db.NoInfo {
+ jsonStr, _ := json.Marshal(db.Store)
+ log.Printf("[Info] Database usage: %.2f%%", float64(len(jsonStr))/float64(maxDBBytes)*100)
+ }
+}
+

main.go

@@ -11,6 +11,10 @@ import (
  "github.com/urfave/cli/v2"
 )
 
+var (
+ maxDBBytes = 1946
+)
+
 func main() {
  log.SetFlags(0)
 
@@ -23,7 +27,7 @@ func main() {
  Aliases: []string{"i"},
  Usage: "initializes a new database",
  Action: func(cCtx *cli.Context) error {
- database := setupDB(cCtx.String("db"))
+ database := setupDB(cCtx)
  database.Init()
  return nil
  },
@@ -33,7 +37,7 @@ func main() {
  Aliases: []string{"s"},
  Usage: "sets a key",
  Action: func(cCtx *cli.Context) error {
- database := setupDB(cCtx.String("db"))
+ database := setupDB(cCtx)
 
  key := cCtx.Args().First()
  val := cCtx.Args().Get(1)
@@ -51,7 +55,7 @@ func main() {
  Aliases: []string{"g"},
  Usage: "get a value from given key",
  Action: func(cCtx *cli.Context) error {
- database := setupDB(cCtx.String("db"))
+ database := setupDB(cCtx)
  fmt.Println(database.Get(cCtx.Args().First()))
  return nil
  },
@@ -61,7 +65,7 @@ func main() {
  Aliases: []string{"l"},
  Usage: "list all keys",
  Action: func(cCtx *cli.Context) error {
- database := setupDB(cCtx.String("db"))
+ database := setupDB(cCtx)
  keys := database.List()
  fmt.Println(strings.Join(keys, "\n"))
  return nil
@@ -74,6 +78,11 @@ func main() {
  Value: "0",
  Usage: "database to use",
  },
+ &cli.BoolFlag{
+ Name: "no-info",
+ Value: false,
+ Usage: "Do not print db usage information",
+ },
  },
  }
 
@@ -83,12 +92,15 @@ func main() {
 
 }
 
-func setupDB(name string) Database {
+func setupDB(ctx *cli.Context) Database {
  token := os.Getenv("HCLOUD_TOKEN")
+ name := ctx.String("db")
+ noInfo := ctx.Bool("no-info")
 
  return Database{
  Client: hcloud.NewClient(hcloud.WithToken(token)),
  Context: context.Background(),
  Name: fmt.Sprintf("hkv-%s", name),
+ NoInfo: noInfo,
  }
 }