From 17f04a167fdd16ab5781c1c57f718a7ad2777970 Mon Sep 17 00:00:00 2001 From: Zack Scholl Date: Mon, 14 Mar 2016 15:36:01 -0400 Subject: [PATCH] Check if opened Former-commit-id: a9ee6c19c060ff6323c9657cb90f2b166a4f7103 [formerly d75dba0b429e79ac8d417dcf044496f8d735d803] [formerly 9413872d5aecf982e873fcf43bf87cec2acf3e98 [formerly 3ef10c946634965775c8448b77caabf0847e4c07]] Former-commit-id: 7513366547dbac6a0778dbb49f1446092fc59842 [formerly f1f5d9eb16be383430f83b397c40c65ae0551ee9] Former-commit-id: bd73123413245683c3a90e6ae46884fa901f6e35 --- db.go | 15 ++++++--------- main.go | 4 +--- 2 files changed, 7 insertions(+), 12 deletions(-) diff --git a/db.go b/db.go index 5b1c213..dd3fd29 100644 --- a/db.go +++ b/db.go @@ -43,11 +43,10 @@ type WikiData struct { } func hasPassword(title string) (bool, error) { - Open(RuntimeArgs.DatabaseLocation) - defer Close() title = strings.ToLower(title) if !open { - return false, fmt.Errorf("db must be opened before loading") + Open(RuntimeArgs.DatabaseLocation) + defer Close() } hasPassword := false err := db.View(func(tx *bolt.Tx) error { @@ -126,11 +125,10 @@ func getCurrentText(title string, version int) (string, []versionsInfo, bool, ti } func (p *WikiData) load(title string) error { - Open(RuntimeArgs.DatabaseLocation) - defer Close() title = strings.ToLower(title) if !open { - return fmt.Errorf("db must be opened before loading!") + Open(RuntimeArgs.DatabaseLocation) + defer Close() } err := db.View(func(tx *bolt.Tx) error { var err error @@ -162,10 +160,9 @@ func (p *WikiData) load(title string) error { } func (p *WikiData) save(newText string) error { - Open(RuntimeArgs.DatabaseLocation) - defer Close() if !open { - return fmt.Errorf("db must be opened before saving") + Open(RuntimeArgs.DatabaseLocation) + defer Close() } err := db.Update(func(tx *bolt.Tx) error { bucket, err := tx.CreateBucketIfNotExists([]byte("datas")) diff --git a/main.go b/main.go index f9a128e..09358bf 100644 --- a/main.go +++ b/main.go @@ -61,9 +61,7 @@ Options:`) if *dumpDataset { fmt.Println("Dumping data to 'dump' folder...") - Open(RuntimeArgs.DatabaseLocation) dumpEverything() - Close() os.Exit(1) } @@ -87,7 +85,7 @@ Options:`) panic(err) } Close() - + // Default page aboutFile, _ := ioutil.ReadFile(path.Join(RuntimeArgs.SourcePath, "templates/aboutpage.md")) p := WikiData{"help", "", []string{}, []string{}, false}