26 lines
498 B
Go
26 lines
498 B
Go
package repository
|
|
|
|
import (
|
|
"butterfliu/internal/model"
|
|
"database/sql"
|
|
)
|
|
|
|
type MediaRepository struct {
|
|
db *sql.DB
|
|
}
|
|
|
|
func NewMediaRepository(db *sql.DB) *MediaRepository {
|
|
return &MediaRepository{db: db}
|
|
}
|
|
|
|
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(
|
|
&m.ID, &m.Path, &m.LibraryID,
|
|
)
|
|
if err != nil {
|
|
return model.MediaFile{}, err
|
|
}
|
|
return m, nil
|
|
}
|