pocket-id-pocket-id-2/backend/internal/bootstrap/bootstrap.go

package bootstrap

import (
	"context"
	"fmt"
	"log"
	"time"

	_ "github.com/golang-migrate/migrate/v4/source/file"

	"github.com/pocket-id/pocket-id/backend/internal/job"
	"github.com/pocket-id/pocket-id/backend/internal/utils"
	"github.com/pocket-id/pocket-id/backend/internal/utils/signals"
)

func Bootstrap() error {
	// Get a context that is canceled when the application is stopping
	ctx := signals.SignalContext(context.Background())

	initApplicationImages()

	// Perform migrations for changes
	migrateConfigDBConnstring()
	migrateKey()

	// Connect to the database
	db := newDatabase()

	// Create all services
	svc, err := initServices(ctx, db)
	if err != nil {
		return fmt.Errorf("failed to initialize services: %w", err)
	}

	// Init the job scheduler
	scheduler, err := job.NewScheduler()
	if err != nil {
		return fmt.Errorf("failed to create job scheduler: %w", err)
	}
	err = registerScheduledJobs(ctx, db, svc, scheduler)
	if err != nil {
		return fmt.Errorf("failed to register scheduled jobs: %w", err)
	}

	// Init the router
	router := initRouter(db, svc)

	// Run all background serivces
	// This call blocks until the context is canceled
	err = utils.
		NewServiceRunner(router, scheduler.Run).
		Run(ctx)
	if err != nil {
		return fmt.Errorf("failed to run services: %w", err)
	}

	// Invoke all shutdown functions
	// We give these a timeout of 5s
	shutdownCtx, shutdownCancel := context.WithTimeout(context.Background(), 5*time.Second)
	defer shutdownCancel()
	err = utils.
		// TODO: Add shutdown services here
		NewServiceRunner().
		Run(shutdownCtx)
	if err != nil {
		log.Printf("Error shutting down services: %v", err)
	}

	return nil
}