From dec0490c46fa2388ba7ecb627abb93c7c2883c55 Mon Sep 17 00:00:00 2001 From: Ryan Yeske Date: Sat, 20 Jun 2026 11:48:40 -0700 Subject: [PATCH] Add timeout --- worker/youtubedownload/youtubedownload.go | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/worker/youtubedownload/youtubedownload.go b/worker/youtubedownload/youtubedownload.go index 0ef7cf4..38c01b4 100644 --- a/worker/youtubedownload/youtubedownload.go +++ b/worker/youtubedownload/youtubedownload.go @@ -5,6 +5,7 @@ import ( "fmt" "log" "os" + "time" "oj/api" @@ -28,6 +29,10 @@ func NewWorker(q *api.Queries) *Worker { return &Worker{Queries: q} } +func (w *Worker) Timeout(job *river.Job[YoutubeDownloadArgs]) time.Duration { + return 5 * time.Minute +} + func dlPath() string { p := os.Getenv("MUSIC_DOWNLOAD_PATH") if p == "" { @@ -37,6 +42,9 @@ func dlPath() string { } func (w *Worker) Work(ctx context.Context, job *river.Job[YoutubeDownloadArgs]) error { + ctx, cancel := context.WithTimeout(ctx, time.Minute) + defer cancel() + trackID := job.Args.MusicTrackID track, err := w.Queries.MusicTrackByID(ctx, trackID)