make it possible to handle errors

1 files changed, 1 insertions, 4 deletions

diff --git a/walk.go b/walk.go
index f7c147d..9e07411 100644
--- a/walk.go
+++ b/walk.go
@@ -7,7 +7,6 @@ package git
 import "C"
 
 import (
-	"io"
 	"runtime"
 	"unsafe"
 )
@@ -158,8 +157,6 @@ func (v *RevWalk) Next(id *Oid) (err error) {
 
 	ret := C.git_revwalk_next(id.toC(), v.ptr)
 	switch {
-	case ret == ITEROVER:
-		err = io.EOF
 	case ret < 0:
 		err = MakeGitError(ret)
 	}
@@ -173,7 +170,7 @@ func (v *RevWalk) Iterate(fun RevWalkIterator) (err error) {
 	oid := new(Oid)
 	for {
 		err = v.Next(oid)
-		if err == io.EOF {
+		if IsErrorCode(err, ErrIterOver) {
 			return nil
 		}
 		if err != nil {