gitbase: ignore missing HEAD references

Signed-off-by: Miguel Molina <[email protected]>

Author: erizocosmico

integration_test.go

@@ -2,6 +2,8 @@ package gitbase_test
 
 import (
 	"context"
+	"os"
+	"path/filepath"
 	"testing"
 
 	"github.com/src-d/gitbase/internal/rule"
@@ -278,6 +280,31 @@ func queryResults(
 	return rows
 }
 
+func TestMissingHeadRefs(t *testing.T) {
+	require := require.New(t)
+
+	path := filepath.Join(
+		os.Getenv("GOPATH"),
+		"src", "github.com", "src-d", "gitbase",
+		"_testdata",
+	)
+
+	pool := gitbase.NewRepositoryPool()
+	require.NoError(pool.AddSivaDir(path))
+
+	engine := sqle.New()
+	engine.AddDatabase(gitbase.NewDatabase("foo"))
+
+	session := gitbase.NewSession(pool)
+	ctx := sql.NewContext(context.TODO(), sql.WithSession(session))
+	_, iter, err := engine.Query(ctx, "SELECT * FROM refs")
+	require.NoError(err)
+
+	rows, err := sql.RowIterToRows(iter)
+	require.NoError(err)
+	require.Len(rows, 56)
+}
+
 func BenchmarkQueries(b *testing.B) {
 	queries := []struct {
 		name  string

iterator.go

@@ -355,7 +355,11 @@ func (i *refIter) New(ctx *sql.Context, repo *Repository) (ChainableIter, error)
 
 	head, err := repo.Repo.Head()
 	if err != nil {
-		return nil, err
+		if err != plumbing.ErrReferenceNotFound {
+			return nil, err
+		}
+
+		logrus.WithField("repo", repo.ID).Debug("unable to get HEAD of repository")
 	}
 
 	return &refIter{
@@ -489,7 +493,13 @@ func (i *repoRefsIter) Advance() error {
 
 			i.head, err = i.repos.Repo().Repo.Head()
 			if err != nil {
-				return err
+				if err != plumbing.ErrReferenceNotFound {
+					return err
+				}
+
+				logrus.WithField("repo", i.repos.Repo().ID).
+					Debug("unable to get HEAD of repository")
+				continue
 			}
 		}
 
@@ -616,7 +626,12 @@ func (i *remoteRefsIter) Advance() error {
 
 			i.head, err = i.repo.Head()
 			if err != nil {
-				return err
+				if err != plumbing.ErrReferenceNotFound {
+					return err
+				}
+
+				logrus.WithField("repo", i.remotes.Remote().RepoID).
+					Debug("unable to get HEAD of repository")
 			}
 		}
 

references.go

@@ -127,7 +127,11 @@ func (i *referenceIter) NewIterator(repo *Repository) (RowRepoIter, error) {
 
 	head, err := repo.Repo.Head()
 	if err != nil {
-		return nil, err
+		if err != plumbing.ErrReferenceNotFound {
+			return nil, err
+		}
+
+		logrus.WithField("repo", repo.ID).Debug("unable to get HEAD of repository")
 	}
 
 	return &referenceIter{
@@ -190,7 +194,11 @@ func (i *filteredReferencesIter) NewIterator(repo *Repository) (RowRepoIter, err
 
 	head, err := repo.Repo.Head()
 	if err != nil {
-		return nil, err
+		if err != plumbing.ErrReferenceNotFound {
+			return nil, err
+		}
+
+		logrus.WithField("repo", repo.ID).Debug("unable to get HEAD of repository")
 	}
 
 	return &filteredReferencesIter{