55 lines
958 B
Go
55 lines
958 B
Go
// Get/set arbitrary flags in the Weave Net persistence DB
|
|
package main
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/weaveworks/weave/db"
|
|
)
|
|
|
|
const dbFlagPrefix = "flag:"
|
|
|
|
func getDBFlag(args []string) error {
|
|
if len(args) != 2 {
|
|
cmdUsage("check-db-flag", "<db-prefix> <flag-name>")
|
|
}
|
|
dbPrefix := args[0]
|
|
flagName := args[1]
|
|
|
|
d, err := db.NewBoltDBReadOnly(dbPrefix)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer d.Close()
|
|
var value string
|
|
nameFound, err := d.Load(dbFlagPrefix+flagName, &value)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if !nameFound {
|
|
return fmt.Errorf("")
|
|
}
|
|
fmt.Print(value)
|
|
return nil
|
|
}
|
|
|
|
func setDBFlag(args []string) error {
|
|
if len(args) != 3 {
|
|
cmdUsage("set-db-flag", "<db-prefix> <flag-name> <flag-value>")
|
|
}
|
|
dbPrefix := args[0]
|
|
flagName := args[1]
|
|
value := args[2]
|
|
|
|
d, err := db.NewBoltDB(dbPrefix)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer d.Close()
|
|
err = d.Save(dbFlagPrefix+flagName, value)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|