From 1675136966473083e970badfe92433a05d2d7fee Mon Sep 17 00:00:00 2001 From: jakka Date: Sat, 12 Apr 2025 14:44:39 +0300 Subject: implemented database and concurrent file scanning --- cmd/cmd.go | 58 ++++++++++++++++++++++++++++++++++++++-------------------- 1 file changed, 38 insertions(+), 20 deletions(-) (limited to 'cmd/cmd.go') diff --git a/cmd/cmd.go b/cmd/cmd.go index 86a1b92..82a9cd5 100644 --- a/cmd/cmd.go +++ b/cmd/cmd.go @@ -1,14 +1,13 @@ package main import ( - "fmt" + "context" "os" "os/signal" - "time" + "github.com/justjakka/reencoder/files" "github.com/tidwall/buntdb" "github.com/urfave/cli/v2" - "golang.org/x/sync/errgroup" ) func runCmd(cCtx *cli.Context) error { @@ -17,32 +16,51 @@ func runCmd(cCtx *cli.Context) error { return err } - db, err := buntdb.Open(ctx.Value("database").(string)) + db, err := buntdb.Open(ctx.Value("dbfile").(string)) if err != nil { return err } defer db.Close() + if err := db.Shrink(); err != nil { + return err + } + + if err := db.CreateIndex("process", "*", buntdb.IndexJSON("process")); err != nil { + if err != buntdb.ErrIndexExists { + return err + } + } + + ctx = context.WithValue(ctx, "database", db) + ctx, stop := signal.NotifyContext(ctx, os.Interrupt) defer stop() - test := new(errgroup.Group) - test.SetLimit(3) - for n := 1; n < 10; n++ { - test.Go(func() error { - select { - case <-ctx.Done(): - return nil - default: - fmt.Println("test") - time.Sleep(3 * time.Second) - return nil - } - }) + if err = files.IndexFlacs(ctx); err != nil { + return err } - _ = test.Wait() - fmt.Println("reached") - return cCtx.Err() + + return nil + + /* + test := new(errgroup.Group) + test.SetLimit(3) + for n := 1; n < 10; n++ { + test.Go(func() error { + select { + case <-ctx.Done(): + return nil + default: + fmt.Println("test") + time.Sleep(3 * time.Second) + return nil + } + }) + } + _ = test.Wait() + fmt.Println("reached") + return cCtx.Err() */ } func start() { -- cgit v1.3.1