lzw-723/butterfliu
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

实现增量扫描
All checks were successful
Go CI / test-and-build (push) Successful in 13s
Details
Web CI / lint-test-build (push) Successful in 31s
Details

Browse Source
This commit is contained in:
lzw-723
2026-04-11 14:38:23 +08:00
parent 3f82e29c6c
commit 79d47c81e7
10 changed files with 351 additions and 91 deletions
Download Patch File Download Diff File Expand all files Collapse all files

79
internal/repository/media_repo.go
View File

@@ -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