dornröschen: add gokrazy-specific main

dornröschen/dornröschen.go

@@ -225,17 +225,14 @@ func sync(NASen []string) {
 	}
 }
 
-func main() {
-	log.SetFlags(log.LstdFlags | log.Lshortfile)
-	flag.Parse()
-
+func run() error {
 	if !*runBackup && !*runSync {
-		log.Fatal("Neither -backup nor -sync enabled, nothing to do.")
+		return fmt.Errorf("Neither -backup nor -sync enabled, nothing to do.")
 	}
 
 	storageList := strings.Split(*storageHosts, ",")
 	if len(storageList) > 2 {
-		log.Fatal("More than 2 -storage_hosts are not supported. Please send a patch to fix.")
+		return fmt.Errorf("More than 2 -storage_hosts are not supported. Please send a patch to fix.")
 	}
 
 	if *runBackup {
@@ -253,7 +250,5 @@ func main() {
 	prometheus.MustRegister(lastSuccess)
 	lastSuccess.Set(float64(time.Now().Unix()))
 
-	if err := prometheus.Push("dornroeschen", "dornroeschen", *pushGateway); err != nil {
-		log.Fatal(err)
-	}
+	return prometheus.Push("dornroeschen", "dornroeschen", *pushGateway)
 }

dornröschen/gokrazy.go

@@ -0,0 +1,44 @@
+// +build gokrazy
+
+package main
+
+import (
+	"log"
+	"time"
+
+	"github.com/gokrazy/gokrazy"
+)
+
+func main() {
+	log.SetFlags(log.LstdFlags | log.Lshortfile)
+
+	gokrazy.WaitForClock()
+
+	// Run forever, trigger a run at 10:00 each Monday through Friday.
+	for {
+		now := time.Now()
+		runToday := now.Hour() < 10 &&
+			now.Weekday() != time.Saturday &&
+			now.Weekday() != time.Sunday
+		today := now.Day()
+		log.Printf("now = %v, runToday = %v", now, runToday)
+		for {
+			if time.Now().Day() != today {
+				// Day changed, re-evaluate whether to run today.
+				break
+			}
+
+			nextHour := time.Now().Truncate(time.Hour).Add(1 * time.Hour)
+			log.Printf("today = %d, runToday = %v, next hour: %v", today, runToday, nextHour)
+			time.Sleep(time.Until(nextHour))
+
+			if time.Now().Hour() >= 10 && runToday {
+				runToday = false
+				log.Println("Running dornröschen")
+				if err := run(); err != nil {
+					log.Printf("failed: %v", err)
+				}
+			}
+		}
+	}
+}

dornröschen/main.go

@@ -0,0 +1,17 @@
+// +build !gokrazy
+
+package main
+
+import (
+	"flag"
+	"log"
+)
+
+func main() {
+	log.SetFlags(log.LstdFlags | log.Lshortfile)
+	flag.Parse()
+
+	if err := run(); err != nil {
+		log.Fatal(err)
+	}
+}