Add error handling for hash computaiton of delta snapshot

Signed-off-by: Swapnil Mhamane <[email protected]>
gardener · Dec 11, 2018 · 21aabc1 · 21aabc1
1 parent d960b82
commit 21aabc1
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/pkg/snapshot/snapshotter/snapshotter.go b/pkg/snapshot/snapshotter/snapshotter.go
@@ -20,6 +20,7 @@ import (
 	"crypto/sha256"
 	"encoding/json"
 	"fmt"
+	"io"
 	"path"
 	"sync"
 	"time"
@@ -250,10 +251,12 @@ func (ssr *Snapshotter) takeDeltaSnapshot() error {
 	}
 	// compute hash
 	hash := sha256.New()
-	hash.Write(data)
+	if _, err := io.Copy(hash, bytes.NewReader(data)); err != nil {
+		return fmt.Errorf("failed to compute hash of events: %v", err)
+	}
 	data = hash.Sum(data)
-	dataReader := bytes.NewReader(data)
 
+	dataReader := bytes.NewReader(data)
 	if err := ssr.config.store.Save(*snap, dataReader); err != nil {
 		ssr.logger.Errorf("Error saving delta snapshots. %v", err)
 		return err