From b7780518cc3d0945b7287f34f5b54bcd253eb2ca Mon Sep 17 00:00:00 2001
From: Todd Gruben <tgruben@umbel.com>
Date: Thu, 15 Sep 2016 13:37:10 -0500
Subject: [PATCH] cache update on import adjustment

---
 fragment.go | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/fragment.go b/fragment.go
index e38867474..1369a0143 100644
--- a/fragment.go
+++ b/fragment.go
@@ -883,7 +883,8 @@ func (f *Fragment) Import(bitmapIDs, profileIDs []uint64) error {
 			}
 
 			// Write to storage.
-			if _, err := f.storage.Add(pos); err != nil {
+			changed, err := f.storage.Add(pos)
+			if err != nil {
 				return err
 			}
 
@@ -898,13 +899,14 @@ func (f *Fragment) Import(bitmapIDs, profileIDs []uint64) error {
 				bmCounter = bitmap.Count()
 				lastID = bitmapID
 			}
-			if bitmap.SetBit(profileID) {
+			if changed {
 				bmCounter += 1
 			}
 
 			// Invalidate block checksum.
 			delete(f.checksums, int(bitmapID/HashBlockSize))
 		}
+		f.cache.Add(lastID, bmCounter)
 
 		f.cache.Invalidate()
 		return nil