cmd/gitbase: register pilosa index on server startup

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

Author: erizocosmico

cmd/gitbase/command/server.go

@@ -2,16 +2,20 @@ package command
 
 import (
 	"net"
+	"os"
 	"path/filepath"
 	"strconv"
 
+	"github.com/prometheus/common/log"
 	"github.com/src-d/gitbase"
 	"github.com/src-d/gitbase/internal/function"
 	"github.com/src-d/gitbase/internal/rule"
 
+	gopilosa "github.com/pilosa/go-pilosa"
 	"github.com/sirupsen/logrus"
 	sqle "gopkg.in/src-d/go-mysql-server.v0"
 	"gopkg.in/src-d/go-mysql-server.v0/server"
+	"gopkg.in/src-d/go-mysql-server.v0/sql/index/pilosa"
 	"gopkg.in/src-d/go-vitess.v0/mysql"
 )
 
@@ -28,13 +32,15 @@ const (
 
 // Server represents the `server` command of gitbase cli tool.
 type Server struct {
-	Verbose  bool     `short:"v" description:"Activates the verbose mode"`
-	Git      []string `short:"g" long:"git" description:"Path where the git repositories are located, multiple directories can be defined. Accepts globs."`
-	Siva     []string `long:"siva" description:"Path where the siva repositories are located, multiple directories can be defined. Accepts globs."`
-	Host     string   `short:"h" long:"host" default:"localhost" description:"Host where the server is going to listen"`
-	Port     int      `short:"p" long:"port" default:"3306" description:"Port where the server is going to listen"`
-	User     string   `short:"u" long:"user" default:"root" description:"User name used for connection"`
-	Password string   `short:"P" long:"password" default:"" description:"Password used for connection"`
+	Verbose   bool     `short:"v" description:"Activates the verbose mode"`
+	Git       []string `short:"g" long:"git" description:"Path where the git repositories are located, multiple directories can be defined. Accepts globs."`
+	Siva      []string `long:"siva" description:"Path where the siva repositories are located, multiple directories can be defined. Accepts globs."`
+	Host      string   `short:"h" long:"host" default:"localhost" description:"Host where the server is going to listen"`
+	Port      int      `short:"p" long:"port" default:"3306" description:"Port where the server is going to listen"`
+	User      string   `short:"u" long:"user" default:"root" description:"User name used for connection"`
+	Password  string   `short:"P" long:"password" default:"" description:"Password used for connection"`
+	PilosaURL string   `long:"pilosa" default:"http://localhost:10101" description:"URL to your pilosa server"`
+	IndexDir  string   `short:"i" long:"index" default:"/var/gitbase/index" description:"Directory where the gitbase indexes information will be persisted."`
 
 	// UnstableSquash quashing tables and pushing down join conditions is still
 	// a work in progress and unstable. To enable it, the GITBASE_UNSTABLE_SQUASH_ENABLE
@@ -104,6 +110,18 @@ func (c *Server) buildDatabase() error {
 	c.engine.Catalog.RegisterFunctions(function.Functions)
 	logrus.Debug("registered all available functions in catalog")
 
+	if err := os.MkdirAll(c.IndexDir, 0755); err != nil {
+		return err
+	}
+
+	client, err := gopilosa.NewClient(c.PilosaURL)
+	if err != nil {
+		return err
+	}
+
+	c.engine.Catalog.RegisterIndexDriver(pilosa.NewDriver(c.IndexDir, client))
+	log.Debug("registered pilosa index driver")
+
 	if c.UnstableSquash {
 		logrus.Warn("unstable squash tables rule is enabled")
 		c.engine.Analyzer.AddRule(rule.SquashJoinsRule, rule.SquashJoins)