Merge pull request #341 from mdaffin/branchiterator-foreach-errors

Add check for ErrIterOver in BranchIterator.ForEach

1 files changed, 4 insertions, 0 deletions

diff --git a/branch.go b/branch.go
index a869054..d381c23 100644
--- a/branch.go
+++ b/branch.go
@@ -73,6 +73,10 @@ func (i *BranchIterator) ForEach(f BranchIteratorFunc) error {
 		}
 	}
 
+	if err != nil && IsErrorCode(err, ErrIterOver) {
+		return nil
+	}
+
 	return err
 }