From ebb2dc540858155152b21549afd77e65118e1474 Mon Sep 17 00:00:00 2001
From: Marc Pervaz Boocha <marcpervaz@qburst.com>
Date: Tue, 8 Apr 2025 16:15:27 +0530
Subject: Lint and bug fixes

---
 encoding.go | 13 +++++++++----
 1 file changed, 9 insertions(+), 4 deletions(-)

(limited to 'encoding.go')

diff --git a/encoding.go b/encoding.go
index fe376b2..a8778ab 100644
--- a/encoding.go
+++ b/encoding.go
@@ -121,6 +121,7 @@ func (d *decoder) DecodeTime() (time.Time, error) {
 	if t.IsZero() {
 		t = zero[time.Time]()
 	}
+
 	return t, nil
 }
 
@@ -231,17 +232,21 @@ func (d *decoder) DecodeStore(s *store) error {
 }
 
 func (s *store) Snapshot(w io.WriteSeeker) error {
-	s.Lock.Lock()
-	defer s.Lock.Unlock()
+	s.Lock.RLock()
+	defer s.Lock.RUnlock()
 
 	if _, err := w.Seek(0, io.SeekStart); err != nil {
 		return err
 	}
 
 	wr := newEncoder(w)
-	defer wr.Flush()
 
-	return wr.EncodeStore(s)
+	err := wr.EncodeStore(s)
+	if err != nil {
+		return err
+	}
+
+	return wr.Flush()
 }
 
 func (s *store) LoadSnapshot(r io.ReadSeeker) error {
-- 
cgit v1.2.3-70-g09d2