From a5c97d68e77ca8c9c5518051f085051487cd0fbf Mon Sep 17 00:00:00 2001
From: bodqhrohro <bodqhrohro@gmail.com>
Date: Tue, 12 Nov 2019 17:50:25 +0200
Subject: Spawn TDlib instances for recovered sessions

---
 yamldb/yamldb.go | 32 ++++++++++++++++----------------
 1 file changed, 16 insertions(+), 16 deletions(-)

(limited to 'yamldb')

diff --git a/yamldb/yamldb.go b/yamldb/yamldb.go
index eec23be..1478cc7 100644
--- a/yamldb/yamldb.go
+++ b/yamldb/yamldb.go
@@ -18,9 +18,7 @@ type YamlDB struct {
 
 // Transaction executes the given callback and safely saves
 // the data after they are modified within the callback
-func (db *YamlDB) Transaction(callback func(), marshaller func() ([]byte, error)) error {
-	var err error
-
+func (db *YamlDB) Transaction(callback func() bool, marshaller func() ([]byte, error)) error {
 	log.Debug("Enter transaction")
 	db.lock.Lock()
 	defer func() {
@@ -28,19 +26,21 @@ func (db *YamlDB) Transaction(callback func(), marshaller func() ([]byte, error)
 		log.Debug("Exit transaction")
 	}()
 
-	callback()
-
-	yamlData, err := marshaller()
-	if err != nil {
-		return errors.Wrap(err, "Data marshalling error")
-	}
-	err = ioutil.WriteFile(db.PathNew, yamlData, 0644)
-	if err != nil {
-		return errors.Wrap(err, "YamlDB write failure")
-	}
-	err = os.Rename(db.PathNew, db.Path)
-	if err != nil {
-		return errors.Wrap(err, "Couldn't rewrite an old YamlDB file")
+	isDataChanged := callback()
+
+	if isDataChanged {
+		yamlData, err := marshaller()
+		if err != nil {
+			return errors.Wrap(err, "Data marshalling error")
+		}
+		err = ioutil.WriteFile(db.PathNew, yamlData, 0644)
+		if err != nil {
+			return errors.Wrap(err, "YamlDB write failure")
+		}
+		err = os.Rename(db.PathNew, db.Path)
+		if err != nil {
+			return errors.Wrap(err, "Couldn't rewrite an old YamlDB file")
+		}
 	}
 
 	return nil
-- 
cgit v1.2.3