106 lines
1.9 KiB
Go
106 lines
1.9 KiB
Go
package moviestore
|
|
|
|
import (
|
|
"database/sql"
|
|
"errors"
|
|
"log/slog"
|
|
"time"
|
|
)
|
|
|
|
type JobStatus string
|
|
|
|
const (
|
|
JobStatusToDo JobStatus = "todo"
|
|
JobStatusDoing JobStatus = "doing"
|
|
JobStatusDone JobStatus = "done"
|
|
)
|
|
|
|
type Action string
|
|
|
|
const (
|
|
interval = 10 * time.Second
|
|
|
|
ActionFetchIMDBReviews Action = "fetch-imdb-reviews"
|
|
)
|
|
|
|
type Job struct {
|
|
ID int
|
|
MovieID string
|
|
Action Action
|
|
Status JobStatus
|
|
}
|
|
|
|
type JobQueue struct {
|
|
db *SQLite
|
|
out chan Job
|
|
logger *slog.Logger
|
|
}
|
|
|
|
func NewJobQueue(db *SQLite, logger *slog.Logger) *JobQueue {
|
|
return &JobQueue{
|
|
db: db,
|
|
out: make(chan Job),
|
|
logger: logger.With("service", "jobqueue"),
|
|
}
|
|
}
|
|
|
|
func (jq *JobQueue) Add(movieID string, action Action) error {
|
|
_, err := jq.db.Exec(`INSERT INTO job_queue (movie_id, action, status)
|
|
VALUES (?, ?, 'todo')`, movieID, action)
|
|
|
|
return err
|
|
}
|
|
|
|
func (jq *JobQueue) Next() chan Job {
|
|
return jq.out
|
|
}
|
|
|
|
func (jq *JobQueue) Run() {
|
|
logger := jq.logger.With("method", "run")
|
|
logger.Info("starting job queue")
|
|
for {
|
|
row := jq.db.QueryRow(`
|
|
SELECT id, movie_id, action
|
|
FROM job_queue
|
|
WHERE status='todo'
|
|
ORDER BY id DESC
|
|
LIMIT 1`)
|
|
|
|
var job Job
|
|
err := row.Scan(&job.ID, &job.MovieID, &job.Action)
|
|
switch {
|
|
case errors.Is(err, sql.ErrNoRows):
|
|
logger.Info("nothing to do")
|
|
time.Sleep(interval)
|
|
continue
|
|
case err != nil:
|
|
logger.Error("could not fetch next job", "error", row.Err())
|
|
time.Sleep(interval)
|
|
continue
|
|
}
|
|
logger.Info("found a job", "id", job.ID)
|
|
|
|
if _, err := jq.db.Exec(`
|
|
UPDATE job_queue
|
|
SET status='doing'
|
|
WHERE id=?`, job.ID); err != nil {
|
|
logger.Error("could not set job to doing", "error")
|
|
time.Sleep(interval)
|
|
continue
|
|
}
|
|
|
|
jq.out <- job
|
|
}
|
|
}
|
|
|
|
func (jq *JobQueue) MarkDone(id string) {
|
|
logger := jq.logger.With("method", "markdone")
|
|
if _, err := jq.db.Exec(`
|
|
UPDATE job_queue SET
|
|
status='done'
|
|
WHERE id=?`, id); err != nil {
|
|
logger.Error("could not mark job done", "error", err)
|
|
}
|
|
return
|
|
}
|