feat: add graceful termination

cmd/server/main.go

@@ -4,6 +4,8 @@ import (
 	"log"
 	"os"
 	"os/exec"
+	"os/signal"
+	"syscall"
 
 	"github.com/moosh3/github-actions-aggregator/pkg/api"
 	"github.com/moosh3/github-actions-aggregator/pkg/config"
@@ -36,9 +38,22 @@ func main() {
 	githubClient := github.NewClient(cfg.GitHub.AccessToken)
 
 	workerPool := worker.NewWorkerPool(database, cfg.WorkerPoolSize)
+	workerPool.Start()
 
 	// Start the API server
-	api.StartServer(cfg, database, githubClient, workerPool)
+	go api.StartServer(cfg, database, githubClient, workerPool)
+
+	// Set up graceful shutdown
+	quit := make(chan os.Signal, 1)
+	signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM)
+	<-quit
+
+	log.Println("Shutting down server...")
+
+	// Stop the worker pool
+	workerPool.Stop()
+
+	log.Println("Server exiting")
 }
 
 func runMigrations() error {

pkg/api/handlers.go

@@ -74,6 +74,7 @@ func GetWorkflowStats(c *gin.Context) {
 		c.JSON(http.StatusBadRequest, gin.H{"error": "start_time must be before end_time"})
 		return
 	}
+
 	// Initialize counters
 	totalRuns := len(runs)
 	successCount := 0
@@ -125,5 +126,4 @@ func GetWorkflowStats(c *gin.Context) {
 		"start_time":            startTime.Format(time.RFC3339),
 		"end_time":              endTime.Format(time.RFC3339),
 	})
-
 }