feat(app): add TaskPoller, wire DI, and add task integration tests

internal/app/app.go

@@ -28,6 +28,8 @@ type App struct {
 	db            *gorm.DB
 	server        *http.Server
 	cancelCleanup context.CancelFunc
+	taskSvc       *service.TaskService
+	taskPoller    *TaskPoller
 }
 
 // NewApp initializes all application dependencies: DB, Slurm client, services, handlers, router.
@@ -43,7 +45,7 @@ func NewApp(cfg *config.Config, logger *zap.Logger) (*App, error) {
 		return nil, err
 	}
 
-	srv, cancelCleanup := initHTTPServer(cfg, gormDB, slurmClient, logger)
+	srv, cancelCleanup, taskSvc, taskPoller := initHTTPServer(cfg, gormDB, slurmClient, logger)
 
 	return &App{
 		cfg:           cfg,
@@ -51,6 +53,8 @@ func NewApp(cfg *config.Config, logger *zap.Logger) (*App, error) {
 		db:            gormDB,
 		server:        srv,
 		cancelCleanup: cancelCleanup,
+		taskSvc:       taskSvc,
+		taskPoller:    taskPoller,
 	}, nil
 }
 
@@ -86,6 +90,14 @@ func (a *App) Run() error {
 func (a *App) Close() error {
 	var errs []error
 
+	if a.taskSvc != nil {
+		a.taskSvc.StopProcessor()
+	}
+
+	if a.taskPoller != nil {
+		a.taskPoller.Stop()
+	}
+
 	if a.cancelCleanup != nil {
 		a.cancelCleanup()
 	}
@@ -145,15 +157,15 @@ func initSlurmClient(cfg *config.Config) (*slurm.Client, error) {
 	return client, nil
 }
 
-func initHTTPServer(cfg *config.Config, db *gorm.DB, slurmClient *slurm.Client, logger *zap.Logger) (*http.Server, context.CancelFunc) {
+func initHTTPServer(cfg *config.Config, db *gorm.DB, slurmClient *slurm.Client, logger *zap.Logger) (*http.Server, context.CancelFunc, *service.TaskService, *TaskPoller) {
+	ctx := context.Background()
+
 	jobSvc := service.NewJobService(slurmClient, logger)
 	clusterSvc := service.NewClusterService(slurmClient, logger)
 	jobH := handler.NewJobHandler(jobSvc, logger)
 	clusterH := handler.NewClusterHandler(clusterSvc, logger)
 
 	appStore := store.NewApplicationStore(db)
-	appSvc := service.NewApplicationService(appStore, jobSvc, cfg.WorkDirBase, logger)
-	appH := handler.NewApplicationHandler(appSvc, logger)
 
 	// File storage initialization
 	minioClient, err := storage.NewMinioClient(cfg.Minio)
@@ -165,9 +177,12 @@ func initHTTPServer(cfg *config.Config, db *gorm.DB, slurmClient *slurm.Client,
 	var fileH *handler.FileHandler
 	var folderH *handler.FolderHandler
 
+	taskStore := store.NewTaskStore(db)
+	fileStore := store.NewFileStore(db)
+	blobStore := store.NewBlobStore(db)
+
+	var stagingSvc *service.FileStagingService
 	if minioClient != nil {
-		blobStore := store.NewBlobStore(db)
-		fileStore := store.NewFileStore(db)
 		folderStore := store.NewFolderStore(db)
 		uploadStore := store.NewUploadStore(db)
 
@@ -179,25 +194,34 @@ func initHTTPServer(cfg *config.Config, db *gorm.DB, slurmClient *slurm.Client,
 		fileH = handler.NewFileHandler(fileSvc, logger)
 		folderH = handler.NewFolderHandler(folderSvc, logger)
 
-		cleanupCtx, cancelCleanup := context.WithCancel(context.Background())
-		go startCleanupWorker(cleanupCtx, uploadStore, minioClient, cfg.Minio.Bucket, logger)
-
-		router := server.NewRouter(jobH, clusterH, appH, uploadH, fileH, folderH, logger)
-
-		addr := ":" + cfg.ServerPort
-
-		return &http.Server{
-			Addr:    addr,
-			Handler: router,
-		}, cancelCleanup
+		stagingSvc = service.NewFileStagingService(fileStore, blobStore, minioClient, cfg.Minio.Bucket, logger)
 	}
 
-	router := server.NewRouter(jobH, clusterH, appH, uploadH, fileH, folderH, logger)
+	taskSvc := service.NewTaskService(taskStore, appStore, fileStore, blobStore, stagingSvc, jobSvc, cfg.WorkDirBase, logger)
+	taskSvc.StartProcessor(ctx)
+
+	appSvc := service.NewApplicationService(appStore, jobSvc, cfg.WorkDirBase, logger, taskSvc)
+	appH := handler.NewApplicationHandler(appSvc, logger)
+
+	poller := NewTaskPoller(taskSvc, 10*time.Second, logger)
+	poller.Start(ctx)
+
+	taskH := handler.NewTaskHandler(taskSvc, logger)
+
+	var cancelCleanup context.CancelFunc
+
+	if minioClient != nil {
+		cleanupCtx, cancel := context.WithCancel(context.Background())
+		cancelCleanup = cancel
+		go startCleanupWorker(cleanupCtx, store.NewUploadStore(db), minioClient, cfg.Minio.Bucket, logger)
+	}
+
+	router := server.NewRouter(jobH, clusterH, appH, uploadH, fileH, folderH, taskH, logger)
 
 	addr := ":" + cfg.ServerPort
 
 	return &http.Server{
 		Addr:    addr,
 		Handler: router,
-	}, nil
+	}, cancelCleanup, taskSvc, poller
 }