Updated redis package for main.go

Author: PythonHacker24

cmd/laclm/main.go

@@ -11,17 +11,19 @@ import (
 	"time"
 
 	"github.com/MakeNowJust/heredoc"
+	"github.com/jackc/pgx/v5"
 	"github.com/joho/godotenv"
 	"github.com/spf13/cobra"
 	"go.uber.org/automaxprocs/maxprocs"
 	"go.uber.org/zap"
 
 	"github.com/PythonHacker24/linux-acl-management-backend/api/routes"
 	"github.com/PythonHacker24/linux-acl-management-backend/config"
+	"github.com/PythonHacker24/linux-acl-management-backend/internal/postgresql"
+	"github.com/PythonHacker24/linux-acl-management-backend/internal/redis"
 	"github.com/PythonHacker24/linux-acl-management-backend/internal/scheduler"
 	"github.com/PythonHacker24/linux-acl-management-backend/internal/scheduler/fcfs"
 	"github.com/PythonHacker24/linux-acl-management-backend/internal/session"
-	"github.com/PythonHacker24/linux-acl-management-backend/internal/redis"
 	"github.com/PythonHacker24/linux-acl-management-backend/internal/transprocessor"
 	"github.com/PythonHacker24/linux-acl-management-backend/internal/utils"
 )
@@ -131,6 +133,14 @@ func run(ctx context.Context) error {
 		zap.L().Fatal("Failed to connect to Redis", zap.Error(err))
 	}
 
+	connPQ, err := pgx.Connect(context.Background(), os.Getenv("DATABASE_URL"))
+    if err != nil {
+            fmt.Fprintf(os.Stderr, "Unable to connect to database: %v\n", err)
+            os.Exit(1)
+    }
+
+	archivalPQ := postgresql.New(connPQ)
+
 	/* 
 		initializing schedular 
 		scheduler uses context to quit - part of waitgroup
@@ -139,7 +149,7 @@ func run(ctx context.Context) error {
 	errCh := make(chan error, 1)
 
 	/* create a session manager */
-	sessionManager := session.NewManager(logRedisClient)
+	sessionManager := session.NewManager(logRedisClient, archivalPQ)
 
 	/* create a permissions processor */
 	permProcessor := transprocessor.NewPermProcessor()
@@ -218,6 +228,9 @@ func run(ctx context.Context) error {
 
 	wg.Wait()
 
+	/* close archival database connection */
+    connPQ.Close(context.Background())
+
 	zap.L().Info("All background processes closed gracefully")
 
 	return err