Use transaction when necessary

cmd/data/articles.go

@@ -140,11 +140,15 @@ func LoadArticleList(filename string) (*ArticleList, error) {
 		return nil, fmt.Errorf("error opening key file: %v", err)
 	}
 
-	articleList := NewArticleList()
+	articleList := initArticleList()
 	if err = gob.NewDecoder(file).Decode(&articleList.articles); err != nil {
 		return nil, fmt.Errorf("error decoding key: %v", err)
 	}
 
+	articleList.wg.Add(1)
+	go articleList.start()
+	articleList.wg.Wait()
+
 	return articleList, nil
 }
 
@@ -189,10 +193,14 @@ func LoadTagList(filename string) (*TagList, error) {
 	}
 	defer file.Close()
 
-	tagList := NewTagList()
+	tagList := initTagList()
 	if err = gob.NewDecoder(file).Decode(&tagList.tags); err != nil {
 		return nil, fmt.Errorf("error decoding key: %v", err)
 	}
 
+	tagList.wg.Add(1)
+	go tagList.start()
+	tagList.wg.Wait()
+
 	return tagList, nil
 }

cmd/data/db.go

@@ -89,12 +89,19 @@ func (db *DB) CheckPassword(id int64, pass string) error {
 }
 
 func (db *DB) ChangePassword(id int64, oldPass, newPass string) error {
+	tx, err := db.Begin()
+	if err != nil {
+		return fmt.Errorf("error starting transaction: %v", err)
+	}
+
 	if err := db.CheckPassword(id, oldPass); err != nil {
+		tx.Rollback()
 		return fmt.Errorf("error checking password: %v", err)
 	}
 
 	newHashedPass, err := bcrypt.GenerateFromPassword([]byte(newPass), bcrypt.DefaultCost)
 	if err != nil {
+		tx.Rollback()
 		return fmt.Errorf("error creating password hash: %v", err)
 	}
 
@@ -104,9 +111,14 @@ func (db *DB) ChangePassword(id int64, oldPass, newPass string) error {
     WHERE id = ?
     `
 	if _, err = db.Exec(query, string(newHashedPass), id); err != nil {
+		tx.Rollback()
 		return fmt.Errorf("error updating password in DB: %v", err)
 	}
 
+	if err = tx.Commit(); err != nil {
+		return fmt.Errorf("error committing transaction: %v", err)
+	}
+
 	return nil
 }
 

cmd/data/rss.go

@@ -70,17 +70,17 @@ func LoadChannel(filename string) (*Channel, error) {
 	}
 	defer file.Close()
 
-	channel := initChannel()
-	channel.wg.Add(1)
-	go channel.start()
-	channel.wg.Wait()
-
 	tmpChannel := new(rss.Channel)
 	if err = gob.NewDecoder(file).Decode(tmpChannel); err != nil {
 		return nil, fmt.Errorf("error decoding channel from file %v: %v", filename, err)
 	}
 
+	channel := initChannel()
+	channel.wg.Add(1)
+	go channel.start()
+	channel.wg.Wait()
 	channel.Set(*tmpChannel)
+
 	return channel, nil
 }