main.go

package main

import (
	"context"
	"os"
	"time"

	"github.com/cybertec-postgresql/pg_timetable/internal/cmdparser"
	"github.com/cybertec-postgresql/pg_timetable/internal/pgengine"
	"github.com/cybertec-postgresql/pg_timetable/internal/scheduler"
)

/**
 * pg_timetable is the daemon application responsible to execute scheduled SQL tasks that cannot be triggered by the
 * PostgreSQL server (PostgreSQL does not support time triggers).
 *
 * This application may run in the same machine as PostgreSQL server and must grant full access permission to the
 * timetable tables.
 */

func main() {
	ctx := context.Background()
	cmdOpts, err := cmdparser.Parse()
	if err != nil {

		pgengine.LogToDB("PANIC", "Error parsing command line arguments: ", err)
		os.Exit(2)
	}
	connctx, cancel := context.WithTimeout(ctx, 90*time.Second)
	defer cancel()
	if !pgengine.InitAndTestConfigDBConnection(connctx, *cmdOpts) {
		os.Exit(2)
	}
	defer pgengine.FinalizeConfigDBConnection()
	if cmdOpts.Upgrade {
		if !pgengine.MigrateDb(ctx) {
			os.Exit(3)
		}
	} else {
		if upgrade, err := pgengine.CheckNeedMigrateDb(ctx); upgrade || err != nil {
			os.Exit(3)
		}
	}
	if cmdOpts.Init {
		os.Exit(0)
	}
	pgengine.SetupCloseHandler()
	for scheduler.Run(ctx) == scheduler.ConnectionDroppped {
		pgengine.ReconnectDbAndFixLeftovers(ctx)
	}
}