Add DB.Check().

2 files changed, 42 insertions, 0 deletions

diff --git a/cmd/bolt/check.go b/cmd/bolt/check.go
new file mode 100644
index 0000000..ec2ea69
--- /dev/null
+++ b/cmd/bolt/check.go
@@ -0,0 +1,34 @@
+package main
+
+import (
+	"os"
+
+	"github.com/boltdb/bolt"
+)
+
+// Check performs a consistency check on the database and prints any errors found.
+func Check(path string) {
+	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()
+
+	// Perform consistency check.
+	if err := db.Check(); err != nil {
+		if errors, ok := err.(bolt.ErrorList); ok {
+			for _, err := range errors {
+				println(err)
+			}
+		}
+		fatalln(err)
+		return
+	}
+	println("OK")
+}
diff --git a/cmd/bolt/main.go b/cmd/bolt/main.go
index 6b28060..bc5fadb 100644
--- a/cmd/bolt/main.go
+++ b/cmd/bolt/main.go
@@ -61,6 +61,14 @@ func NewApp() *cli.App {
 				Pages(path)
 			},
 		},
+		{
+			Name:  "check",
+			Usage: "Performs a consistency check on the database",
+			Action: func(c *cli.Context) {
+				path := c.Args().Get(0)
+				Check(path)
+			},
+		},
 	}
 	return app
 }