Full MVC of gitlab-zoekt-indexer

cmd/gitlab-zoekt-indexer/main.go

@@ -28,7 +28,6 @@ type options struct {
 }
 
 type indexServer struct {
-	indexDir             string
 	indexBuilder         indexBuilder
 	promRegistry         *prometheus.Registry
 	metricsRequestsTotal *prometheus.CounterVec
@@ -49,12 +48,19 @@ type indexRequest struct {
 }
 
 type indexBuilder interface {
-	indexRepository(ctx context.Context, req indexRequest, indexDir string) error
+	getIndexDir() string
+	indexRepository(ctx context.Context, req indexRequest) error
 }
 
-type defaultIndexBuilder struct{}
+type defaultIndexBuilder struct {
+	indexDir string
+}
 
-func (b defaultIndexBuilder) indexRepository(ctx context.Context, req indexRequest, indexDir string) error {
+func (b defaultIndexBuilder) getIndexDir() string {
+	return b.indexDir
+}
+
+func (b defaultIndexBuilder) indexRepository(ctx context.Context, req indexRequest) error {
 	if req.GitalyConnectionInfo == nil {
 		return errors.New("gitalyConnectionInfo is not set")
 	}
@@ -72,7 +78,7 @@ func (b defaultIndexBuilder) indexRepository(ctx context.Context, req indexReque
 	defer cancel()
 
 	idx := &indexer.Indexer{
-		IndexDir:           indexDir,
+		IndexDir:           b.indexDir,
 		ProjectID:          req.RepoID,
 		CorrelationID:      correlation.GetCorrelationID(ctx),
 		GitalyAddress:      req.GitalyConnectionInfo.Address,
@@ -89,8 +95,8 @@ func (b defaultIndexBuilder) indexRepository(ctx context.Context, req indexReque
 }
 
 func (s *indexServer) createIndexDir() {
-	if err := os.MkdirAll(s.indexDir, 0o755); err != nil {
-		log.Fatalf("createIndexDir %s: %v", s.indexDir, err)
+	if err := os.MkdirAll(s.indexBuilder.getIndexDir(), 0o755); err != nil {
+		log.Fatalf("createIndexDir %s: %v", s.indexBuilder.getIndexDir(), err)
 	}
 }
 
@@ -112,7 +118,7 @@ func (s *indexServer) handleStatus() http.HandlerFunc {
 		}
 
 		idx := &indexer.Indexer{
-			IndexDir:  s.indexDir,
+			IndexDir:  s.indexBuilder.getIndexDir(),
 			ProjectID: uint32(repoID),
 		}
 
@@ -174,7 +180,7 @@ func (s *indexServer) handleIndex() http.HandlerFunc {
 			return
 		}
 
-		err = s.indexBuilder.indexRepository(r.Context(), req, s.indexDir)
+		err = s.indexBuilder.indexRepository(r.Context(), req)
 		if err != nil {
 			s.respondWithError(w, r, route, err)
 			return
@@ -214,10 +220,10 @@ func (s *indexServer) handleTruncate() http.HandlerFunc {
 	}
 
 	return func(w http.ResponseWriter, r *http.Request) {
-		err := emptyDirectory(s.indexDir)
+		err := emptyDirectory(s.indexBuilder.getIndexDir())
 
 		if err != nil {
-			err = fmt.Errorf("failed to empty indexDir: %v with error: %v", s.indexDir, err)
+			err = fmt.Errorf("failed to empty indexDir: %v with error: %v", s.indexBuilder.getIndexDir(), err)
 
 			s.respondWithError(w, r, route, err)
 			return
@@ -304,8 +310,9 @@ func main() {
 	opts := parseOptions()
 
 	server := indexServer{
-		indexDir:     opts.indexDir,
-		indexBuilder: defaultIndexBuilder{},
+		indexBuilder: defaultIndexBuilder{
+			indexDir: opts.indexDir,
+		},
 	}
 
 	server.startIndexingApi(opts.listen)