diff options
Diffstat (limited to 'cmd/bolt/main.go')
| -rw-r--r-- | cmd/bolt/main.go | 220 |
1 files changed, 31 insertions, 189 deletions
diff --git a/cmd/bolt/main.go b/cmd/bolt/main.go index 9a51ec5..6b28060 100644 --- a/cmd/bolt/main.go +++ b/cmd/bolt/main.go @@ -5,9 +5,7 @@ import ( "fmt" "log" "os" - "strconv" - "github.com/boltdb/bolt" "github.com/codegangsta/cli" ) @@ -21,208 +19,52 @@ func NewApp() *cli.App { app := cli.NewApp() app.Name = "bolt" app.Usage = "BoltDB toolkit" + app.Version = "0.1.0" app.Commands = []cli.Command{ { - Name: "get", - Usage: "Retrieve a value for given key in a bucket", - Action: GetCommand, + Name: "get", + Usage: "Retrieve a value for given key in a bucket", + Action: func(c *cli.Context) { + path, name, key := c.Args().Get(0), c.Args().Get(1), c.Args().Get(2) + Get(path, name, key) + }, }, { - Name: "set", - Usage: "Sets a value for given key in a bucket", - Action: SetCommand, + Name: "set", + Usage: "Sets a value for given key in a bucket", + Action: func(c *cli.Context) { + path, name, key, value := c.Args().Get(0), c.Args().Get(1), c.Args().Get(2), c.Args().Get(3) + Set(path, name, key, value) + }, }, { - Name: "keys", - Usage: "Retrieve a list of all keys in a bucket", - Action: KeysCommand, + Name: "keys", + Usage: "Retrieve a list of all keys in a bucket", + Action: func(c *cli.Context) { + path, name := c.Args().Get(0), c.Args().Get(1) + Keys(path, name) + }, }, { - Name: "buckets", - Usage: "Retrieves a list of all buckets", - Action: BucketsCommand, + Name: "buckets", + Usage: "Retrieves a list of all buckets", + Action: func(c *cli.Context) { + path := c.Args().Get(0) + Buckets(path) + }, }, { - Name: "pages", - Usage: "Dumps page information for a database", - Action: PagesCommand, + Name: "pages", + Usage: "Dumps page information for a database", + Action: func(c *cli.Context) { + path := c.Args().Get(0) + Pages(path) + }, }, } return app } -// GetCommand retrieves the value for a given bucket/key. -func GetCommand(c *cli.Context) { - path, name, key := c.Args().Get(0), c.Args().Get(1), c.Args().Get(2) - if _, err := os.Stat(path); os.IsNotExist(err) { - fatal(err) - return - } - - db, err := bolt.Open(path, 0600) - if err != nil { - fatal(err) - return - } - defer db.Close() - - err = db.View(func(tx *bolt.Tx) error { - // Find bucket. - b := tx.Bucket(name) - if b == nil { - fatalf("bucket not found: %s", name) - return nil - } - - // Find value for a given key. - value := b.Get([]byte(key)) - if value == nil { - fatalf("key not found: %s", key) - return nil - } - - println(string(value)) - return nil - }) - if err != nil { - fatal(err) - return - } -} - -// SetCommand sets the value for a given key in a bucket. -func SetCommand(c *cli.Context) { - path, name, key, value := c.Args().Get(0), c.Args().Get(1), c.Args().Get(2), c.Args().Get(3) - if _, err := os.Stat(path); os.IsNotExist(err) { - fatal(err) - return - } - - db, err := bolt.Open(path, 0600) - if err != nil { - fatal(err) - return - } - defer db.Close() - - err = db.Update(func(tx *bolt.Tx) error { - // Find bucket. - b := tx.Bucket(name) - if b == nil { - fatalf("bucket not found: %s", name) - return nil - } - - // Set value for a given key. - return b.Put([]byte(key), []byte(value)) - }) - if err != nil { - fatal(err) - return - } -} - -// KeysCommand retrieves a list of keys for a given bucket. -func KeysCommand(c *cli.Context) { - path, name := c.Args().Get(0), c.Args().Get(1) - if _, err := os.Stat(path); os.IsNotExist(err) { - fatal(err) - return - } - - db, err := bolt.Open(path, 0600) - if err != nil { - fatal(err) - return - } - defer db.Close() - - err = db.View(func(tx *bolt.Tx) error { - // Find bucket. - b := tx.Bucket(name) - if b == nil { - fatalf("bucket not found: %s", name) - return nil - } - - // Iterate over each key. - return b.ForEach(func(key, _ []byte) error { - println(string(key)) - return nil - }) - }) - if err != nil { - fatal(err) - return - } -} - -// BucketsCommand retrieves a list of all buckets. -func BucketsCommand(c *cli.Context) { - path := c.Args().Get(0) - if _, err := os.Stat(path); os.IsNotExist(err) { - fatal(err) - return - } - - db, err := bolt.Open(path, 0600) - if err != nil { - fatal(err) - return - } - defer db.Close() - - err = db.View(func(tx *bolt.Tx) error { - for _, b := range tx.Buckets() { - println(b.Name()) - } - return nil - }) - if err != nil { - fatal(err) - return - } -} - -// PagesCommand prints a list of all pages in a database. -func PagesCommand(c *cli.Context) { - path := c.Args().Get(0) - if _, err := os.Stat(path); os.IsNotExist(err) { - fatal(err) - return - } - - db, err := bolt.Open(path, 0600) - if err != nil { - fatal(err) - return - } - defer db.Close() - - println("ID TYPE ITEMS OVRFLW") - println("======== ========== ====== ======") - - db.Update(func(tx *bolt.Tx) error { - var id int - for { - p, err := tx.Page(id) - if err != nil { - fatalf("page error: %d: %s", id, err) - } else if p == nil { - break - } - - var overflow string - if p.OverflowCount > 0 { - overflow = strconv.Itoa(p.OverflowCount) - } - printf("%-8d %-10s %-6d %-6s\n", p.ID, p.Type, p.Count, overflow) - id += 1 + p.OverflowCount - } - return nil - }) -} - var logger = log.New(os.Stderr, "", 0) var logBuffer *bytes.Buffer |