Files
butterfliu/internal/repository/media_repo.go
lzw-723 a8d976b97e
All checks were successful
Go CI / test-and-build (push) Successful in 43s
清理代码库
2026-04-06 13:17:57 +08:00

27 lines
546 B
Go

package repository
import "database/sql"
type MediaRepository struct {
db *sql.DB
}
func NewMediaRepository(db *sql.DB) *MediaRepository {
return &MediaRepository{db: db}
}
func (r *MediaRepository) Get(id int) (MediaFile, error) {
var m MediaFile
rows, err := r.db.Query("SELECT id, path, library_id FROM media_files WHERE id = ?", id)
if err != nil {
return MediaFile{}, err
}
defer rows.Close()
if rows.Next() {
if err := rows.Scan(&m.ID, &m.Path, &m.LibraryID); err != nil {
return MediaFile{}, err
}
}
return m, nil
}