Refactor main function: replace Redis client with server initialization and add video handler for processing tasks

Author: Prateet-Github

cmd/worker/main.go

@@ -4,19 +4,29 @@ import (
 	"log"
 
 	"github.com/Prateet-Github/worker-go/internal/config"
+	"github.com/Prateet-Github/worker-go/internal/handlers"
 	"github.com/Prateet-Github/worker-go/internal/queue"
+
+	"github.com/hibiken/asynq"
 )
 
 func main() {
 	cfg := config.Load()
 
-	client := queue.NewClient(cfg)
-	defer client.Close()
+	server := queue.NewServer(cfg)
 
-	if err := queue.Ping(client); err != nil {
-		log.Fatal("Redis connection failed:", err)
-	}
+	videoHandler := handlers.NewVideoHandler()
+
+	mux := asynq.NewServeMux()
+
+	mux.HandleFunc(
+		queue.TypeProcessVideo,
+		videoHandler.ProcessVideo,
+	)
 
-	log.Println("Connected to Redis")
 	log.Println("Worker started")
+
+	if err := server.Run(mux); err != nil {
+		log.Fatal(err)
+	}
 }

internal/handlers/video.go

@@ -0,0 +1,34 @@
+package handlers
+
+import (
+	"context"
+	"encoding/json"
+	"log"
+
+	"github.com/Prateet-Github/worker-go/internal/queue"
+	"github.com/hibiken/asynq"
+)
+
+type VideoHandler struct{}
+
+func NewVideoHandler() *VideoHandler {
+	return &VideoHandler{}
+}
+
+func (h *VideoHandler) ProcessVideo(
+	ctx context.Context,
+	task *asynq.Task,
+) error {
+
+	var payload queue.VideoTask
+
+	if err := json.Unmarshal(task.Payload(), &payload); err != nil {
+		return err
+	}
+
+	log.Println("Processing video...")
+	log.Println("Video ID:", payload.VideoID)
+	log.Println("S3 Key:", payload.S3Key)
+
+	return nil
+}