实现增量扫描

2026-04-11 14:38:23 +08:00
parent 3f82e29c6c
commit 79d47c81e7
10 changed files with 351 additions and 91 deletions
--- a/internal/repository/library.go
+++ b/internal/repository/library.go
@@ -76,14 +76,34 @@ func (r *LibraryRepository) Delete(id int) error {
 	}
 	defer tx.Rollback()

-	// Delete the library — CASCADE removes media_files and songs automatically
 	_, err = tx.Exec("DELETE FROM libraries WHERE id = ?", id)
 	if err != nil {
 		return err
 	}

-	// Clean up orphan albums (no songs reference them)
-	_, err = tx.Exec(`
+	if err := cleanupOrphansTx(tx); err != nil {
+		return err
+	}
+
+	return tx.Commit()
+}
+
+func (r *LibraryRepository) CleanupOrphans() error {
+	tx, err := r.db.Begin()
+	if err != nil {
+		return err
+	}
+	defer tx.Rollback()
+
+	if err := cleanupOrphansTx(tx); err != nil {
+		return err
+	}
+
+	return tx.Commit()
+}
+
+func cleanupOrphansTx(tx *sql.Tx) error {
+	_, err := tx.Exec(`
 		DELETE FROM albums WHERE id NOT IN (
 			SELECT DISTINCT album_id FROM songs WHERE album_id IS NOT NULL
 		)
@@ -92,7 +112,6 @@ func (r *LibraryRepository) Delete(id int) error {
 		return err
 	}

-	// Clean up orphan artists (no songs and no albums reference them)
 	_, err = tx.Exec(`
 		DELETE FROM artists WHERE id NOT IN (
 			SELECT DISTINCT artist_id FROM songs WHERE artist_id IS NOT NULL
@@ -104,5 +123,5 @@ func (r *LibraryRepository) Delete(id int) error {
 		return err
 	}

-	return tx.Commit()
+	return nil
 }
--- a/internal/repository/media_repo.go
+++ b/internal/repository/media_repo.go
@@ -3,12 +3,16 @@ package repository
 import (
 	"butterfliu/internal/model"
 	"database/sql"
+	"time"
 )

 type ExistingMediaFile struct {
 	Path        string
 	MediaFileID int
 	HasSong     bool
+	FileSize    int64
+	FileMtimeNs int64
+	LastSeenAt  time.Time
 }

 type MediaRepository struct {
@@ -21,8 +25,19 @@ func NewMediaRepository(db *sql.DB) *MediaRepository {

 func (r *MediaRepository) Get(id int) (model.MediaFile, error) {
 	var m model.MediaFile
-	err := r.db.QueryRow("SELECT id, path, library_id FROM media_files WHERE id = ?", id).Scan(
+	err := r.db.QueryRow(`
+		SELECT id, path, library_id,
+		       COALESCE(file_size, 0), COALESCE(file_mtime_ns, 0), COALESCE(format, ''),
+		       COALESCE(bit_rate, 0), COALESCE(sample_rate, 0),
+		       COALESCE(last_seen_at, '1970-01-01T00:00:00Z'),
+		       COALESCE(created_at, '1970-01-01T00:00:00Z'),
+		       COALESCE(updated_at, '1970-01-01T00:00:00Z')
+		FROM media_files WHERE id = ?
+	`, id).Scan(
 		&m.ID, &m.Path, &m.LibraryID,
+		&m.FileSize, &m.FileMtimeNs, &m.Format,
+		&m.BitRate, &m.SampleRate, &m.LastSeenAt,
+		&m.CreatedAt, &m.UpdatedAt,
 	)
 	if err != nil {
 		return model.MediaFile{}, err
@@ -33,8 +48,9 @@ func (r *MediaRepository) Get(id int) (model.MediaFile, error) {
 func (r *MediaRepository) GetByPath(path string) (model.MediaFile, error) {
 	row := r.db.QueryRow(`
 		SELECT id, path, library_id,
-		       COALESCE(file_size, 0), COALESCE(format, ''),
+		       COALESCE(file_size, 0), COALESCE(file_mtime_ns, 0), COALESCE(format, ''),
 		       COALESCE(bit_rate, 0), COALESCE(sample_rate, 0),
+		       COALESCE(last_seen_at, '1970-01-01T00:00:00Z'),
 		       COALESCE(created_at, '1970-01-01T00:00:00Z'),
 		       COALESCE(updated_at, '1970-01-01T00:00:00Z')
 		FROM media_files WHERE path = ?
@@ -42,8 +58,8 @@ func (r *MediaRepository) GetByPath(path string) (model.MediaFile, error) {
 	var mediaFile model.MediaFile
 	if err := row.Scan(
 		&mediaFile.ID, &mediaFile.Path, &mediaFile.LibraryID,
-		&mediaFile.FileSize, &mediaFile.Format,
-		&mediaFile.BitRate, &mediaFile.SampleRate,
+		&mediaFile.FileSize, &mediaFile.FileMtimeNs, &mediaFile.Format,
+		&mediaFile.BitRate, &mediaFile.SampleRate, &mediaFile.LastSeenAt,
 		&mediaFile.CreatedAt, &mediaFile.UpdatedAt,
 	); err != nil {
 		return model.MediaFile{}, err
@@ -51,8 +67,18 @@ func (r *MediaRepository) GetByPath(path string) (model.MediaFile, error) {
 	return mediaFile, nil
 }

-func (r *MediaRepository) Create(path string, libraryID int) (model.MediaFile, error) {
-	result, err := r.db.Exec("INSERT INTO media_files (path, library_id) VALUES (?, ?)", path, libraryID)
+func (r *MediaRepository) Create(mediaFile model.MediaFile) (model.MediaFile, error) {
+	result, err := r.db.Exec(
+		`INSERT INTO media_files (path, library_id, file_size, file_mtime_ns, format, bit_rate, sample_rate, last_seen_at) VALUES (?, ?, ?, ?, ?, ?, ?, ?)`,
+		mediaFile.Path,
+		mediaFile.LibraryID,
+		mediaFile.FileSize,
+		mediaFile.FileMtimeNs,
+		mediaFile.Format,
+		mediaFile.BitRate,
+		mediaFile.SampleRate,
+		mediaFile.LastSeenAt,
+	)
 	if err != nil {
 		return model.MediaFile{}, err
 	}
@@ -62,16 +88,15 @@ func (r *MediaRepository) Create(path string, libraryID int) (model.MediaFile, e
 		return model.MediaFile{}, err
 	}

-	return model.MediaFile{
-		ID:        int(id),
-		Path:      path,
-		LibraryID: libraryID,
-	}, nil
+	mediaFile.ID = int(id)
+	return mediaFile, nil
 }

 func (r *MediaRepository) ListExistingByLibrary(libraryID int) ([]ExistingMediaFile, error) {
 	rows, err := r.db.Query(`
-		SELECT mf.path, mf.id, CASE WHEN s.id IS NULL THEN 0 ELSE 1 END AS has_song
+		SELECT mf.path, mf.id, CASE WHEN s.id IS NULL THEN 0 ELSE 1 END AS has_song,
+		       COALESCE(mf.file_size, 0), COALESCE(mf.file_mtime_ns, 0),
+		       COALESCE(mf.last_seen_at, '1970-01-01T00:00:00Z')
 		FROM media_files mf
 		LEFT JOIN songs s ON s.media_file_id = mf.id
 		WHERE mf.library_id = ?
@@ -85,7 +110,7 @@ func (r *MediaRepository) ListExistingByLibrary(libraryID int) ([]ExistingMediaF
 	for rows.Next() {
 		var item ExistingMediaFile
 		var hasSong int
-		if err := rows.Scan(&item.Path, &item.MediaFileID, &hasSong); err != nil {
+		if err := rows.Scan(&item.Path, &item.MediaFileID, &hasSong, &item.FileSize, &item.FileMtimeNs, &item.LastSeenAt); err != nil {
 			return nil, err
 		}
 		item.HasSong = hasSong == 1
@@ -94,6 +119,34 @@ func (r *MediaRepository) ListExistingByLibrary(libraryID int) ([]ExistingMediaF
 	return existing, nil
 }

+func (r *MediaRepository) UpdateMetadata(mediaFile model.MediaFile) error {
+	_, err := r.db.Exec(
+		`UPDATE media_files SET file_size = ?, file_mtime_ns = ?, format = ?, bit_rate = ?, sample_rate = ?, last_seen_at = ?, updated_at = CURRENT_TIMESTAMP WHERE id = ?`,
+		mediaFile.FileSize,
+		mediaFile.FileMtimeNs,
+		mediaFile.Format,
+		mediaFile.BitRate,
+		mediaFile.SampleRate,
+		mediaFile.LastSeenAt,
+		mediaFile.ID,
+	)
+	return err
+}
+
+func (r *MediaRepository) TouchLastSeen(mediaFileID int, seenAt time.Time) error {
+	_, err := r.db.Exec(
+		`UPDATE media_files SET last_seen_at = ?, updated_at = CURRENT_TIMESTAMP WHERE id = ?`,
+		seenAt,
+		mediaFileID,
+	)
+	return err
+}
+
+func (r *MediaRepository) Delete(id int) error {
+	_, err := r.db.Exec("DELETE FROM media_files WHERE id = ?", id)
+	return err
+}
+
 func (r *MediaRepository) GetSongsByLibraryWithDetails(libraryID int) ([]model.SongDetail, error) {
 	rows, err := r.db.Query(`
 		SELECT s.id, s.title, a.name as artist_name, al.title as album_title, s.duration, mf.path
--- a/internal/repository/song_repo.go
+++ b/internal/repository/song_repo.go
@@ -130,3 +130,11 @@ func (r *SongRepository) Create(title string, artistID, albumID, duration, media
 		MediaFileID: mediaFileID,
 	}, nil
 }
+
+func (r *SongRepository) UpdateByMediaFileID(title string, artistID, albumID, duration, mediaFileID int) error {
+	_, err := r.db.Exec(
+		"UPDATE songs SET title = ?, artist_id = ?, album_id = ?, duration = ? WHERE media_file_id = ?",
+		title, artistID, albumID, duration, mediaFileID,
+	)
+	return err
+}