Remove internal _done_ channel.

Signed-off-by: kuba-- <[email protected]>

Author: kuba--

repository_pool.go

@@ -272,7 +272,6 @@ type rowRepoIter struct {
 	iter           RowRepoIter
 	session        *Session
 	ctx            *sql.Context
-	done           chan struct{}
 }
 
 // NewRowRepoIter initializes a new repository iterator.
@@ -303,7 +302,6 @@ func NewRowRepoIter(
 		iter:           iter,
 		session:        s,
 		ctx:            ctx,
-		done:           make(chan struct{}),
 	}
 
 	return &repoIter, nil
@@ -324,21 +322,18 @@ func (i *rowRepoIter) Next() (sql.Row, error) {
 				repo, err := i.repositoryIter.Next()
 				if err != nil {
 					if err == io.EOF {
-						close(i.done)
 						return nil, io.EOF
 					}
 
 					if i.session.SkipGitErrors {
 						continue
 					}
 
-					close(i.done)
 					return nil, err
 				}
 
 				i.currRepoIter, err = i.iter.NewIterator(repo)
 				if err != nil {
-					close(i.done)
 					return nil, err
 				}
 			}
@@ -355,7 +350,6 @@ func (i *rowRepoIter) Next() (sql.Row, error) {
 					continue
 				}
 
-				close(i.done)
 				return nil, err
 			}
 

repository_pool_test.go

@@ -356,18 +356,13 @@ func testCaseRepositoryErrorIter(
 		for {
 			_, err := repoIter.Next()
 			if err != nil {
+				cancel()
 				break
 			}
 		}
 	}()
 
-	select {
-	case <-repoIter.done:
-	case <-repoIter.ctx.Done():
-
-	}
-
-	cancel()
+	<-repoIter.ctx.Done()
 }
 
 func TestRepositoryErrorIter(t *testing.T) {