add service

cmd/server/main.go

@@ -0,0 +1,88 @@
+package main
+
+import (
+	"context"
+	"log"
+
+	"github.com/jackc/pgx/v5/pgxpool"
+	_ "github.com/jackc/pgx/v5/stdlib"
+	rkboot "github.com/rookie-ninja/rk-boot/v2"
+	"github.com/rookie-ninja/rk-entry/v2/entry"
+	rkgrpc "github.com/rookie-ninja/rk-grpc/v2/boot"
+	"google.golang.org/grpc"
+
+	"smart-search-back/internal/config"
+	"smart-search-back/internal/database"
+	grpcServer "smart-search-back/internal/grpc"
+	"smart-search-back/internal/repository"
+	"smart-search-back/internal/worker"
+)
+
+func main() {
+	cfg, err := config.Load("config/config.yaml")
+	if err != nil {
+		log.Fatalf("Failed to load config: %v", err)
+	}
+
+	boot := rkboot.NewBoot(rkboot.WithBootConfigPath("config/boot.yaml", nil))
+
+	ctx := context.Background()
+	boot.Bootstrap(ctx)
+
+	grpcEntry := rkgrpc.GetGrpcEntry("smart-search-service")
+	if grpcEntry == nil {
+		log.Fatal("Failed to get gRPC entry from rk-boot")
+	}
+
+	loggerEntry := rkentry.GlobalAppCtx.GetLoggerEntry("smart-search-service")
+	if loggerEntry == nil {
+		loggerEntry = rkentry.GlobalAppCtx.GetLoggerEntryDefault()
+	}
+	logger := loggerEntry.Logger
+
+	if err := database.RunMigrations(cfg.DatabaseURL()); err != nil {
+		log.Fatalf("Failed to run migrations: %v", err)
+	}
+
+	pool, err := pgxpool.New(ctx, cfg.DatabaseURL())
+	if err != nil {
+		log.Fatalf("Failed to connect to database: %v", err)
+	}
+	defer pool.Close()
+
+	if err := pool.Ping(ctx); err != nil {
+		log.Fatalf("Failed to ping database: %v", err)
+	}
+
+	log.Println("Successfully connected to database")
+
+	sessionRepo := repository.NewSessionRepository(pool)
+	inviteRepo := repository.NewInviteRepository(pool)
+
+	sessionCleaner := worker.NewSessionCleaner(ctx, sessionRepo)
+	sessionCleaner.Start()
+	defer sessionCleaner.Stop()
+
+	inviteCleaner := worker.NewInviteCleaner(ctx, inviteRepo)
+	inviteCleaner.Start()
+	defer inviteCleaner.Stop()
+
+	authHandler, userHandler, inviteHandler, requestHandler, supplierHandler := grpcServer.NewHandlers(
+		pool,
+		cfg.Security.JWTSecret,
+		cfg.Security.CryptoSecret,
+		cfg.AI.OpenAIKey,
+		cfg.AI.PerplexityKey,
+		logger,
+	)
+
+	grpcEntry.AddRegFuncGrpc(func(s *grpc.Server) {
+		grpcServer.RegisterServices(s, authHandler, userHandler, inviteHandler, requestHandler, supplierHandler)
+	})
+
+	log.Println("gRPC server started via rk-boot on port 9091")
+
+	boot.WaitForShutdownSig(ctx)
+
+	log.Println("Server stopped gracefully")
+}