Merge pull request #18 from jfontan/add-repositories-table

Add repositories table

Author: jfontan

database.go

@@ -6,12 +6,13 @@ import (
 
 const (
 	// TODO 'references' is a reserved keyword into the parser
-	referencesTableName  = "refs"
-	commitsTableName     = "commits"
-	tagsTableName        = "tags"
-	blobsTableName       = "blobs"
-	treeEntriesTableName = "tree_entries"
-	objectsTableName     = "objects"
+	referencesTableName   = "refs"
+	commitsTableName      = "commits"
+	tagsTableName         = "tags"
+	blobsTableName        = "blobs"
+	treeEntriesTableName  = "tree_entries"
+	objectsTableName      = "objects"
+	repositoriesTableName = "repositories"
 )
 
 type Database struct {
@@ -22,6 +23,7 @@ type Database struct {
 	ter  sql.Table
 	br   sql.Table
 	or   sql.Table
+	rer  sql.Table
 }
 
 func NewDatabase(name string, pool *RepositoryPool) sql.Database {
@@ -33,6 +35,7 @@ func NewDatabase(name string, pool *RepositoryPool) sql.Database {
 		br:   newBlobsTable(pool),
 		ter:  newTreeEntriesTable(pool),
 		or:   newObjectsTable(pool),
+		rer:  newRepositoriesTable(pool),
 	}
 }
 
@@ -42,11 +45,12 @@ func (d *Database) Name() string {
 
 func (d *Database) Tables() map[string]sql.Table {
 	return map[string]sql.Table{
-		commitsTableName:     d.cr,
-		tagsTableName:        d.tr,
-		referencesTableName:  d.rr,
-		blobsTableName:       d.br,
-		treeEntriesTableName: d.ter,
-		objectsTableName:     d.or,
+		commitsTableName:      d.cr,
+		tagsTableName:         d.tr,
+		referencesTableName:   d.rr,
+		blobsTableName:        d.br,
+		treeEntriesTableName:  d.ter,
+		objectsTableName:      d.or,
+		repositoriesTableName: d.rer,
 	}
 }

database_test.go

@@ -41,6 +41,7 @@ func TestDatabase_Tables(t *testing.T) {
 		tagsTableName,
 		blobsTableName,
 		objectsTableName,
+		repositoriesTableName,
 	}
 	sort.Strings(expected)
 

repositories.go

@@ -0,0 +1,79 @@
+package gitquery
+
+import (
+	"io"
+
+	"gopkg.in/src-d/go-mysql-server.v0/sql"
+)
+
+type repositoriesTable struct {
+	pool *RepositoryPool
+}
+
+func newRepositoriesTable(pool *RepositoryPool) sql.Table {
+	return &repositoriesTable{pool: pool}
+}
+
+func (repositoriesTable) Resolved() bool {
+	return true
+}
+
+func (repositoriesTable) Name() string {
+	return repositoriesTableName
+}
+
+func (repositoriesTable) Schema() sql.Schema {
+	return sql.Schema{
+		{Name: "id", Type: sql.Text, Nullable: false},
+	}
+}
+
+func (r *repositoriesTable) TransformUp(f func(sql.Node) sql.Node) sql.Node {
+	return f(r)
+}
+
+func (r *repositoriesTable) TransformExpressionsUp(
+	f func(sql.Expression) sql.Expression) sql.Node {
+
+	return r
+}
+
+func (r repositoriesTable) RowIter() (sql.RowIter, error) {
+	iter := &repositoriesIter{}
+
+	rowRepoIter, err := NewRowRepoIter(r.pool, iter)
+	if err != nil {
+		return nil, err
+	}
+
+	return rowRepoIter, nil
+}
+
+func (repositoriesTable) Children() []sql.Node {
+	return nil
+}
+
+type repositoriesIter struct {
+	visited bool
+	id      string
+}
+
+func (i *repositoriesIter) NewIterator(repo *Repository) (RowRepoIter, error) {
+	return &repositoriesIter{
+		visited: false,
+		id:      repo.ID,
+	}, nil
+}
+
+func (i *repositoriesIter) Next() (sql.Row, error) {
+	if i.visited {
+		return nil, io.EOF
+	}
+
+	i.visited = true
+	return sql.NewRow(i.id), nil
+}
+
+func (i *repositoriesIter) Close() error {
+	return nil
+}

repositories_test.go

@@ -0,0 +1,65 @@
+package gitquery
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/require"
+	"gopkg.in/src-d/go-git-fixtures.v3"
+	"gopkg.in/src-d/go-mysql-server.v0/sql"
+)
+
+func TestRepositoriesTable_Name(t *testing.T) {
+	require := require.New(t)
+
+	f := fixtures.Basic().One()
+	table := getTable(require, f, repositoriesTableName)
+	require.Equal(repositoriesTableName, table.Name())
+}
+
+func TestRepositoriesTable_Children(t *testing.T) {
+	require := require.New(t)
+
+	f := fixtures.Basic().One()
+	table := getTable(require, f, repositoriesTableName)
+	require.Equal(0, len(table.Children()))
+}
+
+func TestRepositoriesTable_RowIter(t *testing.T) {
+	require := require.New(t)
+
+	repoIDs := []string{
+		"one", "two", "three", "four", "five", "six",
+		"seven", "eight", "nine",
+	}
+
+	pool := NewRepositoryPool()
+
+	for _, id := range repoIDs {
+		pool.Add(id, nil)
+	}
+
+	db := NewDatabase(repositoriesTableName, &pool)
+	require.NotNil(db)
+
+	tables := db.Tables()
+	table, ok := tables[repositoriesTableName]
+
+	require.True(ok)
+	require.NotNil(table)
+
+	rows, err := sql.NodeToRows(table)
+	require.Nil(err)
+	require.Len(rows, len(repoIDs))
+
+	idArray := make([]string, len(repoIDs))
+	for i, row := range rows {
+		idArray[i] = row[0].(string)
+	}
+	require.ElementsMatch(idArray, repoIDs)
+
+	schema := table.Schema()
+	for idx, row := range rows {
+		err := schema.CheckRow(row)
+		require.Nil(err, "row %d doesn't conform to schema", idx)
+	}
+}