From bfee747b170cf483cf687dd6d627bdb1c666f11e Mon Sep 17 00:00:00 2001
From: NiseVoid <nisevoid@gmail.com>
Date: Mon, 14 May 2018 12:49:54 +0200
Subject: [PATCH] Retrieve data every minute

---
 main.go | 18 ++++++++++++------
 1 file changed, 12 insertions(+), 6 deletions(-)

diff --git a/main.go b/main.go
index d245225..333deaf 100644
--- a/main.go
+++ b/main.go
@@ -19,10 +19,9 @@ var db *qbdb.DB
 func main() {
 	initDB()
 
-	startDate := GetStartDate()
-
-	for time.Now().After(startDate) {
-		start, end := startDate, startDate.AddDate(0, 0, 7)
+	start := GetStartDate()
+	for {
+		end := start.Add(time.Hour)
 
 		data := GetFortopData(start, end)
 		ranges := map[string][]Range{}
@@ -46,7 +45,14 @@ func main() {
 
 		insertSets(sets)
 
-		startDate = end
+		if end.Before(time.Now()) {
+			start = end.Add(time.Second)
+			continue
+		}
+
+		time.Sleep(time.Minute)
+
+		start = GetStartDate()
 	}
 }
 
@@ -58,7 +64,7 @@ func GetStartDate() time.Time {
 	q := m.Select(qf.Max(m.Time))
 	err := db.QueryRow(q).Scan(&t)
 	if t == nil || err == sql.ErrNoRows {
-		return time.Date(2018, 1, 1, 0, 0, 0, 0, time.Local)
+		return time.Date(2018, 4, 1, 0, 0, 0, 0, time.Local)
 	}
 	if err != nil {
 		panic(err)