iters: add IsClosed field, fixes #4

Author: mcuadros

commit.go

@@ -113,17 +113,28 @@ func (i *CommitIter) Next() (*Commit, error) {
 type iter struct {
 	ch chan core.Object
 	r  *Repository
+
+	IsClosed bool
 }
 
 func newIter(r *Repository) iter {
 	ch := make(chan core.Object, 1)
-	return iter{ch, r}
+	return iter{ch: ch, r: r}
 }
 
 func (i *iter) Add(o core.Object) {
+	if i.IsClosed {
+		return
+	}
+
 	i.ch <- o
 }
 
 func (i *iter) Close() {
+	if i.IsClosed {
+		return
+	}
+
+	defer func() { i.IsClosed = true }()
 	close(i.ch)
 }

commit_test.go

@@ -0,0 +1,16 @@
+package git
+
+import (
+	. "gopkg.in/check.v1"
+	"gopkg.in/src-d/go-git.v2/core"
+)
+
+type CommitCommon struct{}
+
+var _ = Suite(&CommitCommon{})
+
+func (s *CommitCommon) TestIterClose(c *C) {
+	i := &iter{ch: make(chan core.Object, 1)}
+	i.Close()
+	i.Close()
+}

repository_test.go

@@ -69,3 +69,14 @@ func (s *SuiteRepository) TestCommits(c *C) {
 
 	c.Assert(count, Equals, 8)
 }
+
+func (s *SuiteRepository) TestCommitIterClosePanic(c *C) {
+	r, err := NewRepository(RepositoryFixture, nil)
+	r.Remotes["origin"].upSrv = &MockGitUploadPackService{}
+
+	c.Assert(err, IsNil)
+	c.Assert(r.Pull("origin", "refs/heads/master"), IsNil)
+
+	commits := r.Commits()
+	commits.Close()
+}