From 4ab47545f7e48f6575dfa086a62754d363fb7aab Mon Sep 17 00:00:00 2001 From: Andrew Slotin Date: Fri, 13 Oct 2017 19:01:25 +0700 Subject: [PATCH] Pass db open timeout via command-line flag --- boltbrowser.go | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/boltbrowser.go b/boltbrowser.go index 4c38de6..19fa968 100644 --- a/boltbrowser.go +++ b/boltbrowser.go @@ -23,9 +23,15 @@ var currentFilename string const DefaultDBOpenTimeout = time.Second +var args struct { + DBOpenTimeout time.Duration +} + func init() { + flag.DurationVar(&args.DBOpenTimeout, "timeout", DefaultDBOpenTimeout, "DB file open timeout") flag.Usage = func() { - fmt.Fprintf(os.Stdout, "Usage: %s \n", ProgramName) + fmt.Fprintf(os.Stdout, "Usage: %s [OPTIONS] \nOptions:\n", ProgramName) + flag.PrintDefaults() } } @@ -50,7 +56,7 @@ func main() { databaseFiles := flag.Args() for _, databaseFile := range databaseFiles { currentFilename = databaseFile - db, err = bolt.Open(databaseFile, 0600, &bolt.Options{Timeout: DefaultDBOpenTimeout}) + db, err = bolt.Open(databaseFile, 0600, &bolt.Options{Timeout: args.DBOpenTimeout}) if err == bolt.ErrTimeout { termbox.Close() fmt.Printf("File %s is locked. Make sure it's not used by another app and try again\n", databaseFile)