fix drone and sqlc generation

main.go

@@ -23,12 +23,14 @@ import (
 )
 
 var (
-	bindDisableTls   bool
-	sha1ver          string // sha1 revision used to build the program
-	buildTime        string // when the executable was built
-	cronFrequency    time.Duration
-	cronInvFrequency time.Duration
-	encryptionKey    = []byte("5L1l3B5KvwOCzUHMAlCgsgUTRAYMfSpa")
+	bindDisableTls         bool
+	sha1ver                string // sha1 revision used to build the program
+	buildTime              string // when the executable was built
+	cronFrequency          time.Duration
+	cronInvFrequency       time.Duration
+	cronSnapshotFrequency  time.Duration
+	cronAggregateFrequency time.Duration
+	encryptionKey          = []byte("5L1l3B5KvwOCzUHMAlCgsgUTRAYMfSpa")
 )
 
 func main() {
@@ -189,6 +191,30 @@ func main() {
 	}
 	logger.Debug("Setting VM inventory polling cronjob frequency to", "frequency", cronInvFrequency)
 
+	cronSnapshotFrequencyString := os.Getenv("VCENTER_INVENTORY_SNAPSHOT_SECONDS")
+	if cronSnapshotFrequencyString != "" {
+		cronSnapshotFrequency, err = time.ParseDuration(cronSnapshotFrequencyString)
+		if err != nil {
+			slog.Error("Can't convert VCENTER_INVENTORY_SNAPSHOT_SECONDS value to time duration. Defaulting to 3600", "value", cronSnapshotFrequencyString, "error", err)
+			cronSnapshotFrequency = time.Hour
+		}
+	} else {
+		cronSnapshotFrequency = time.Hour
+	}
+	logger.Debug("Setting VM inventory snapshot cronjob frequency to", "frequency", cronSnapshotFrequency)
+
+	cronAggregateFrequencyString := os.Getenv("VCENTER_INVENTORY_AGGREGATE_SECONDS")
+	if cronAggregateFrequencyString != "" {
+		cronAggregateFrequency, err = time.ParseDuration(cronAggregateFrequencyString)
+		if err != nil {
+			slog.Error("Can't convert VCENTER_INVENTORY_AGGREGATE_SECONDS value to time duration. Defaulting to 86400", "value", cronAggregateFrequencyString, "error", err)
+			cronAggregateFrequency = time.Hour * 24
+		}
+	} else {
+		cronAggregateFrequency = time.Hour * 24
+	}
+	logger.Debug("Setting VM inventory aggregation cronjob frequency to", "frequency", cronAggregateFrequency)
+
 	// start background processing for events stored in events table
 	startsAt := time.Now().Add(time.Second * 10)
 	job, err := c.NewJob(
@@ -219,6 +245,65 @@ func main() {
 	}
 	logger.Debug("Created vcenter inventory cron job", "job", job2.ID(), "starting_at", startsAt2)
 
+	startsAt3 := time.Now().Add(cronSnapshotFrequency)
+	if cronSnapshotFrequency == time.Hour {
+		startsAt3 = time.Now().Truncate(time.Hour).Add(time.Hour)
+	}
+	job3, err := c.NewJob(
+		gocron.DurationJob(cronSnapshotFrequency),
+		gocron.NewTask(func() {
+			ct.RunVcenterSnapshotHourly(ctx, logger)
+		}), gocron.WithSingletonMode(gocron.LimitModeReschedule),
+		gocron.WithStartAt(gocron.WithStartDateTime(startsAt3)),
+	)
+	if err != nil {
+		logger.Error("failed to start vcenter inventory snapshot cron job", "error", err)
+		os.Exit(1)
+	}
+	logger.Debug("Created vcenter inventory snapshot cron job", "job", job3.ID(), "starting_at", startsAt3)
+
+	startsAt4 := time.Now().Add(cronAggregateFrequency)
+	if cronAggregateFrequency == time.Hour*24 {
+		now := time.Now()
+		startsAt4 = time.Date(now.Year(), now.Month(), now.Day()+1, 0, 0, 0, 0, now.Location())
+	}
+	job4, err := c.NewJob(
+		gocron.DurationJob(cronAggregateFrequency),
+		gocron.NewTask(func() {
+			ct.RunVcenterDailyAggregate(ctx, logger)
+		}), gocron.WithSingletonMode(gocron.LimitModeReschedule),
+		gocron.WithStartAt(gocron.WithStartDateTime(startsAt4)),
+	)
+	if err != nil {
+		logger.Error("failed to start vcenter inventory aggregation cron job", "error", err)
+		os.Exit(1)
+	}
+	logger.Debug("Created vcenter inventory aggregation cron job", "job", job4.ID(), "starting_at", startsAt4)
+
+	job5, err := c.NewJob(
+		gocron.CronJob("0 0 1 * *", false),
+		gocron.NewTask(func() {
+			ct.RunVcenterMonthlyAggregate(ctx, logger)
+		}), gocron.WithSingletonMode(gocron.LimitModeReschedule),
+	)
+	if err != nil {
+		logger.Error("failed to start vcenter monthly aggregation cron job", "error", err)
+		os.Exit(1)
+	}
+	logger.Debug("Created vcenter monthly aggregation cron job", "job", job5.ID())
+
+	job6, err := c.NewJob(
+		gocron.CronJob("0 30 2 * *", false),
+		gocron.NewTask(func() {
+			ct.RunSnapshotCleanup(ctx, logger)
+		}), gocron.WithSingletonMode(gocron.LimitModeReschedule),
+	)
+	if err != nil {
+		logger.Error("failed to start snapshot cleanup cron job", "error", err)
+		os.Exit(1)
+	}
+	logger.Debug("Created snapshot cleanup cron job", "job", job6.ID())
+
 	// start cron scheduler
 	c.Start()