75 lines
2.2 KiB
Go
75 lines
2.2 KiB
Go
package scanner2
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"github.com/google/go-pipeline/pkg/pipeline"
|
|
"github.com/navidrome/navidrome/log"
|
|
"github.com/navidrome/navidrome/model"
|
|
"github.com/navidrome/navidrome/model/request"
|
|
"github.com/navidrome/navidrome/scanner"
|
|
)
|
|
|
|
type scanner2 struct {
|
|
processCtx context.Context
|
|
ds model.DataStore
|
|
}
|
|
|
|
func New(ctx context.Context, ds model.DataStore) scanner.Scanner {
|
|
return &scanner2{processCtx: ctx, ds: ds}
|
|
}
|
|
|
|
func (s *scanner2) RescanAll(requestCtx context.Context, fullRescan bool) error {
|
|
ctx := request.AddValues(s.processCtx, requestCtx)
|
|
|
|
libs, err := s.ds.Library(ctx).GetAll()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
startTime := time.Now()
|
|
log.Info(ctx, "Scanner: Starting scan", "fullRescan", fullRescan, "numLibraries", len(libs))
|
|
|
|
err = s.runPipeline(
|
|
pipeline.NewProducer(s.produceFolders(ctx, libs, fullRescan), pipeline.Name("read folders from disk")),
|
|
pipeline.NewStage(s.processFolder(ctx), pipeline.Name("process folder")),
|
|
pipeline.NewStage(s.logFolder(ctx), pipeline.Name("log results")),
|
|
)
|
|
|
|
if err != nil {
|
|
log.Error(ctx, "Scanner: Error scanning libraries", "duration", time.Since(startTime), err)
|
|
} else {
|
|
log.Info(ctx, "Scanner: Finished scanning all libraries", "duration", time.Since(startTime))
|
|
}
|
|
return err
|
|
}
|
|
|
|
func (s *scanner2) runPipeline(producer pipeline.Producer[*folderEntry], stages ...pipeline.Stage[*folderEntry]) error {
|
|
if log.CurrentLevel() >= log.LevelDebug {
|
|
metrics, err := pipeline.Measure(producer, stages...)
|
|
log.Trace(metrics.String())
|
|
return err
|
|
}
|
|
return pipeline.Do(producer, stages...)
|
|
}
|
|
|
|
func (s *scanner2) logFolder(ctx context.Context) func(folder *folderEntry) (out *folderEntry, err error) {
|
|
return func(folder *folderEntry) (out *folderEntry, err error) {
|
|
log.Debug(ctx, "Scanner: Completed processing folder", "_path", folder.path,
|
|
"audioCount", len(folder.audioFiles), "imageCount", len(folder.imageFiles), "plsCount", len(folder.playlists))
|
|
return folder, nil
|
|
}
|
|
}
|
|
|
|
func (s *scanner2) Status(context.Context) (*scanner.StatusInfo, error) {
|
|
return &scanner.StatusInfo{}, nil
|
|
}
|
|
|
|
//nolint:unused
|
|
func (s *scanner2) doScan(ctx context.Context, fullRescan bool, folders <-chan string) error {
|
|
return nil
|
|
}
|
|
|
|
var _ scanner.Scanner = (*scanner2)(nil)
|