feat: server supports to cancel jobs

chore(deps): update ffmpeg build script to latest

Author: iaymerich

Dockerfile

@@ -1,8 +1,6 @@
-ARG BASE_IMAGE=ubuntu:22.04@sha256:77906da86b60585ce12215807090eb327e7386c8fafb5402369e421f44eff17e
+ARG BASE_IMAGE=ubuntu:24.04
 FROM ${BASE_IMAGE} AS builder-ffmpeg
 
-ARG FFMPEG_BUILD_SCRIPT_VERSION=1.48
-
 ENV DEBIAN_FRONTEND=noninteractive
 
 RUN apt-get update \
@@ -20,9 +18,10 @@ RUN apt-get update \
 
 WORKDIR /app
 
+ARG FFMPEG_BUILD_SCRIPT_VERSION=7ea5427c1f91d5eb3400f47dfe5e425844e4af35
 # ADD doesn't cache when used from URL
 RUN curl -sLO \
-    https://raw.githubusercontent.com/markus-perl/ffmpeg-build-script/v${FFMPEG_BUILD_SCRIPT_VERSION}/build-ffmpeg && \
+    https://raw.githubusercontent.com/markus-perl/ffmpeg-build-script/${FFMPEG_BUILD_SCRIPT_VERSION}/build-ffmpeg && \
     chmod 755 ./build-ffmpeg && \
     SKIPINSTALL=yes ./build-ffmpeg \
         --build \

server/scheduler/scheduler.go

@@ -31,6 +31,7 @@ type Scheduler interface {
 	GetChecksum(ctx context.Context, uuid string) (string, error)
 	RequestJob(ctx context.Context, workerName string) (*model.TaskEncode, error)
 	HandleWorkerEvent(ctx context.Context, taskEvent *model.TaskEvent) error
+	CancelJob(ctx context.Context, id string) error
 }
 
 type SchedulerConfig struct {
@@ -97,6 +98,30 @@ func (R *RuntimeScheduler) HandleWorkerEvent(ctx context.Context, jobEvent *mode
 	return nil
 }
 
+func (R *RuntimeScheduler) CancelJob(ctx context.Context, id string) error {
+	job, err := R.repo.GetJob(ctx, id)
+	if err != nil {
+		return err
+	}
+
+	status := job.Events.GetStatus()
+	switch {
+	case status == model.CompletedNotificationStatus:
+		return fmt.Errorf("job already completed")
+	case status == model.FailedNotificationStatus:
+		return fmt.Errorf("job is failed")
+	case status == model.CanceledNotificationStatus:
+		return fmt.Errorf("job already canceled")
+	case status == model.AssignedNotificationStatus, status == model.StartedNotificationStatus:
+		newEvent := job.AddEventComplete(model.NotificationEvent, model.JobNotification, model.CanceledNotificationStatus, "Job canceled by user")
+		err = R.repo.AddNewTaskEvent(ctx, newEvent)
+		if err != nil {
+			return err
+		}
+	}
+	return fmt.Errorf("job %s is in unknown state", id)
+}
+
 func (R *RuntimeScheduler) processEvent(ctx context.Context, taskEvent *model.TaskEvent) error {
 	var err error
 	switch taskEvent.EventType {

server/web/web.go

@@ -98,6 +98,19 @@ func (W *WebServer) addJobs(writer http.ResponseWriter, request *http.Request) {
 	writer.Write(b)
 }
 
+func (W *WebServer) cancelJobs(writer http.ResponseWriter, request *http.Request) {
+	vars := mux.Vars(request)
+	jobId := vars["jobid"]
+	if jobId == "" {
+		webError(writer, fmt.Errorf("jobId is mandatory"), 400)
+		return
+	}
+	err := W.scheduler.CancelJob(W.ctx, jobId)
+	if webError(writer, err, 500) {
+		return
+	}
+}
+
 func (W *WebServer) upload(writer http.ResponseWriter, request *http.Request) {
 	workerName := request.Header.Get("workerName")
 	if workerName == "" {
@@ -251,6 +264,7 @@ func NewWebServer(config WebServerConfig, scheduler scheduler.Scheduler, updater
 		},
 	}
 	rtr.Handle("/api/v1/job/", webServer.AuthFunc(webServer.addJobs)).Methods("POST")
+	rtr.Handle("/api/v1/job/{jobid}", webServer.AuthFunc(webServer.cancelJobs)).Methods("DELETE")
 	rtr.Handle("/api/v1/job/request", webServer.AuthFunc(webServer.requestJob)).Methods("GET")
 	rtr.Handle("/api/v1/event", webServer.AuthFunc(webServer.handleWorkerEvent)).Methods("POST")
 	rtr.HandleFunc("/api/v1/download", webServer.download).Methods("GET")