feat(backend): dedup CLI (hash backfill + pairs rescan)
cmd/dedup is the offline maintenance tool for duplicate detection. It reuses the server's config and runs two phases (both by default; -hashes / -pairs to pick): - hashes: compute the perceptual hash of every live image/video missing one — images from their bytes, videos from a middle frame via DiskStorage. VideoFrameMiddle. Per-file failures are reported and counted, not fatal. - pairs: rebuild data.duplicate_pairs from all current hashes (DuplicateService. Rescan). Idempotent and safe to re-run: hashing only touches NULL phashes, the pairs rebuild is a full replace. This is how video phashes and any backlog get computed, and how newly uploaded duplicates become visible. Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
This commit is contained in:
@@ -0,0 +1,165 @@
|
|||||||
|
// Command dedup is the offline maintenance tool for duplicate detection. It runs
|
||||||
|
// in two phases:
|
||||||
|
//
|
||||||
|
// hashes — compute the perceptual hash of every live image/video that has none
|
||||||
|
// yet (images from their bytes, videos from a middle frame via ffmpeg).
|
||||||
|
// pairs — rebuild data.duplicate_pairs from all current hashes.
|
||||||
|
//
|
||||||
|
// Both phases run by default; pass -hashes or -pairs to run only one. It reuses
|
||||||
|
// the server's configuration (DATABASE_URL, FILES_PATH, THUMBS_CACHE_PATH, …) and
|
||||||
|
// is safe to re-run: hashing only touches files whose phash is NULL, and the
|
||||||
|
// pairs rebuild is a full replace.
|
||||||
|
//
|
||||||
|
// go run ./cmd/dedup # hashes, then pairs
|
||||||
|
// go run ./cmd/dedup -pairs # only rebuild pairs
|
||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"flag"
|
||||||
|
"fmt"
|
||||||
|
"io"
|
||||||
|
"os"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
"github.com/google/uuid"
|
||||||
|
|
||||||
|
"tanabata/backend/internal/config"
|
||||||
|
"tanabata/backend/internal/db/postgres"
|
||||||
|
"tanabata/backend/internal/imagehash"
|
||||||
|
"tanabata/backend/internal/service"
|
||||||
|
"tanabata/backend/internal/storage"
|
||||||
|
)
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
hashesOnly := flag.Bool("hashes", false, "only (re)compute missing perceptual hashes")
|
||||||
|
pairsOnly := flag.Bool("pairs", false, "only rebuild the duplicate pairs table")
|
||||||
|
flag.Parse()
|
||||||
|
|
||||||
|
// No flag, or both, means run everything.
|
||||||
|
doHashes := *hashesOnly || !*pairsOnly
|
||||||
|
doPairs := *pairsOnly || !*hashesOnly
|
||||||
|
|
||||||
|
cfg, err := config.Load()
|
||||||
|
if err != nil {
|
||||||
|
fatal("load config", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
ctx := context.Background()
|
||||||
|
pool, err := postgres.NewPool(ctx, cfg.DatabaseURL)
|
||||||
|
if err != nil {
|
||||||
|
fatal("connect to database", err)
|
||||||
|
}
|
||||||
|
defer pool.Close()
|
||||||
|
|
||||||
|
diskStorage, err := storage.NewDiskStorage(
|
||||||
|
cfg.FilesPath, cfg.ThumbsCachePath,
|
||||||
|
cfg.ThumbWidth, cfg.ThumbHeight,
|
||||||
|
cfg.PreviewWidth, cfg.PreviewHeight,
|
||||||
|
cfg.ThumbMaxPixels, cfg.ThumbConcurrency,
|
||||||
|
)
|
||||||
|
if err != nil {
|
||||||
|
fatal("init storage", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
fileRepo := postgres.NewFileRepo(pool)
|
||||||
|
pairRepo := postgres.NewDuplicatePairRepo(pool)
|
||||||
|
dismissalRepo := postgres.NewDismissalRepo(pool)
|
||||||
|
aclRepo := postgres.NewACLRepo(pool)
|
||||||
|
auditRepo := postgres.NewAuditRepo(pool)
|
||||||
|
tagRepo := postgres.NewTagRepo(pool)
|
||||||
|
categoryRepo := postgres.NewCategoryRepo(pool)
|
||||||
|
poolRepo := postgres.NewPoolRepo(pool)
|
||||||
|
transactor := postgres.NewTransactor(pool)
|
||||||
|
|
||||||
|
aclSvc := service.NewACLService(aclRepo, fileRepo, tagRepo, categoryRepo, poolRepo, transactor)
|
||||||
|
auditSvc := service.NewAuditService(auditRepo)
|
||||||
|
dupSvc := service.NewDuplicateService(
|
||||||
|
fileRepo, pairRepo, dismissalRepo, aclSvc, auditSvc, transactor, cfg.DuplicateHashThreshold,
|
||||||
|
)
|
||||||
|
|
||||||
|
if doHashes {
|
||||||
|
if err := backfillHashes(ctx, fileRepo, diskStorage); err != nil {
|
||||||
|
fatal("backfill hashes", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if doPairs {
|
||||||
|
fmt.Printf("rebuilding duplicate pairs (threshold %d)...\n", cfg.DuplicateHashThreshold)
|
||||||
|
if err := dupSvc.Rescan(ctx, func(done, total int) {
|
||||||
|
fmt.Printf("\r hashed %d/%d", done, total)
|
||||||
|
}); err != nil {
|
||||||
|
fatal("rescan pairs", err)
|
||||||
|
}
|
||||||
|
fmt.Println("\n done")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// backfillHashes computes and stores a perceptual hash for every live image/video
|
||||||
|
// that lacks one. Failures on individual files are counted and reported, not
|
||||||
|
// fatal, so one unreadable file doesn't abort the whole run.
|
||||||
|
func backfillHashes(ctx context.Context, files *postgres.FileRepo, store *storage.DiskStorage) error {
|
||||||
|
pending, err := files.ListMissingPHash(ctx)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
total := len(pending)
|
||||||
|
fmt.Printf("hashing %d files without a perceptual hash...\n", total)
|
||||||
|
|
||||||
|
var hashed, skipped, failed int
|
||||||
|
for i, f := range pending {
|
||||||
|
ph, err := hashOne(ctx, store, f.ID, f.MIMEType)
|
||||||
|
switch {
|
||||||
|
case err != nil:
|
||||||
|
failed++
|
||||||
|
fmt.Fprintf(os.Stderr, "\n %s (%s): %v\n", f.ID, f.MIMEType, err)
|
||||||
|
case ph == nil:
|
||||||
|
skipped++ // not decodable; leave phash NULL
|
||||||
|
default:
|
||||||
|
if err := files.SetPHash(ctx, f.ID, ph); err != nil {
|
||||||
|
return fmt.Errorf("set phash for %s: %w", f.ID, err)
|
||||||
|
}
|
||||||
|
hashed++
|
||||||
|
}
|
||||||
|
if (i+1)%200 == 0 || i+1 == total {
|
||||||
|
fmt.Printf("\r processed %d/%d", i+1, total)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
fmt.Printf("\n hashed %d, skipped %d, failed %d\n", hashed, skipped, failed)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// hashOne returns the perceptual hash for one file, or nil when it isn't hashable
|
||||||
|
// (e.g. an image that won't decode). Images are hashed from their bytes; videos
|
||||||
|
// from a middle frame.
|
||||||
|
func hashOne(ctx context.Context, store *storage.DiskStorage, id uuid.UUID, mime string) (*int64, error) {
|
||||||
|
switch {
|
||||||
|
case strings.HasPrefix(mime, "image/"):
|
||||||
|
rc, err := store.Read(ctx, id)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
defer rc.Close()
|
||||||
|
data, err := io.ReadAll(rc)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if h, ok := imagehash.FromBytes(data); ok {
|
||||||
|
return &h, nil
|
||||||
|
}
|
||||||
|
return nil, nil
|
||||||
|
case strings.HasPrefix(mime, "video/"):
|
||||||
|
img, err := store.VideoFrameMiddle(ctx, id)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
h := imagehash.FromImage(img)
|
||||||
|
return &h, nil
|
||||||
|
default:
|
||||||
|
return nil, nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func fatal(what string, err error) {
|
||||||
|
fmt.Fprintf(os.Stderr, "dedup: %s: %v\n", what, err)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user