fix: data race

Author: jibon57

internal/scheduler.go

@@ -1,19 +1,20 @@
 package internal
 
 import (
+	"context"
 	"log"
 	"os"
 	"path"
 	"time"
 )
 
-func StartScheduler(shutdown chan os.Signal) {
+func StartScheduler(ctx context.Context) {
 	hourlyChecker := time.NewTicker(1 * time.Hour)
 	defer hourlyChecker.Stop()
 
 	for {
 		select {
-		case <-shutdown:
+		case <-ctx.Done():
 			log.Println("Stopping scheduler")
 			return
 		case <-hourlyChecker.C:

main.go

@@ -1,11 +1,13 @@
 package main
 
 import (
+	"context"
 	"fmt"
 	"log"
 	"os"
 	"os/signal"
 	"syscall"
+	"time"
 
 	"github.com/jibon57/secure-file-download-server/internal"
 	"gopkg.in/yaml.v3"
@@ -34,17 +36,17 @@ func main() {
 	sigChan := make(chan os.Signal, 1)
 	signal.Notify(sigChan, syscall.SIGINT, syscall.SIGTERM, syscall.SIGQUIT)
 
+	ctx, cancel := context.WithCancel(context.Background())
+	defer cancel()
+
 	// start scheduler
-	go internal.StartScheduler(sigChan)
+	go internal.StartScheduler(ctx)
 
 	go func() {
 		sig := <-sigChan
+		ctx.Done()
 		log.Println("exit requested, shutting down", "signal", sig)
-		err = router.Shutdown()
-		if err != nil {
-			log.Fatalln(err)
-		}
-		log.Println("server shutdown")
+		router.ShutdownWithTimeout(time.Second)
 	}()
 
 	err = router.Listen(fmt.Sprintf(":%d", internal.AppCnf.Port))