From e0314dcf56a063542df7d01e58d5aaef5a7f02c9 Mon Sep 17 00:00:00 2001
From: ston1th <ston1th@giftfish.de>
Date: Thu, 6 Oct 2022 18:42:10 +0200
Subject: [PATCH] implement scan and index queue

---
 pkg/server/handler.go |  4 +--
 pkg/server/helper.go  | 31 -------------------
 pkg/server/queue.go   | 71 +++++++++++++++++++++++++++++++++++++++++++
 pkg/server/server.go  |  8 +++++
 4 files changed, 81 insertions(+), 33 deletions(-)
 create mode 100644 pkg/server/queue.go

diff --git a/pkg/server/handler.go b/pkg/server/handler.go
index f8751c2..7f30b27 100644
--- a/pkg/server/handler.go
+++ b/pkg/server/handler.go
@@ -166,7 +166,7 @@ func indexHandler(ctx *Context) {
 		if fpath, ok = hasTrimPrefix(path, core.IndexPrefix); ok {
 			log = log.WithValues("file", fpath)
 			log.V(2).Info("started indexing file")
-			err = scanFile(ctx.Srv, fpath)
+			err = ctx.Srv.SQ.Add(fpath)
 			if err != nil {
 				log.Error(err, "error indexing file")
 			}
@@ -214,7 +214,7 @@ func indexHandler(ctx *Context) {
 				ctx.Srv.DB.IndexMutex.Set(uint64(len(files)))
 				for _, f := range files {
 					if !ctx.Srv.DB.IsIndexed(f) {
-						err := scanFile(ctx.Srv, f)
+						err := ctx.Srv.SQ.Add(f)
 						if err != nil {
 							log.Error(err, "error indexing file", "file", f)
 						}
diff --git a/pkg/server/helper.go b/pkg/server/helper.go
index 58a6a0e..2fccedc 100644
--- a/pkg/server/helper.go
+++ b/pkg/server/helper.go
@@ -28,37 +28,6 @@ func getStats(i *bleve.Index) *core.Stats {
 	}
 }
 
-func scanFile(srv *HTTPServer, path string) (err error) {
-	err = srv.FS.AddScan(path)
-	if err != nil {
-		return
-	}
-	go func() {
-		defer srv.FS.RemoveScan(path)
-		log := srv.Log.WithValues("file", path)
-		file, txt, err := srv.Scanner.Scan(path)
-		if err != nil {
-			log.Error(err, "error scanning file")
-			return
-		}
-		tags, err := srv.DB.GetAllRTags()
-		if err != nil {
-			log.Error(err, "error getting tags")
-		}
-		found := tags.Match(txt)
-		id, err := srv.DB.Index.Add(txt, found)
-		if err != nil {
-			log.Error(err, "error adding file to index")
-			return
-		}
-		err = srv.DB.NewFile(id, file, found)
-		if err != nil {
-			log.Error(err, "error adding file to DB")
-		}
-	}()
-	return
-}
-
 // Parts taken from strings.TrimPrefix
 func hasTrimPrefix(s, prefix string) (string, bool) {
 	if len(s) >= len(prefix) && s[0:len(prefix)] == prefix {
diff --git a/pkg/server/queue.go b/pkg/server/queue.go
new file mode 100644
index 0000000..278a2e1
--- /dev/null
+++ b/pkg/server/queue.go
@@ -0,0 +1,71 @@
+// Copyright (C) 2022 Marius Schellenberger
+
+package server
+
+import "context"
+
+type ScanQueue struct {
+	ctx   context.Context
+	srv   *HTTPServer
+	pchan chan string
+}
+
+func NewScanQueue(ctx context.Context, srv *HTTPServer, buffer int) *ScanQueue {
+	return &ScanQueue{
+		ctx:   ctx,
+		srv:   srv,
+		pchan: make(chan string, buffer),
+	}
+}
+
+func (sq *ScanQueue) Add(path string) (err error) {
+	err = sq.srv.FS.AddScan(path)
+	if err != nil {
+		return
+	}
+	go func() {
+		select {
+		case <-sq.ctx.Done():
+			return
+		case sq.pchan <- path:
+		}
+	}()
+	return
+}
+
+func (sq *ScanQueue) Scan() {
+	for {
+		select {
+		case <-sq.ctx.Done():
+			//close(sq.pchan)
+			return
+		case p := <-sq.pchan:
+			sq.scanFile(p)
+		}
+	}
+}
+
+func (sq *ScanQueue) scanFile(path string) {
+	srv := sq.srv
+	defer srv.FS.RemoveScan(path)
+	log := srv.Log.WithValues("file", path)
+	file, txt, err := srv.Scanner.Scan(path)
+	if err != nil {
+		log.Error(err, "error scanning file")
+		return
+	}
+	tags, err := srv.DB.GetAllRTags()
+	if err != nil {
+		log.Error(err, "error getting tags")
+	}
+	found := tags.Match(txt)
+	id, err := srv.DB.Index.Add(txt, found)
+	if err != nil {
+		log.Error(err, "error adding file to index")
+		return
+	}
+	err = srv.DB.NewFile(id, file, found)
+	if err != nil {
+		log.Error(err, "error adding file to DB")
+	}
+}
diff --git a/pkg/server/server.go b/pkg/server/server.go
index e3fbf45..2925d9e 100644
--- a/pkg/server/server.go
+++ b/pkg/server/server.go
@@ -39,21 +39,27 @@ type HTTPServer struct {
 	JWT     *jwt.JWT
 	Scanner *scan.Scanner
 	FS      *fs.Filesystem
+	SQ      *ScanQueue
 	Log     logr.Logger
 	plain   logr.Logger
 
+	cancel func()
+
 	templ map[string]*template.Template
 	res   map[string][]byte
 }
 
 func NewHTTPServer(log logr.Logger, cfg *core.Config, l net.Listener, s *scan.Scanner) (srv *HTTPServer) {
+	ctx, cancel := context.WithCancel(context.Background())
 	srv = &HTTPServer{
 		Config:   cfg,
 		listener: l,
 		Scanner:  s,
 		Log:      log.WithName("server"),
 		plain:    log,
+		cancel:   cancel,
 	}
+	srv.SQ = NewScanQueue(ctx, srv, 100)
 	srv.loadTemplates()
 	srv.srv = &http.Server{}
 	return
@@ -88,6 +94,7 @@ func (s *HTTPServer) Start() (err error) {
 	} else {
 		s.srv.Handler = s.handler
 	}
+	go s.SQ.Scan()
 	go func() {
 		err = s.srv.Serve(s.listener)
 		if err != nil && err != http.ErrServerClosed {
@@ -98,6 +105,7 @@ func (s *HTTPServer) Start() (err error) {
 }
 
 func (s *HTTPServer) Stop() error {
+	s.cancel()
 	s.DB.Close()
 	s.JWT.Stop()
 	ctx, cancel := context.WithTimeout(context.Background(), time.Second*5)