main.go

package main

import (
	"context"
	"log"
	"os"
	"os/signal"
	"syscall"
	"time"
)

func main() {
	// Load configuration from environment variables
	cfg, err := LoadConfig()
	if err != nil {
		log.Fatalf("failed to load configuration: %v", err)
	}

	// Create a new VM manager
	vmManager, err := NewVMManager(cfg)
	if err != nil {
		log.Fatalf("failed to create VM manager: %v", err)
	}
	defer vmManager.Close()

	// Initialize database tables
	if err := vmManager.InitDatabase(); err != nil {
		log.Fatalf("failed to initialize database: %v", err)
	}

	// Setup NATS subscriptions
	if err := vmManager.SetupNATSSubscriptions(); err != nil {
		log.Fatalf("failed to set up NATS subscriptions: %v", err)
	}

	ctx, cancel := context.WithCancel(context.Background())
	defer cancel()

	// Start the background VM expiration checker
	vmManager.StartExpirationChecker(ctx, 1*time.Minute)

	// Start the background cooldown expiration checker
	vmManager.StartCooldownExpirationChecker(ctx, 1*time.Minute)

	log.Println("VM manager service started and listening for NATS messages...")

	// Wait for termination signal
	done := make(chan os.Signal, 1)
	signal.Notify(done, syscall.SIGINT, syscall.SIGTERM)

	// Block until a signal is received
	<-done
	log.Println("shutting down...")
}