postgres

cmd/main.go

@@ -13,10 +13,9 @@ import (
 	"gitea.d3m0k1d.ru/HellreigN/Control-plane/internal/auth"
 	"gitea.d3m0k1d.ru/HellreigN/Control-plane/internal/config"
 	"github.com/gin-gonic/gin"
+	"github.com/jackc/pgx/v5/pgxpool"
 	"github.com/swaggo/files"
 	"github.com/swaggo/gin-swagger"
-	"go.mongodb.org/mongo-driver/v2/mongo"
-	"go.mongodb.org/mongo-driver/v2/mongo/options"
 )
 
 // @title         AegisGuard API
@@ -38,23 +37,23 @@ func main() {
 	ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
 	defer cancel()
 
-	client, err := mongo.Connect(options.Client().ApplyURI(cfg.MongoURI).SetTimeout(10 * time.Second))
+	pool, err := pgxpool.New(ctx, cfg.DatabaseURL)
 	if err != nil {
-		log.Fatalf("failed to create mongodb client: %v", err)
+		log.Fatalf("failed to create postgres pool: %v", err)
 	}
+	defer pool.Close()
 
-	if err := client.Ping(ctx, nil); err != nil {
-		log.Fatalf("failed to ping mongodb: %v", err)
+	if err := pool.Ping(ctx); err != nil {
+		log.Fatalf("failed to ping postgres: %v", err)
 	}
-	log.Println("connected to mongodb")
+	log.Println("connected to postgres")
 
-	db := client.Database(cfg.MongoDB)
+	repo := auth.NewRepository(pool)
 
-	repo := auth.NewRepository(db)
-
-	if err := repo.EnsureIndexes(ctx); err != nil {
-		log.Printf("warning: failed to ensure indexes: %v", err)
+	if err := repo.Migrate(ctx); err != nil {
+		log.Fatalf("failed to run migrations: %v", err)
 	}
+	log.Println("migrations applied")
 
 	svc := auth.NewService(repo, cfg.JWTSecret, cfg.JWTExpiration, cfg.JWTRefreshExpiration)
 	handler := auth.NewHandler(svc)
@@ -106,9 +105,7 @@ func main() {
 		log.Fatalf("server forced to shutdown: %v", err)
 	}
 
-	if err := client.Disconnect(shutdownCtx); err != nil {
-		log.Printf("failed to disconnect mongodb: %v", err)
-	}
+	pool.Close()
 
 	log.Println("server stopped")
 }